Your browser does not support javascript! Please enable it, otherwise web will not work for you.

Staff Machine Learning Engineer

Home > Python programming jobs

Staff Machine Learning Engineer in USA

  • San Francisco, CA

Responsibilities

  • Drive transformative initiatives for ML production infrastructure, including queueing, orchestration, cloud identity, deployment, and infrastructure retirement.
  • Build and operate multi-region model-serving systems, customer-specific model pipelines, and CI and deployment tooling.
  • Set the technical direction for model training, deployment, serving, observability, reliability, and cost efficiency.
  • Lead incident response and production-quality practices for ML systems.
  • Coordinate technical initiatives across teams that share infrastructure, deployment tooling, and data systems.
  • Improve engineering quality through design reviews, code reviews, production-readiness reviews, and mentoring.
  • Represent the team’s technical perspective to product and engineering leadership and connect technical decisions to business outcomes.

Requirements

  • 8+ years building and operating distributed systems in production, including deployment, operations, scale, reliability, CI/CD, and infrastructure as code.
  • Hands-on experience operating ML workloads in production, such as training pipelines, model serving, feature systems, or ML platform tooling.
  • Experience owning technical direction, leading multi-quarter cross-team initiatives, mentoring senior engineers, and maintaining high personal output.
  • Deep working knowledge of Kubernetes and cloud infrastructure, including identity and access management, networking, and cost management.
  • Strong verbal and written communication skills with the ability to build consensus and drive decisions.
  • Preferred experience with Celery, RabbitMQ, Kafka, Ray, MLflow, model registries, feature stores, ML observability, Python, Ruby on Rails, MongoDB, Redis, SOX, HIPAA, customer engagement, personalization, or marketing technology.

Benefits

  • Base pay for U.S.-based candidates is listed at $184,000-$314,000 per year; the role may also include equity and other compensation components.
  • Hybrid ways of working with a curated in-office employee experience.
  • Retirement and Employee Stock Purchase Plans, flexible paid time off, and medical, dental, vision, life, and disability coverage.
  • Fertility benefits and equal paid parental leave.
  • Professional development, formal career pathing, learning platforms, and a yearly learning stipend.
  • Volunteer Week, donation matching, Employee Resource Groups, and a collaborative workplace culture.

Salary: $184k - $314k/yr

Braze

Braze is the leading customer engagement platform that empowers brands to Be Absolutely Engaging.™ Braze allows any marketer to collect and take action on any amount of data from any source, so they can creatively engage with customers in real time, across channels from one platform. From cross-c...

Similar positions

Senior, Machine Learning Engineer

  • Torc Robotics
  • Full time
  • USA
  • 09/22/2026
  • Salary: $177k - $213k/yr
  • Remote

Senior Software Engineer

  • Commonwealth Bank of Australia
  • Full time
  • Australia
  • 09/22/2026
  • Sydney

Deployed Engineer, Professional Services

  • LangChain
  • Full time
  • Remote
  • 09/22/2026
  • Salary: Competitive
  • APAC

Senior AI Engineer

  • Jeeves
  • Full time
  • Remote
  • 09/22/2026
  • Brazil

Senior Product Security Engineer

  • AlphaSense
  • Full time
  • USA
  • 09/22/2026
  • Salary: $182k - $228k/yr
  • Remote