AI/ML Manager, ML Foundations

Stripe • San Francisco, CA

95d • $306,000 - $459,000

About The Position

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. 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. 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 $1.4T in payments volume per year, which is roughly 1.3% of the world's GDP. We process petabytes of financial data using our ML platform to build features, train models, and deploy them to production. We use a combination of highly scalable and explainable models such as linear/logistic regression and random forests, along with the latest deep neural networks from transformers to LLMs. Some of our latest innovations have been around figuring out how best to bring transformers and LLMs to improve existing models and also enable entirely new product ideas that are only made possible by GenAI.

Requirements

• 10+ years of experience in machine learning, with a strong emphasis on modern technologies such as DNNs, Transformers, and Foundation Models.
• Proven experience in managing and developing a team of managers, fostering their growth and ensuring alignment with strategic objectives.
• A strong builder mentality, with the ability to define a team's charter and lead the development of complex systems from scratch.
• Proven ability to shepherd large, complex ML projects and drive transformational change in an organization.
• Deep passion for solving really interesting problems and for building the latest technologies rather than relying on outdated methods.

Nice To Haves

• A PhD or Master's degree with a research-oriented background, with the ability to dive into papers and stay current with academic publications.
• Experience with a large-scale, data-rich product in a domain such as payments, commerce, search, or social media.
• Knowledge of the challenges and opportunities in applying ML to fraud prevention, merchant intelligence, or financial services.

Responsibilities

• Lead the development of foundation models for payments, merchants, and other core product areas.
• Develop universal AI Assistants and Retrieval-Augmented Generation (RAG) and LLM systems to automate support and answer questions across Stripe's platforms.
• Drive the adoption of new ML and AI systems to automate merchant onboarding and workflow automation for a majority of our merchants.
• Build and lead a world-class team of ML engineers (MLEs), fostering a culture of technical excellence, collaboration, and continuous learning.

Benefits

• Equity
• Company bonus or sales commissions/bonuses
• 401(k) plan
• Medical, dental, and vision benefits
• Wellness stipends

What This Job Offers

• Job Type: Full-time
• Career Level: Senior
• Industry: Credit Intermediation and Related Activities
• Education Level: Master's degree