Senior Machine Learning Engineer (AI)

TL;DR

Senior Machine Learning Engineer (AI): Researching and implementing 3D rendering of sensor data, leveraging GenAI/ML and 3D graphics to improve the fidelity of sensor simulation. Focus on developing realism metrics and improving rendering and ML inference tooling for generating realistic data at scale.

Location: Foster City, CA

Salary: $239,000 - $333,000 a year

Company

Zoox is developing the first ground-up, fully autonomous vehicle fleet and the supporting ecosystem required to bring this technology to market.

What you will do

• Research, implement, and optimize state of the art 3D rendering of sensor data, leveraging GenAI/ML and 3D graphics.
• Develop realism metrics with V&V to show measurable impact of your improved sensor fidelity.
• Collaborate with Perception and Safety teams to improve realism of sensor simulation for high fidelity synthetic data.
• Improve rendering and ML inference tooling for generating realistic data at scale.

Requirements

• 5+ years of industry experience, and/or PhD, developing neural rendering techniques like Gaussian Splatting, NeRFs, or 3D reconstruction
• 5+ years of experience developing software with Python and/or modern C++
• Expertise with machine learning frameworks PyTorch or Tensorflow
• Familiarity with 3D graphics algorithms, such as 3D geometry and camera models
• Strong mathematical skills and understanding of 3D linear algebra and probabilistic techniques

Nice to have

• Master's or PhD in computer science, mathematics, physics, or related field
• Experience with generative models for 3D content pipelines using applications like Houdini, Maya, or Blender.
• Experience in 3D rendering for simulation, games, cloud computing, or VFX

Culture & Benefits

• Comprehensive package of benefits, including paid time off (e.g. sick leave, vacation, bereavement), unpaid time off, Zoox Stock Appreciation Rights, Amazon RSUs, health insurance, long-term care insurance, long-term and short-term disability insurance, and life insurance.