Staff Software Engineer, Machine Learning Platform
Core
Designing and building the platforms and services that enable ML engineers and data scientists to take data and build features and models from prototype to production at scale.
Role type
Staff Software Engineer, Machine Learning Platform
Builds
ML infrastructure, model inference systems, feature stores, and real-time monitoring tools for Stripe's ML-driven products.
Domain
Financial technology / Machine Learning Infrastructure
Deliverable
production ML models | infrastructure
Required skills
service-oriented architecture, large-scale distributed systems, technical leadership, product instincts, cross-functional collaboration, system design, low-latency optimization, LLM/agent orchestration
Preferred skills
large-scale serving/data infrastructure, LLM frameworks, agentic AI patterns, cloud services (AWS, SageMaker, Bedrock, Databricks), model training and shipping, distributed team management
Technologies
AWS, SageMaker, Bedrock, Databricks, OpenAI
Responsibilities
Take ownership of end-to-end architecture for large, complex ML Platform projects; Define technical directions for projects with high ambiguity; Design system architecture for low-latency model inference and large-scale feature stores; Scope and lead large projects with significant business impact; Work with ML engineers and data scientists to translate needs into scalable solutions; Arbitrate critical decisions balancing latency, reliability, cost, and security; Mentor and grow other engineers.
Seniority
Staff, technical lead & strategy