Senior / Staff Machine Learning Engineer - 3D Sensor Simulation

Location: Hybrid in Foster City, California or Seattle, Washington, United States

Company

Zoox develops autonomous vehicle technology and robotics systems for self-driving mobility.

What you will do

• Design, train, and validate machine learning models that improve simulated lidar, radar, and camera data.
• Integrate 3DGS and NeRF-based 3D reconstruction with simulation across sensor modalities.
• Build and curate datasets from real and simulated sensor data, including GenAI-based augmentation.
• Define sensor realism metrics and address high-impact sim-to-real gaps with simulation, verification, validation, and perception engineers.
• Mentor engineers and interns and establish best practices for training and deploying machine learning models on sensor data.

Requirements

• At least 5 years of experience developing machine learning models in Python with PyTorch or TensorFlow.
• At least 2 years of prior experience as a technical lead.
• Industry experience applying machine learning to lidar, radar, and/or camera data.
• Strong understanding of customizing, tuning, and validating machine learning training pipelines.

Nice to have

• Experience with diffusion, flow matching, or VAE-based generative techniques.
• Experience with 3D reconstruction, 3DGS, NeRFs, or differentiable rendering.
• Experience developing autonomy for autonomous vehicles or other robotics and computer vision systems.

Culture & Benefits

• Full-time employment within the Software Systems department.
• Work in collaboration with Synthetic Data, HIL Simulation, Simulation V&V, and Perception engineering groups.
• Work focused on improving autonomous vehicle simulation and the Zoox robotics autonomy stack.