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 improve diversity and blur the line between simulation and reality. You will have access to the best sensor data in the world and an incredible infrastructure for testing and validating your algorithms. Our ML group is working on generating rich and diverse 3D environments, humans, and environmental conditions, and you will be able to contribute to any (or all!) of these.

In this role, you will:

• Research, implement, and optimize state of the art machine learning approaches to improve simulation variation and fidelity
• Collaborate with ML Researchers, Technical Artists and Simulation engineers to bring new ML techniques to our production simulator
• 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 programming and algorithm design
• Fluency in C++ and Python
• Experience with modern 3D vision techniques like CNNs, Diffusion, and NeRF
• Publications or production experience in human-centric 3D vision, reconstruction, or style transfer
• Strong mathematical skills and understanding of probabilistic techniques

Bonus Qualifications:

• Bachelor's, Master's or PhD in computer science, mathematics, physics, or related field
• Publications in your field (CVPR, ICCV, RSS, ICRA preferred)
• Experience with generative models for 3D content
• Experience automating 3D content pipelines using applications like Houdini, Maya, or Blender
• Experience in distributed systems and cloud computing platforms
• Experience in 3D rendering for simulation, games, or VFX

Zoox offers a comprehensive package of benefits including paid time off, Zoox Stock Appreciation Rights, Amazon RSUs, health insurance, long-term care insurance, long-term and short-term disability insurance, and life insurance.

Zoox is committed to building a diverse team and encourages applications from candidates with a variety of backgrounds, experiences, and skills.