Software Engineer, Machine Learning Infrastructure

Stripe; Toronto; Hybrid; CAD 172,000 - 258,000; Full time

Benefits offered by this job

Equity

Company bonus or sales commissions/bon

Retirement plans

Health benefits

Wellness stipends

Job summary

Stripe is seeking an ML Infra engineer to help scale the ML lifecycle across data, analytics and AI/ML platforms. You will collaborate with ML engineers, data scientists and product teams to build services that accelerate experimentation and production deployment.

You’ll contribute to the next generation of ML infra capabilities, improving development velocity and MLOps maturity across Stripe’s systems, in a hybrid work model.

Qualifications

• 2+ years of professional software development experience with service oriented architectures.
• Experience with large-scale distributed systems and production ML platforms.
• Experience with ML workflows, MLOps, or LLM applications.

Responsibilities

• Design and build scalable, secure ML infra services for notebooks, model training, experimentation and serving.
• Collaborate with product engineers to improve day-to-day productivity and reliability.
• Own technical and product challenges across diverse systems and technologies.

Skills

Software development

Distributed systems

MLOps

LLM applications

Operations for high availability

Tools

LLM Frameworks

Job description

About Stripe

Stripe is a financial infrastructure platform for businesses. Millions of companies—from the world’s largest enterprises to the most ambitious startups—use Stripe to accept payments, grow their revenue, and accelerate new business opportunities. Our mission is to increase the GDP of the internet, and we have a staggering amount of work ahead. That means you have an unprecedented opportunity to put the global economy within everyone’s reach while doing the most important work of your career.

Who we are

Stripe is a financial infrastructure platform for businesses. Millions of companies—from the world’s largest enterprises to the most ambitious startups—use Stripe to accept payments, grow their revenue, and accelerate new business opportunities. Our mission is to increase the GDP of the internet, and we have a staggering amount of work ahead. That means you have an unprecedented opportunity to put the global economy within everyone’s reach while doing the most important work of your career.

About The Team

Stripe processes over $1T in payments volume per year, which is roughly 1% of the world’s GDP. The tremendous amount of data makes Stripe one of the best places to do machine learning. The ML Infra team builds services and tools that power every step in the ML lifecycle, including data exploration, feature generation, experimentation, training, deploying, serving ML models, and building LLM applications. With the phenomenal developments happening in the field of AI, we are positioned to accelerate the adoption of AI/ML across all parts of the company by building highly scalable and reliable foundational infrastructure.

What you’ll do

You will work closely with machine learning engineers, data scientists, and product engineering teams to enable seamless end-to-end experience in building solutions across data, analytics, and AI/ML platforms. You will build the next generation of ML Infra services and major new capabilities that substantially improve ML development velocity and MLOps maturity across the company.

Responsibilities

• Designing and building scalable, reliable, and secure services for notebooks, ML model training, experimentation, serving, and LLM applications across multiple regions.
• Creating services and libraries that enable ML engineers at Stripe to seamlessly transition from experimentation to production across Stripe’s systems.
• Working directly with product teams and ML engineers to improve their day-to-day productivity.
• Taking ownership of and finding solutions for technical and product challenges by working with a diverse set of systems, processes, and technologies.

Who you are

We’re looking for people with a strong background or interest in building successful products or systems; you’re passionate about solving business problems and making impact, you are comfortable in dealing with lots of moving pieces; and you’re comfortable learning new technologies and systems. You are comfortable working with other Stripe teams across the US and Canada.

Minimum Requirements

• 2+ years of professional software development experience with a solid background on service oriented architecture and large-scale distributed systems
• Experience working through the full life cycle of software development, from talking to users, to design and implementation, to testing and deployment, to operations
• Experience working on production ML platforms, MLOps solutions, or building LLM applications
• Experience running operations for high availability, low latency systems
• Experience partnering with other teams to drive business outcomes
• A sense of pragmatism: you know when to aim for the ideal solution and when to adjust course

Preferred Qualifications

• Experience building and shipping production AI agents
• Familiarity with the LLMs and LLM Frameworks
• Experience training and shipping machine learning models to production to solve critical business problems

Hybrid work at Stripe

This role is available either in an office or a remote location (35+ miles or 56+ km from a Stripe office).

In-office expectations

Office-assigned Stripes spend at least 50% of the time in a given month in their local office or with users. This hits a balance between bringing people together for in-person collaboration and learning from each other, while supporting flexibility about how to do this in a way that makes sense for individuals and their teams.

Working remotely at Stripe

A remote location is defined as being 35 miles (56 kilometers) or more from one of our offices. While you would be welcome to come into the office for team/business meetings, on-sites, meet-ups, and events, our expectation is you would regularly work from home rather than a Stripe office. Stripe does not cover the cost of relocating to a remote location. We encourage you to apply for roles that match the location where you currently live or plan to live.

Pay and benefits

The annual salary range for this role in the primary location is CA$172,000 - CA$258,000. This range may change if you are hired in another location. For sales roles, the range provided is the role’s On Target Earnings (“OTE”) range, meaning that the range includes both the sales commissions/sales bonuses target and annual base salary for the role. This salary range may be inclusive of several career levels at Stripe and will be narrowed during the interview process based on a number of factors, including the candidate’s experience, qualifications, and specific location. Applicants interested in this role and who are not located in the primary location may request the annual salary range for their location during the interview process.

Specific benefits and details about what compensation is included in the salary range listed above will vary depending on the applicant’s location and can be discussed in more detail during the interview process.

• equity
• company bonus or sales commissions/bonuses
• retirement plans
• health benefits
• wellness stipends

We look forward to hearing from you.

At Stripe, we're looking for people with passion, grit, and integrity. You're encouraged to apply even if your experience doesn't precisely match the job description. Your skills and passion will stand out—and set you apart—especially if your career has taken some extraordinary twists and turns. At Stripe, we welcome diverse perspectives and people who think rigorously and aren't afraid to challenge assumptions. Join us.