Drimva

Braze

Staff Machine Learning Engineer, ML Platform

$184K–$348K

San Francisco +1 · Staff · Full-time · Posted Sep 16, 2026

Location
San Francisco
New York City
Experience
8+ years building and operating distributed systems in production.

About the job

Own and evolve the production ML platform that trains, deploys, serves, and monitors customer-specific models at global scale.

What you'll do

  • Lead major infrastructure initiatives and deliver complex ML platform changes from design through production.
  • Set technical direction and production quality standards; lead incident response and improve reliability and cost efficiency.
  • Coordinate across partner teams, review designs and code, mentor senior engineers and data scientists, and connect technical choices to customer and business outcomes.

What they ask for

  • 8+ years building and operating distributed production systems, including deployment and operations.
  • Hands-on experience with production ML workloads, such as training pipelines, model serving, feature systems, or ML platform tooling.
  • Technical leadership across team direction and multi-quarter initiatives, plus experience growing senior engineers while maintaining strong individual contribution.
  • Deep working knowledge of Kubernetes and cloud infrastructure, including identity and access management, networking, and cost.

Nice to have

  • Queueing/orchestration systems, ML platform tooling, or experience with the listed stack (Python, Ruby on Rails, MongoDB, Redis, Kubernetes).
  • Experience with compliance regimes such as SOX or HIPAA, or customer engagement, personalization, or marketing technology.

Pay

$184,000 and $314,000/year USD — base pay range; $204,000 and $348,000/year USD — expected On Target Earnings (OTE), including bonus or commission

Tools

Kubernetes, Celery, RabbitMQ, Kafka, Ray, MLflow, Python, Ruby on Rails

AI focus

Build and operate production ML systems for personalized customer experiences, including customer-specific model training pipelines and high-throughput prediction serving.

Model-written summary · the original posting is authoritative

First seen Oct 27, 2026 · Last verified Nov 5, 2026 (UTC)