Taro Logo

Machine Learning Engineer - Generative AI

Zoox is an autonomous vehicle company developing self-driving technology and robotaxis.
Foster City, CA, USABoston, MA, USA
$204,000 - $277,000
Machine Learning
Senior Software Engineer
Hybrid
1,000 - 5,000 Employees
3+ years of experience
AI · Automotive · Robotics

Job Description

Zoox, a leading autonomous vehicle company, is seeking a Machine Learning Engineer specializing in Generative AI to join their Perception team. This role sits at the intersection of cutting-edge AI and autonomous driving technology, where you'll be responsible for developing synthetic data generation systems that enhance the training and evaluation of Zoox's autonomous driving capabilities.

The position offers an exciting opportunity to work with large-scale proprietary data and implement state-of-the-art generative AI techniques, including world models and radiance fields. You'll be directly contributing to the safety and reliability of autonomous systems through your work on sensor-level scenario generation.

As a senior member of the team, you'll lead the development of generative models, design sophisticated training techniques, and collaborate across multiple teams including perception, planning, safety, and simulation. The role requires expertise in Python, deep learning frameworks, and experience with large-scale model training.

The compensation package is competitive, ranging from $204,000 to $277,000 in base salary, plus additional components including Amazon RSUs and Zoox Stock Appreciation Rights. The position offers a comprehensive benefits package and the flexibility of hybrid work arrangements in either Foster City, CA or Boston, MA.

This is an excellent opportunity for someone with a strong academic background (MS/PhD) and relevant industry experience who wants to make a significant impact in the autonomous vehicle industry. The role combines theoretical machine learning expertise with practical applications in a rapidly evolving field, making it ideal for candidates passionate about both AI and autonomous systems.

Last updated a month ago

Responsibilities For Machine Learning Engineer - Generative AI

  • Define and execute the ML roadmap for synthetic data generation using generative AI
  • Lead the development of generative models from small scale objects to complete scenarios
  • Design effective model architectures and sophisticated training techniques
  • Collaborate with perception, planning, safety, simulation, and systems teams
  • Validate and optimize solutions using real-world driving scenarios

Requirements For Machine Learning Engineer - Generative AI

Python
  • MS or PhD in Computer Science, Machine Learning, or related technical field with 3+ years of industry experience
  • Demonstrated experience architecting, training and deploying large models such as diffusion, flow matching, GANs and/or NeRFs
  • Experience building and maintaining ML training pipelines
  • Proficiency in Python and ML libraries (PyTorch, NumPy)
  • Experience training with large scale datasets

Benefits For Machine Learning Engineer - Generative AI

Medical Insurance
Dental Insurance
Vision Insurance
Equity
  • Paid time off (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
  • Life insurance

Related Jobs