Senior Machine Learning Engineer (AI)

TL;DR

Senior Machine Learning Engineer (AI): Developing and optimizing 3D rendering of sensor data, combining GenAI/ML and modern 3D graphics techniques. Focus on improving sensor fidelity and generating realistic data at scale.

Location: Seattle, WA

Salary: $227,000 - $316,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 to show measurable impact of 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.
• Zoox Stock Appreciation Rights, Amazon RSUs, health insurance, long-term care insurance, long-term and short-term disability insurance, and life insurance.