Machine Learning Engineer, Foundation Model

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.

About The Team

Our Applied ML team aims to reform how our users interact with Stripe. We are doing so by (a) automating the easy tasks, and (b) assisting our users in the difficult tasks. Some examples include helping our users resolve issues with Stripe faster or making it easier for our users to sign up and navigate Stripe. We are using the latest LLMs as well as fine-tuning our own models. We're an end-to-end team going from ideas to models to shipping in production.

What You’ll Do

As a machine learning engineer, you will be responsible for analyzing opportunities, proposing ideas, training & evaluating ML models, running experiments, and deploying everything to production. You will also have the opportunity to contribute to and influence ML architecture at Stripe as well as be a part of a larger ML community.

Responsibilities

• Evaluate systems offline & online.
• Improve performance to match (and beat) humans.
• Ensure model quality doesn’t degrade online.
• Fine-tune LLMs for better performance.
• Invest in the right OSS and in-house platforms.

Who You Are

We are looking for ML Engineers who are passionate about using ML to improve products and delight customers. You have experience developing streaming feature pipelines, building ML models, and deploying them to production, even if it involves making substantial changes to backend code. You are comfortable with ambiguity, love to take initiative, and have a bias towards action.

Minimum Requirements

• At least 3 years of experience shipping ML systems in production.
• High standards for working with production systems.
• Ownership and drive for projects to business impact.
• Thrive in a collaborative environment.

Preferred Qualifications

• 5+ years of experience in full-time software development roles.
• Experience shipping LLM integrations to user products with high quality.
• Experience operating in highly ambiguous environments.
• Knowledge about driving a hypothesis from data.

Hybrid Work at Stripe

This role is available either in an office or a remote location (typically, 35+ miles or 56+ km from a Stripe office). 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.

A remote location, in most cases, 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 or plan to live.

Pay and Benefits

The annual US base salary range for this role is $173,100 - $316,800. 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.