← All jobs · Stripe

Software Engineer, Machine Learning Infrastructure

Stripe ·
73
AI-Agency
B78 U65
📍 US 🌐 Remote/hybrid Mid 2+ yrs
PythonKubernetesdistributed systemsMLOps
TL;DR

Software engineer at Stripe building ML infrastructure services for the ML lifecycle, including training, serving, and LLM applications. Focus on scalable, reliable systems enabling seamless experimentation-to-production workflows.

Apply at Stripe →
you'll be redirected to the company's career page

Job description

<h2>Who we are</h2> <h3>About Stripe</h3> <p><span style="font-weight: 400;">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.</span></p> <h3>About the team</h3> <p>Stripe processes over $1T in payments volume per year, which is roughly 1% of the world’s GDP. The tremendous amount of data makes Stripe one of the best places to do machine learning. The ML Infra team builds services and tools that power every step in the ML lifecycle, including data exploration, feature generation, experimentation, training, deploying, serving ML models, and building LLM applications. With the phenomenal developments happening in the field of AI, we are positioned to accelerate the adoption of AI/ML across all parts of the company by building highly scalable and reliable foundational infrastructure.</p> <h2>What you’ll do</h2> <p>You will work closely with machine learning engineers, data scientists, and product engineering teams to enable seamless end-to-end experience in building solutions across data, analytics, and AI/ML platforms. You will build the next generation of ML Infra services and major new capabilities that substantially improve ML development velocity and MLOps maturity across the company.</p> <h3>Responsibilities<span style="font-weight: 400;">&nbsp;</span></h3> <ul> <li>Designing and building scalable, reliable, and secure services for notebooks, ML model training, experimentation, serving, and LLM applications across multiple regions.&nbsp;</li> <li>Creating services and libraries that enable ML engineers at Stripe to seamlessly transition from experimentation to production across Stripe’s systems.&nbsp;</li> <li>Working directly with product teams and ML engineers to improve their day-to-day productivity.&nbsp;</li> <li>Taking ownership of and finding solutions for technical and product challenges by working with a diverse set of systems, processes, and technologies.</li> </ul> <h2><strong>Who you are</strong></h2> <p>We’re looking for people with a strong background or interest in building successful products or systems; you’re passionate about solving business problems and making impact, you are comfortable in dealing with lots of moving pieces; and you’re comfortable learning new technologies and systems. You are comfortable working with other Stripe teams across the US and Canada.</p> <h3>Minimum requirements</h3> <ul> <li>2+ years of professional software development experience with a solid background on service oriented architecture and large-scale distributed systems</li> <li>Experience working through the full life cycle of software development, from talking to users, to design and implementation, to testing and deployment, to operations</li> <li>Experience working on production ML platforms, MLOps solutions, or building LLM applications</li> <li>Experience running operations for high availability, low latency systems</li> <li>Experience partnering with other teams to drive business outcomes</li> <li>A sense of pragmatism: you know when to aim for the ideal solution and when to adjust course</li> </ul> <h3><strong>Preferred qualifications</strong><span style="font-weight: 400;">&nbsp;</span></h3> <ul> <li>Experience building and shipping production AI agents</li> <li>Familiarity with the LLMs and LLM Frameworks</li> <li>Experience training and shipping machine learning models to production to solve critical business problems</li> </ul>
Apply at Stripe →

More open roles at Stripe

Stripe ·
Forward Deployed AI Accelerator, Marketing
📍 US 🌐 Remote 🛠 AI tools welcome · Mid
Forward Deployed AI Accelerator at Stripe's Marketing organization. Embed with marketing teams to build custom AI agents, automations, and tools that transform workflows. Coach marketers toward AI-first operations and scale successful patterns across the organization.
ClaudeClaude CodePythonAPI integrations
83
AI-core
Stripe ·
Forward Deployed AI Accelerator, Marketing
📍 US 🌐 Remote 🛠 AI tools welcome · Mid
Forward Deployed AI Accelerator at Stripe's Marketing organization. Embed with marketing teams to build custom AI agents, automations, and tools that transform workflows. Coach marketers toward AI-first operations and scale successful patterns across the organization.
ClaudeClaude CodePythonAPI integrations
83
AI-core
Stripe ·
Forward Deployed AI Accelerator, Marketing
📍 Singapore, SG 🛠 AI tools welcome · Mid
Forward Deployed AI Accelerator at Stripe embedded with marketing teams to build custom AI tools, agents, and automations. Coach marketers through AI adoption and systematically scale workflow transformations across the organization.
ClaudeClaude CodeAPI integrationsworkflow automation tools
81
AI-core
Stripe ·
Machine Learning Engineer, Stripe Assistant
📍 US 🌐 Remote 🛠 AI tools welcome · Senior
Senior Machine Learning Engineer at Stripe building the Stripe Assistant, an LLM-powered agent platform for merchant support and automation. Focus on agentic architecture, RAG, fine-tuning, and safe execution of high-trust actions at scale.
PythonLLMsRAGembeddingsRLHFagentic systems
81
AI-core
Stripe ·
Treasury Finance AI and Quantitative Analytics
📍 New York, US 🛠 AI tools welcome · Mid
Treasury Finance AI and Quantitative Analytics role at Stripe building autonomous agents and AI-powered tools for treasury workflows. Combines finance domain expertise with LLM frameworks (LangChain, LangGraph) and Python to automate liquidity management and risk analysis.
PythonPyTorchTensorFlowLangChainLangGraphStreamlit
79
AI-core