Senior / Staff Machine Learning Engineer - 3D Sensor Simulation

Benefits-forward role in Seattle with Zoox’s 3D Sensor Simulation team. You’ll apply machine learning to LiDAR, radar, and camera data to make simulated sensors more realistic and to generate synthetic data that supports training and testing for autonomous vehicle AI. This onsite position offers a base salary range of USD 242,000 to 348,000 per year, plus additional compensation components.

What you’ll do

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

What you’ll bring

• 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.

Relevant technologies

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

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.

Compensation and time off

Zoox compensation includes base salary, Amazon Restricted Stock Units (RSUs), and Zoox Stock Appreciation Rights. A sign-on bonus may be offered as part of the compensation package. The posting’s salary range applies only to the base salary and may vary by geographic location and level. Final leveling and positioning depend on factors such as years of relevant experience, domain knowledge, and interview performance.

• Paid time off (sick leave, vacation, bereavement)
• Unpaid time off

Insurance and coverage

• Health insurance
• Long-term care insurance
• Long-term and short-term disability insurance
• Life insurance

Accommodations

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

Final note: 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, or assessing responses and identifying potential inconsistencies or verification signals in application materials based on available information.