Braze
Staff Machine Learning Engineer, ML Platform
$184K–$348K
Austin +1 · Staff · Full-time · Posted Sep 16, 2026
- Location
- Austin
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 and AI-powered marketing solutions at global scale.
What you'll do
- Lead complex ML infrastructure initiatives from design through production, including model-serving fleets, pipelines, and CI/deployment tooling.
- Set technical direction and production quality standards; lead incident response and improve reliability, efficiency, and cost.
- Coordinate platform work across teams, review designs and code, mentor engineers and data scientists, and connect technical decisions to business outcomes.
What they ask for
- 8+ years building and operating distributed production systems, including deployment and operations.
- Hands-on production experience with ML workloads such as training pipelines, model serving, feature systems, or ML platform tooling.
- Technical leadership across teams and multi-quarter initiatives, with experience mentoring senior engineers.
- Deep working knowledge of Kubernetes and cloud infrastructure, including IAM, networking, and cost management.
Nice to have
- Queueing/orchestration or ML platform tools; experience with the stated stack, compliance regimes, or customer engagement/personalization/marketing technology.
Pay
$184,000 and $314,000/year USD base salary; $204,000 and $348,000/year USD total pay (OTE, including bonus or commission)
Tools
Kubernetes, Celery, RabbitMQ, Kafka, Ray, MLflow, Python, Ruby on Rails
AI focus
Build and operate ML and AI-enhanced marketing solutions, including customer-specific model training pipelines and high-throughput prediction serving; deep modeling experience is a plus, not a requirement.
Model-written summary · the original posting is authoritative