Drimva

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

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