Staff Machine Learning Engineer (Tech Lead), Stripe Assistant

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

The Stripe Assistant team is transforming how users interact with Stripe by building an intelligent and proactive assistant that not only answers users’ queries but efficiently resolves issues and provides valuable business insights. We leverage LLMs and agentic systems to elevate the user experience across Stripe—from the dashboard to support surfaces—and we enable other teams to build and integrate their AI agents on our platform. We’re evolving from a helpful support tool to a trusted pilot that anticipates, optimizes, and executes on behalf of our users.

What you’ll do

As a Staff Machine Learning Engineer on the Stripe Assistant team, you’ll own the end-to-end ML and agent architecture that makes Stripe Assistant safe, reliable, and deeply useful. You’ll set the strategy for how the Assistant executes high-trust actions, delivers accurate analytical answers across Stripe and the broader web, orchestrates capabilities across many tools and agents, and grounds responses in authoritative Stripe and user data—so users can resolve issues quickly and confidently.

You’ll drive conversation continuity and personalization across surfaces, evolve the Assistant into a proactive partner that anticipates user needs, and deepen its presence in the dashboard to streamline critical workflows. You’ll establish rigorous evaluation and SLOs and deliver step‑change improvements in quality, latency, cost, and availability—paving the way for configurable levels of autonomy and, ultimately, a dependable operating layer over a merchant’s Stripe account.

Responsibilities

Our team operates fluidly and here are some problems you may tackle:

  • Establish trustworthy, human-in-the-loop execution for high-trust “write” actions—prioritizing user control, transparency, accountability, and auditability so customers can delegate with confidence.
  • Define and evolve the Assistant’s capability and governance model across hundreds of tools and agents, balancing power, permissions, and consistency at scale.
  • Raise answer quality and usefulness by grounding in authoritative Stripe knowledge and live user data, building cross-surface memory and personalization, and making the Assistant proactive and present in the dashboard.
  • Explore and apply optimal machine learning methods to improve Stripe Assistant’s overall performance, including but not limited to fine-tuning LLMs with RLHF, synthetic data generation, optimizing RAG pipelines via domain‑specific embedding and retriever fine‑tuning, and automatic prompt tuning, etc.
  • Make quality and reliability a product: set and meet SLOs, build rigorous evaluation and benchmarking loops, and drive sustained improvements in latency, cost, and availability.
  • Lead as a tech lead: mentor and grow engineers, uphold high bars for code quality, security, observability, and operational rigor, and align cross‑functionally to ship safely and fast.

Who you are

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

  • 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.

Preferred qualifications

  • Experience operating ML systems at global scale with stringent SLOs—balancing reliability, latency, and cost—with privacy, security, and compliance by design.
  • Experience building products where AI/ML is core; as well as balancing short-term product priorities with long-term AI/ML improvements.
  • Track record building ML platforms, especially those that enable multiple teams to collaborate together.
  • Strong technical leadership and communication: mentoring and elevating engineers, elevating AI/ML awareness and posture within organizations, setting architectural direction, and driving alignment in ambiguity.

Join us to build a trustworthy, proactive AI operating layer for every Stripe merchant—advancing safety, reliability, and insight at global scale. If you’re ready to help take Stripe Assistant from copilot to full autopilot and shape how businesses connect with Stripe, we’d love to hear from you.

In-office expectations

Office-assigned Stripes in most of our locations are currently expected to spend at least 50% of the time in a given month in their local office or with users. This expectation may vary depending on role, team and location. For example, Stripes in our Bucharest, Romania site have an 80% in-office expectation, and those in Stripe Delivery Center roles in Mexico City, Mexico and Bengaluru, India work 100% from the office. Also, some teams have greater in-office attendance requirements, to appropriately support our users and workflows, which the hiring manager will discuss. This approach helps strike a balance between bringing people together for in-person collaboration and learning from each other, while supporting flexibility when possible.

Pay and benefits

The annual US base salary range for this role is $212,000 - $318,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.

Additional benefits for this role may include: equity, company bonus or sales commissions/bonuses; 401(k) plan; medical, dental, and vision benefits; and wellness stipends.

Office locations

South San Francisco HQ, New York, or Seattle

Team

Machine Learning

Job type

Full time

Please find our California applicant personal information notice here.

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.