Staff Engineer, Machine Learning Platform

Stripe Stripe · Fintech · Canada · 8212 ML Foundations

Stripe's ML Platform team is seeking a Staff Engineer to lead the technical direction and architecture for their ML infrastructure. This role involves building and scaling platforms for ML engineers and data scientists, focusing on areas like low-latency inference, feature stores, monitoring, and LLM/agent orchestration. The goal is to increase ML velocity and MLOps maturity across the company.

What you'd actually do

  1. Take ownership of end-to-end architecture and system design for large, complex projects across ML Platform.
  2. Define technical directions for projects with high ambiguity, transforming complex user needs into long-lasting platform strategy.
  3. Design the system architecture and solutions for the most challenging problems in the ML Platform domain, including low-latency model inference, large-scale feature stores, real-time monitoring, and LLM/agent orchestration.
  4. Turn high-leverage ideas into tangible, robust solutions that shape platform and product roadmap , combining technical excellence with creative problem-solving.
  5. Scope and lead large projects with significant business impact, driving them from requirements through design, implementation, and production operation.

Skills

Required

  • 10+ years of professional software development experience
  • service-oriented architecture
  • large-scale distributed systems
  • technical lead
  • production ML platform services
  • product instincts
  • business context understanding
  • communication skills
  • cross-functional collaboration
  • autonomy and responsibility
  • ambiguous environments
  • Hands on experience using AI tools

Nice to have

  • building large-scale serving or data infrastructure for machine learning use cases
  • model inference
  • feature stores
  • real-time feature computation
  • model registries
  • LLMs
  • LLM frameworks
  • agentic AI patterns
  • tool use
  • multi-agent orchestration
  • retrieval-augmented generation
  • rapidly developing prototypes
  • cloud services (e.g., AWS)
  • cloud-based AI/ML services (e.g., SageMaker, Bedrock, Databricks, OpenAI)
  • training and shipping machine learning models to production
  • synthesize ideas across the organization
  • setting a compelling technical vision
  • working with geographically distributed teams
  • side-projects, open source, or self-driven technical initiatives

What the JD emphasized

  • production ML platform services
  • low-latency model inference
  • large-scale feature stores
  • LLM/agent orchestration
  • MLOps maturity

Other signals

  • ML Platform
  • low-latency model inference
  • large-scale feature stores
  • LLM/agent orchestration
  • MLOps maturity