Senior/Staff Machine Learning Engineer, Autonomy Validation

Zoox, a pioneering company in autonomous mobility, is seeking a Senior or Staff Machine Learning Engineer to join their System Design and Mission Assurance (SDMA) team. This role is crucial in developing and enhancing automated testing and validation processes for their robotaxi service.

The position focuses on applying cutting-edge machine learning techniques to improve the efficiency and scalability of testing processes across massive datasets. The engineer will work with both real-world fleet logs and synthetic data, directly impacting the validation of software changes and ensuring the safety and reliability of Zoox's robotaxi service.

Key responsibilities include leading technical initiatives in ML and data science, improving model interpretability, enhancing feature representation for driving scenarios, integrating AV performance data, and optimizing models through data science. The role requires collaboration with various teams including system safety, data science, software, and fleet operations.

The ideal candidate should have 5+ years of experience or a PhD in a relevant field, with strong expertise in machine learning concepts and programming skills in Python. Experience with ML libraries like PyTorch, TensorFlow, and Jax is essential, as is knowledge of large-scale data processing and distributed computing. Background in robotics or autonomous vehicles is highly valued.

Zoox offers a comprehensive compensation package including a base salary range of $210,000 - $289,000, Amazon RSUs, and Zoox Stock Appreciation Rights. Additional benefits include health insurance, paid time off, and various insurance coverages. The company provides a transparent, respectful culture with strong leadership and opportunities for career growth through demonstrated achievement.

This high-impact position offers the chance to work at the forefront of autonomous vehicle technology, combining machine learning expertise with practical applications in vehicle safety and validation. The role presents an excellent opportunity for those passionate about advancing autonomous vehicle technology while ensuring public safety.