# Machine Learning Engineer - Simulation Framework

**Company:** [Zoox](https://hotfix.jobs/companies/zoox)
**Location:** Foster City, CA, Seattle, WA
**Role:** ML Engineering
**Salary:** $151k – $257k/yr
**Experience:** 7+ years
**Skills:** JAX, PyTorch, C++, Python, Reinforcement Learning, CUDA, Gpu Programming, Deep Learning
**Posted:** 2026-06-24

> Machine Learning Engineer focused on GPU-based simulation frameworks, reinforcement learning, and bridging sim-to-real gaps for autonomous vehicle safety validation. Requires MS/PhD and strong C++/Python experience.

## Job Description

## Responsibilities
- Develop and optimize our GPU-based simulation framework to support complex machine learning training and validation pipelines.
- Apply reinforcement learning concepts to solve complex behavioral and path planning challenges in simulation environments.
- Identify and resolve "sim-to-sim" and "sim-to-real" fidelity gaps to ensure parity between high-speed ML simulations, high-fidelity 3D environments, and physical vehicle execution.
- Build systems that allow autonomy users to self-serve data generation and accelerate their training iterations.
- Write robust, production-ready code to integrate advanced ML algorithms directly into our core simulation architecture.

## Requirements
- PhD or Master’s in computer science, robotics, machine learning, or a related field.
- Deep understanding of reinforcement learning and its application in simulated or robotic environments.
- Hands-on experience developing, training, and fine-tuning deep learning models using modern frameworks (e.g., JAX or PyTorch).
- Strong proficiency in C++ and Python for building and deploying production machine learning systems.
- Experience analyzing and bridging fidelity gaps between synthetic training data and real-world execution.

## Nice-to-Haves
- Experience with GPU programming (CUDA) or high-performance compute clusters.
- Automotive or autonomous robotics industry experience.
- Strong background in deterministic systems and latency optimization.

## Similar roles

- [Senior C++ Software Engineer](https://hotfix.jobs/jobs/2e187e05-ea6e-4e57-acf8-9b3e3d25eeb8) - Applied Intuition - Sunnyvale, CA - $150k – $250k/yr
- [Senior C++ Software Engineer](https://hotfix.jobs/jobs/6fa17829-84c2-459c-af02-99f0a3be9db6) - Applied Intuition - Poway, CA - $150k – $225k/yr
- [Senior C++ Software Engineer](https://hotfix.jobs/jobs/d437d7dc-bb7e-4df2-bf62-7a4391d93214) - Applied Intuition - Ann Arbor, MI - $150k – $200k/yr
- [Senior Machine Learning Engineer, Computer Vision](https://hotfix.jobs/jobs/a3511009-a06a-4bce-9981-cd4669df5771) - Metropolis - Seattle, WA - $150k – $200k/yr
- [Senior Neural Rendering Software Engineer](https://hotfix.jobs/jobs/edb7e3c2-feb5-4a67-9220-45a797134aba) - Applied Intuition - Sunnyvale, CA - $150k – $250k/yr

**Apply:** https://hotfix.jobs/jobs/f38163d6-88a9-4ed1-9c29-b1f4e610f320
**Canonical:** https://hotfix.jobs/jobs/f38163d6-88a9-4ed1-9c29-b1f4e610f320