Machine Learning Engineer – Simulation Framework

Zoox

Machine Learning Engineer focusing on GPU-based simulation frameworks at Zoox. Developing complex simulation systems to improve ML efficiency and vehicle behavior alignment.

Posted 6/25/2026full-timeFoster City • California • 🇺🇸 United StatesMid-LevelSenior💰 $151,000 - $257,000 per yearWebsite

Tech Stack

Tools & technologies

PythonPyTorch

About the role

Key responsibilities & impact

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

What you’ll need

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.

Benefits

Comp & perks

Health insurance
Long-term care insurance
Long-term and short-term disability insurance
Life insurance
Paid time off (e.g. sick leave, vacation, bereavement)
Unpaid time off
Zoox Stock Appreciation Rights
Amazon Restricted Stock Units (RSUs)

ATS Keywords

✓ Tailor your resume

Applicant Tracking System Keywords

Tip: use these terms in your resume and cover letter to boost ATS matches.

Hard Skills & Tools

GPU-based simulationreinforcement learningdeep learning modelsC++Pythonproduction-ready codedata generationML algorithmssimulation architecturefidelity gap analysis

Certifications

PhDMaster’s