# Machine Learning Engineer

**Company:** [Zoox](https://hotfix.jobs/companies/zoox)
**Location:** Foster City, CA
**Role:** ML Engineering
**Salary:** $189k – $270k/yr
**Experience:** 5+ years
**Skills:** Reinforcement Learning, imitation learning, Transformers, Python, C++, machine learning pipelines, planning, prediction, Deep Learning
**Posted:** 2026-07-23

> Develop deep learning models using imitation and reinforcement learning for autonomous driving agents and planning. Requires advanced degree plus experience with RL, transformers, and production ML pipelines.

## Job Description

## Responsibilities
- Develop new deep learning models that use imitation learning and reinforcement learning to generate driving plans for human-like agents.
- Work on novel techniques to estimate the quality of those driving plans along the dimensions of safety, progress, comfort and realism.
- Contribute to our large-scale machine learning infrastructure to discover new solutions and push the boundaries of the field.
- Develop metrics and tools to analyze errors and understand improvements of our systems.
- Collaborate with engineers on Perception, Planning, Simulation, and Validation to solve the overall Autonomous Driving problem.

## Requirements
- PhD degree in computer science or related field or master's degree and 5+ years of professional experience in a relevant field.
- Experience in Planning and / or Prediction using Reinforcement Learning techniques.
- Experience with training and deploying transformer-based model architectures.
- Experience with production Machine Learning pipelines: dataset creation, training frameworks, metrics pipelines.
- Fluency in Python with a basic understanding of C++.

## Nice-to-Haves
- Top tier publications (NeurIPS, ICML, CVPR).

## Similar roles

- [Software Engineer - Behavior Capabilities](https://hotfix.jobs/jobs/07184ed0-bd8d-46fe-92cd-a18258309fb4) - Zoox - Foster City, CA - $189k – $270k/yr
- [Software Engineer - Learned Trajectory Machine Learning Engineer](https://hotfix.jobs/jobs/58417c3f-47c5-46eb-8ffb-27ad8502a9a3) - Zoox - Foster City, CA - $189k – $270k/yr
- [AI Engineer](https://hotfix.jobs/jobs/3e81f9d7-b85f-486a-849e-d1417b241b1e) - Zoox - Foster City, CA - $190k – $250k/yr
- [Software Engineer, Applied AI](https://hotfix.jobs/jobs/be4d97f1-9f2f-4973-bac2-0de1198d845e) - Norm Ai - New York, NY - $190k – $280k/yr
- [ML Engineer](https://hotfix.jobs/jobs/98ad2c38-046c-4b23-9f02-cb395af69959) - Sesame - San Francisco, CA - $190k – $320k/yr

**Apply:** https://hotfix.jobs/jobs/2dd18a1c-65ac-4e43-9a8a-6261305bb40e
**Canonical:** https://hotfix.jobs/jobs/2dd18a1c-65ac-4e43-9a8a-6261305bb40e