Senior/Staff Machine Learning Engineer, 3D Simulation

The 3D Simulation group at Zoox is looking for machine learning engineers to bring the latest research in 3D vision to accelerate autonomy by blurring the line between simulation and reality. Our ML group is working on generating rich and diverse 3D environments, humans, and environmental conditions using reconstruction and generative models. You will have access to the best sensor data in the world and an incredible infrastructure for testing and validation to bring new algorithms into our robotics stack.

In this role, you will:

• Research, implement, and optimize state of the art machine learning approaches to improve 3D simulation variation and fidelity
• Collaborate with ML Researchers, Technical Artists and Simulation engineers to bring new ML techniques to our production simulator
• Train 4D generative and world models for simulating rare scenarios
• Build scalable, useable cloud pipelines for machine learning solutions in 3D simulation
• Improve rendering and tooling for generating realistic data at scale

Qualifications:

7+ years of experience with Python/C++ programming and algorithm design
Experience building and extending radiance field ML models
Experience with generative models for 3D or video content
Publications or production experience in human-centric 3D vision, 3D reconstruction, or style transfer
Strong mathematical skills and understanding of probabilistic techniques

Bonus Qualifications:

• Experience automating 3D content pipelines using applications like Houdini, Maya, or Blender
• Experience in distributed systems and cloud computing platforms
• Experience in generative camera, lidar, radar, or LWIR