Senior / Staff Machine Learning Engineer - 3D Sensor Simulation

Zoox is building autonomous vehicle technology, and this role supports that mission through machine learning for 3D sensor simulation. You will help advance realism for simulated LiDAR, radar, and camera outputs, and contribute to synthetic data generation used to train and validate autonomy software.

This Senior / Staff Machine Learning Engineer position is based in Seattle, WA (onsite). The base salary range is USD 242,000 - 348,000 per year, with compensation also considering Amazon Restricted Stock Units (RSUs) and Zoox Stock Appreciation Rights. A sign-on bonus may be offered as part of the package. Leveling and positioning within a level depend on factors such as relevant experience, domain knowledge, and interview performance.

What you’ll do

• Design, train, and validate ML models that improve the realism of simulated LiDAR, radar, and camera data in AV simulation pipelines.
• Work with Synthetic Data and HIL Sim teams to integrate 3DGS and NeRF-based 3D reconstruction alongside simulation for multiple sensor modalities.
• Build and curate training datasets combining real and simulated sensor data, including GenAI-based data augmentation.
• Partner with Sim V&V and Perception engineers to define sensor realism metrics and reduce sim-to-real gaps with the highest impact on autonomy.
• Mentor engineers and interns, and establish best practices for training and deploying ML on Zoox sensor data.

Core skills and experience

• 2+ years of prior experience as a Technical Lead.
• 5+ years developing ML models in Python using PyTorch or TensorFlow.
• Industry experience applying ML to sensor data (LiDAR, radar, and/or cameras).
• Strong understanding of customizing, tuning, and validating ML training pipelines.

Technologies

• Python, PyTorch, TensorFlow
• 3DGS, NeRFs
• GenAI
• Diffusion, flow matching, VAEs

Benefits

• Paid time off (including sick leave, vacation, and bereavement)
• Unpaid time off
• Zoox Stock Appreciation Rights
• Amazon RSUs
• Health insurance
• Long-term care insurance
• Long-term and short-term disability insurance
• Life insurance

Bonus qualifications

• Experience with generative techniques such as diffusion, flow matching, or VAEs.
• Experience with 3D reconstruction (3DGS, NeRFs) or differentiable rendering.
• Prior experience developing autonomy in AVs or other robotics and computer vision.

If you need an accommodation to participate in the application or interview process, contact [email protected] or your assigned recruiter.

You do not need to match every listed expectation to apply. Zoox may use AI tools to support parts of the hiring process, such as reviewing applications, analyzing resumes, and assessing responses to identify potential inconsistencies or verification signals based on available information.