Stripe Logo

Stripe

Machine Learning Engineer, Growth Platform

Posted 2 Days Ago
In-Office
San Francisco, CA, USA
Mid level
In-Office
San Francisco, CA, USA
Mid level
Build and operate production machine learning systems for recommendations, ranking, personalization, contextual bandits, and agent-based recommendations. Responsibilities include model development, feature and data pipelines, evaluation, deployment, monitoring, experimentation, and reliability improvements. The role partners with engineering, data science, product, marketing, and sales teams to translate business problems into measurable ML outcomes.
The summary above was generated by AI
Who we areAbout 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.
HQ

Stripe San Francisco, California, USA Office

510 Townsend St, San Francisco, CA, United States, 94103

Similar Jobs

One Month Ago
In-Office
San Francisco, CA, USA
150K-220K Annually
Mid level
150K-220K Annually
Mid level
Information Technology • Productivity • Software
As a founding Machine Learning Engineer, you will improve core models, optimize LLM inference, develop data pipelines, and ship features directly to users.
Top Skills: Chrome ExtensionCloudflare WorkersElectron AppLlmsNode.jsReactTransformer ArchitecturesTypescriptVector Databases
One Month Ago
Hybrid
Menlo Park, CA, USA
200K-386K Annually
Expert/Leader
200K-386K Annually
Expert/Leader
AdTech • Cloud • Machine Learning • Mobile • Retail
As a Machine Learning Engineer, you'll design and deploy large-scale models for advertising and commerce systems, optimizing for performance and relevance while maintaining ML pipelines.
Top Skills: AWSBigQueryBigtableGCPJaxKubernetesPythonPyTorchSparkSQLTensorFlow
26 Days Ago
In-Office
San Francisco, CA, USA
148K-240K Annually
Mid level
148K-240K Annually
Mid level
Artificial Intelligence • Software
The Machine Learning Engineer will design and implement ML pipelines and productionize models, collaborating with scientists and engineers on scalable systems.
Top Skills: CudaFastapiGrpcHuggingfacePythonPyTorch

What you need to know about the San Francisco Tech Scene

San Francisco and the surrounding Bay Area attracts more startup funding than any other region in the world. Home to Stanford University and UC Berkeley, leading VC firms and several of the world’s most valuable companies, the Bay Area is the place to go for anyone looking to make it big in the tech industry. That said, San Francisco has a lot to offer beyond technology thanks to a thriving art and music scene, excellent food and a short drive to several of the country’s most beautiful recreational areas.

Key Facts About San Francisco Tech

  • Number of Tech Workers: 365,500; 13.9% of overall workforce (2024 CompTIA survey)
  • Major Tech Employers: Google, Apple, Salesforce, Meta
  • Key Industries: Artificial intelligence, cloud computing, fintech, consumer technology, software
  • Funding Landscape: $50.5 billion in venture capital funding in 2024 (Pitchbook)
  • Notable Investors: Sequoia Capital, Andreessen Horowitz, Bessemer Venture Partners, Greylock Partners, Khosla Ventures, Kleiner Perkins
  • Research Centers and Universities: Stanford University; University of California, Berkeley; University of San Francisco; Santa Clara University; Ames Research Center; Center for AI Safety; California Institute for Regenerative Medicine

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account