Machine Learning Engineer, Foundation Model

Who we are

About Stripe

Stripe’s mission is to accelerate global economic and technological development. We offer financial infrastructure and a variety of services to serve the needs of a wide range of users, from startups to enterprises, with global scale and industry-leading reliability and product quality. All financial services businesses face a trade-off between access, which we want to expand, and risk, which we want to minimize. We use machine learning to scalably and intelligently optimize across both.

Machine learning is an integral part of almost every service at Stripe. It is a key investment area with products and use cases that span merchant and transaction risk, payments optimization, identity, and merchant data analytics and insights. We are also using the latest generative AI technologies (such as LLMs and FMs) to re-imagine product experiences and developing AI Assistants and Agents both for our customers (e.g. Radar Assistant and Sigma Assistant), and also to make Stripes more productive across Support, Marketing, Sales, and Engineering roles within the company.

About the team

We are dedicated to building and shipping the foundational AI and machine learning systems that will power our entire product suite. Our mission is to fundamentally transform how Stripe uses ML, leveraging our extensive and rich dataset to solve some of the most challenging problems in payments and fraud. We work closely with our partners in Risk, Payments, and Support to build transformative technologies that have a direct impact on our users.

From a data perspective, Stripe handles over per year, which is roughly . We process petabytes of financial data using our ML platform to build features, train models, and deploy them to production. We focus on seeing how LLMs can solve some of our hardest problems in merchant risk and understanding how we can align language to our immense ocean of payments data. Some of our latest innovations have been around understanding how to best represent payments using transformers and enabling entirely new product ideas that are only made possible by GenAI.

What you'll do

As a Machine Learning Engineer on the Foundation Model team, you'll solve some of Stripe's most challenging technical problems that span multiple teams and directly impact our research and engineering efforts around building the Foundation Models that power our payments and risk solutions . You'll be responsible for both hands-on technical contributions and driving strategic initiatives that shape how ML systems operate at scale across Stripe.

Responsibilities

• Develop foundation models for payments, merchants, and consumers that span Stripe product areas.
• Fine-tune and optimize LLMs using SOTA techniques, including multimodal alignment, knowledge distillation, quantization, and reasoning to build highly specialized models.
• Drive technical excellence through hands-on contributions to the design and development of state-of-the-art AI / ML systems, conducting architecture reviews, and maintaining high code quality
• Partner with engineering and product leaders across Stripe to identify and prioritize foundational investments such as foundation models, assistants, and agents that unlock new capabilities for product teams
• Contribute to Stripe's technical strategy by representing AI / ML engineering perspectives in company-wide technical decisions and roadmap planning
• Mentor ML engineers across Stripe on ML experimentation, helping teams navigate complex technical trade-offs, and adopt platform capabilities effectively

Who you are

We're looking for someone who meets the minimum requirements to be considered for the role. If you meet these requirements, you are encouraged to apply. The preferred qualifications are a bonus, not a requirement.

Minimum requirements

Preferred qualifications

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 US base salary range for this role is $212,000 - $318,000. 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 location. Applicants interested in this role and who are not located in the US may request the annual salary range for their location during the interview process.

Additional benefits for this role may include : equity, company bonus or sales commissions / bonuses; 401(k) plan; medical, dental, and vision benefits; and wellness stipends.