Machine Learning Engineer, Growth Platform

Stripe; United States; Remote; USD 180,000 - 270,000; Full time

Benefits offered by this job

equity

401(k) plan

medical, dental, and vision benefits

wellness stipends

Job summary

Stripe is seeking a machine learning engineer to build and operate production ML systems that improve how we recommend products and actions to users. You will own from problem definition through deployment, working with data scientists, engineers, and product partners to drive measurable outcomes.

You will design, train, evaluate, and monitor models, optimize latency and cost, and translate business problems into scalable technical solutions while collaborating across Stripe teams.

Qualifications

• 3+ years of industry experience building and shipping ML models in production.
• Strong programming in Python and writing maintainable, tested production code.
• Experience designing, training, and evaluating ML models with PyTorch, TensorFlow, XGBoost, or scikit-learn.
• Experience building data or feature pipelines, proficiency in SQL, and familiarity with Spark / PySpark.
• A solid understanding of statistics, model evaluation, and experimentation with leakage awareness.
• Experience deploying, monitoring, debugging production ML systems with latency and cost considerations.
• Ability to translate open-ended business problems into technical approaches and collaborate cross-functionally.

Responsibilities

• Design, train, evaluate, deploy, and maintain models for recommendations and ranking.
• Improve contextual bandit and policy-learning approaches including exploration and reward design.
• Develop agent-based recommendation capabilities using business context for relevant products.
• Build data pipelines for training and inference with data freshness and feature quality.
• Create reusable tooling for model evaluation, retraining, and safe rollout.
• Own the quality, monitoring, and reliability of ML components and pipelines.

Skills

ML engineering experience

Python programming

Model evaluation

Production ML systems

Cross-functional collaboration

Tools

PyTorch

TensorFlow

XGBoost

scikit-learn

SQL

Spark / PySpark

Job description

Who we are

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

Growth Platform builds the machine learning systems that help businesses discover and use the Stripe products that meet their needs. Our recommendations reach users across the Dashboard, email, onboarding, documentation, and AI agent interfaces. We combine an understanding of each business with models that decide which recommendation is useful, when to show it, and how to learn from the outcome.

Our work spans recommendation and ranking models, contextual bandits, agent-based recommendations, and the data and evaluation systems behind them. We build shared capabilities that product, marketing, and sales teams can use across Stripe. Success means helping businesses take useful actions and adopt products that help them grow, while keeping recommendations relevant and avoiding unnecessary messages.

What you’ll do

You will build and operate production ML systems that improve how Stripe recommends products, content, and next steps to businesses. You will own work from problem definition and feature development through training, evaluation, deployment, monitoring, and iteration. Working with data scientists, engineers, and product partners, you will turn model improvements into measurable user and business outcomes.

Responsibilities

• Design, train, evaluate, deploy, and maintain models for recommendation, ranking, and personalized action selection across Growth Platform surfaces.
• Improve contextual bandit and policy-learning approaches, including exploration, reward design, and how recommendations adapt to user context and feedback.
• Build agent-based recommendation capabilities that use business context to identify relevant products and integration options, with evaluations that test recommendation quality and usefulness.
• Develop reliable data and feature pipelines for training and inference. Improve data freshness, feature quality, and consistency between training and production.
• Build reusable tooling for model evaluation, retraining, and safe rollout so the team can test and ship improvements faster.
• Own the quality and operation of the team's ML components: write tested production code, monitor models and pipelines, investigate failures, and improve reliability, latency, and cost.
• Design and analyze online experiments with data science partners. Connect offline evaluation to product adoption and incremental impact, with guardrails for dismissals, unsubscribes, and user experience.
• Partner with product engineering to integrate models into recommendation delivery systems, and with ML infrastructure teams to use and improve Stripe's shared training, feature, and serving capabilities.
• Work with product, marketing, and sales partners to identify problems that shared ML capabilities can solve, and make practical choices about where modeling adds value.

Who you are

You are a machine learning engineer with a builder mindset. You care about the business problem, the quality of the model, and what happens after it ships. You can move between modeling and software engineering, make practical tradeoffs, and take ownership of an ambiguous problem through production and measurement.

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

• 3+ years of industry experience in machine learning engineering, software engineering, or applied data science, with hands‑on experience building and shipping ML models in production.
• Strong programming skills in Python and experience writing maintainable, tested production code.
• Practical experience designing, training, and evaluating ML models using frameworks such as PyTorch, TensorFlow, XGBoost, or scikit-learn.
• Experience building data or feature pipelines, proficiency in SQL, and familiarity with distributed data processing tools such as Spark or PySpark.
• A strong understanding of statistics, model evaluation, and experimentation, including the ability to recognize data leakage and distinguish offline model improvements from business impact.
• Experience deploying, monitoring, and debugging production ML systems, and evaluating tradeoffs among model quality, reliability, latency, and cost.
• Ability to turn an open‑ended business problem into a technical approach and collaborate effectively with engineering, data science, product, and business partners.

Preferred qualifications

• Experience with recommendation systems, ranking, personalization, or marketplace and advertising optimization.
• Experience with contextual bandits, policy learning, causal inference, or off‑policy evaluation.
• Experience building and evaluating LLM applications, including structured extraction, embeddings, or recommendations grounded in user and business context.
• Experience building reusable ML capabilities used by multiple products or teams, including training automation, feature systems, or model monitoring.
• Experience with product growth, lifecycle messaging, or systems that balance short‑term engagement with longer‑term user outcomes.

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 $180,000 - $270,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.

• equity, company bonus or sales commissions/bonuses
• 401(k) plan
• medical, dental, and vision 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.

Please find our California applicant personal information notice here.

The application window will remain open for 100 days after the Job Post is published. However, this opportunity will remain open based on the needs of the business, which may cause the application window to close before or after the 100-day mark.

Company

Stripe

Team

Data & Data Science

Office locations

Chicago, New York, Seattle, South San Francisco HQ

Remote location

Remote in United States

Employment type

Full time