Zoox

Senior / Staff Machine Learning Engineer - 3D Sensor Simulation

Zoox
Apply
1 day ago
Foster City, CA, USA or Seattle, WA, USASenior / Staff+
H1B sponsor

Base Salary

$242k - $348k/yr

Responsibilities

  • Design, train, and validate ML models that improve the realism of simulated lidar, radar, and camera data.
  • Integrate 3DGS and NeRF-based 3D reconstruction with simulation across sensor modalities.
  • Build and curate training 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 and perception teams.
  • Mentor engineers and interns and establish best practices for training and deploying ML on Zoox sensor data.

Requirements

  • 2+ years of prior experience as a Technical Lead.
  • 5+ years of experience developing ML models in Python with PyTorch or TensorFlow.
  • Industry experience applying ML to lidar, radar, and/or camera sensor data.
  • Strong understanding of customizing, tuning, and validating ML training pipelines.
  • Preferred experience with diffusion, flow matching, VAEs, 3DGS, NeRFs, differentiable rendering, autonomous vehicles, robotics, or computer vision.

Tech Stack

PythonPyTorchTensorFlow

Categories

Zoox

About Zoox

1,001-5,000 employees

Zoox builds a fully autonomous, purpose-built electric vehicle and the end-to-end platform to operate it as an urban robotaxi service. The company designs the vehicle, self-driving software, and fleet operations in-house to offer mobility-as-a-service rather than selling cars. Founded in 2014 and headquartered in Foster City, California, Zoox is a subsidiary of Amazon.

Contact me