Machine Learning Engineer, Agent Simulation

at  Zoox

The Agent Simulation group at Zoox is in search of machine learning engineers to play a crucial role in enhancing agent behaviors, assessing performance, and developing tools to prioritize simulation results. You will be instrumental in the development of reliable and validated simulations, and have the opportunity to work with a wealth of real-world driving data and an exceptional infrastructure for testing and validating your algorithms. Our ML group is focused on creating diverse and innovative agent behaviors, and your contributions will be key in this endeavor.

QUALIFICATIONS

• 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

BONUS QUALIFICATIONS

• Experience with LLMs, reinforcement learning, imitation learning
• Conference or journal publications in machine learning or robotics
• Prior experience with agent behaviors, prediction, autonomous vehicles or robotics

ABOUT ZOOX

Zoox is developing the first ground-up, fully autonomous vehicle fleet and the supporting ecosystem required to bring this technology to market. Sitting at the intersection of robotics, machine learning, and design, Zoox aims to provide the next generation of mobility-as-a-service in urban environments. We’re looking for top talent that shares our passion and wants to be part of a fast-moving and highly execution-oriented team.
Follow us on LinkedIn

• 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 our large-scale machine-learning infrastructure to discover new solutions
• Analyze the difference between behaviors in simulation and real-world
• Work cross-functionally with our safety and autonomy engineers
• Build scalable, useable cloud pipelines for machine learning solutions for simulation