Taro Logo

Senior/Staff Machine Learning Engineer, Agent Simulation

Zoox is developing the first ground-up, fully autonomous vehicle fleet and supporting ecosystem for next-generation mobility-as-a-service in urban environments.
Foster City, CA, USA
$200,000 - $300,000
Machine Learning
Staff Software Engineer
Hybrid
5+ years of experience
AI · Automotive · Robotics
This job posting may no longer be active. You may be interested in these related jobs instead:

Description For Senior/Staff Machine Learning Engineer, Agent Simulation

Zoox, a pioneering company in autonomous vehicle technology, is seeking a Senior/Staff Machine Learning Engineer to join their Agent Simulation group. This role presents an exciting opportunity to work at the intersection of artificial intelligence, robotics, and autonomous vehicles.

The position involves developing and enhancing agent behaviors through advanced machine learning techniques, working with extensive real-world driving data, and utilizing state-of-the-art infrastructure for testing and validation. As a key member of the ML group, you'll be responsible for creating diverse and innovative agent behaviors that contribute to the company's mission of developing fully autonomous vehicles.

The role offers a competitive compensation package ranging from $200,000 to $300,000, plus Amazon RSUs and Zoox Stock Appreciation Rights. The company provides comprehensive benefits including various insurance options and flexible time off policies.

Working in a hybrid setting in Foster City, CA, you'll collaborate with safety and autonomy engineers, leveraging your expertise in deep learning, Python, and ML frameworks to build scalable solutions. The ideal candidate should possess strong mathematical skills and experience with production ML pipelines.

This is an excellent opportunity for someone passionate about autonomous vehicles and machine learning to make a significant impact in a fast-moving, execution-oriented team. The role offers the chance to work on groundbreaking technology that will shape the future of urban mobility.

Zoox values diversity and encourages applications from candidates with varied backgrounds and experiences. The company maintains that candidates need not match every listed qualification to be considered, fostering an inclusive environment that drives innovation through diverse perspectives.

Last updated a month ago

Responsibilities For Senior/Staff Machine Learning Engineer, Agent Simulation

  • Research, implement, and optimize state-of-the-art machine learning approaches to improve plausible agent behaviors
  • Find innovative solutions for agent behaviors and simulation analysis
  • Prove machine-learned algorithms have better performance than heuristics
  • Leverage large-scale machine-learning infrastructure to discover new solutions
  • Analyze the difference between behaviors in simulation and real-world
  • Work cross-functionally with safety and autonomy engineers
  • Build scalable, useable cloud pipelines for machine learning solutions for simulation

Requirements For Senior/Staff Machine Learning Engineer, Agent Simulation

Python
  • BS, MS, or PhD degree in Computer Science or a related field
  • Experience with training and deploying deep learning models
  • Experience with production machine learning pipelines: dataset creation, training frameworks, metrics pipelines
  • Fluency in Python and a basic understanding of C++
  • Fluency with Numpy and PyTorch, TensorFlow or JAX
  • Strong mathematics skills

Benefits For Senior/Staff Machine Learning Engineer, Agent Simulation

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
  • Sign-on bonus may be offered

Interested in this job?