Staff Machine Learning Engineer – Tech Lead
Stripe
full-time
Posted on:
Location Type: Remote
Location: Remote • California • 🇺🇸 United States
Visit company websiteJob Level
Senior
Tech Stack
Distributed SystemsPythonRuby
About the role
- Own the end-to-end ML and agent architecture that makes Stripe Assistant safe, reliable, and deeply useful.
- Set the strategy for how the Assistant executes high-trust actions, delivers accurate analytical answers across Stripe and the broader web.
- Drive conversation continuity and personalization across surfaces, evolve the Assistant into a proactive partner that anticipates user needs.
- Establish rigorous evaluation and SLOs, deliver step‑change improvements in quality, latency, cost, and availability.
- Lead as a tech lead: mentor and grow engineers, uphold high bars for code quality, security, observability, and operational rigor.
Requirements
- 8+ years in AI/ML and backend engineering (4+ years building and operating production ML systems) with technical leadership.
- Deep and up-to-date applied LLM experience: RAG/embeddings, tool use/function calling, agentic planning/orchestration, fine-tuning, code generation, evaluations, etc.
- Proficient in Python (Ruby is a plus); strong distributed systems fundamentals.
- Experience working closely with product management, design, other engineers, and other cross-functional partners.
Benefits
- Not specified 📊 Resume Score Upload your resume to see if it passes auto-rejection tools used by recruiters Check Resume Score
Applicant Tracking System Keywords
Tip: use these terms in your resume and cover letter to boost ATS matches.
Hard skills
machine learningbackend engineeringproduction ML systemsapplied LLMRAGembeddingstool usefunction callingagentic planningPython
Soft skills
technical leadershipmentoringcode qualitysecurityobservabilityoperational rigorcollaborationcommunication