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

  • Austin, TX

Responsibilities

  • Drive transformative initiatives for ML production infrastructure, including queueing, orchestration, deployment, cloud identity, and infrastructure retirement.
  • Build and operate multi-region model-serving fleets, customer-specific model pipelines, and CI and deployment tooling.
  • Set the technical vision and production quality bar for model training, deployment, serving, observability, reliability, and cost efficiency.
  • Lead incident response for ML systems and coordinate initiatives across shared infrastructure, deployment, and data-platform teams.
  • Raise engineering quality through design reviews, code reviews, production readiness practices, and mentoring.
  • Connect technical decisions to customer and business outcomes and represent the team’s technical perspective to leadership.

Requirements

  • 8+ years building and operating distributed systems in production, including deployment, operations, scalability, reliability, CI/CD, and infrastructure as code.
  • Hands-on experience with production ML workloads such as training pipelines, model serving, feature systems, or ML platform tooling.
  • Technical leadership experience setting team direction, leading multi-quarter cross-team initiatives, and developing senior engineers while 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. candidates is listed at $184,000-$314,000 per year, with potential equity through restricted stock units; compensation figures excluded from structured benefits.
  • Hybrid work arrangement 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 through 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

Graduate Software Engineer

  • Synack
  • Full time
  • UK
  • 10/04/2026
  • Remote

Software Engineer - Cloud Images

  • Canonical
  • Full time
  • Remote
  • 10/04/2026
  • EMEA, Americas

GTM Engineer

  • Higgsfield
  • Full time
  • USA
  • 10/04/2026
  • Salary: $130k - $175k/yr
  • San Francisco, CA

Research Engineer

  • Clera
  • Full time
  • Others
  • 10/04/2026
  • Zürich, Switzerland

Lead Automation Quality Engineer

  • London Stock Exchange Group
  • Full time
  • Others
  • 10/04/2026
  • Salary: Competitive
  • Bengaluru, India