Machine Learning Engineer (ML Agents and Planning)

Join Zoox, a company at the forefront of autonomous driving technology. As a Machine Learning Engineer, you will develop cutting-edge machine learning models and pipelines, collaborate with various teams, and contribute to our large-scale machine learning infrastructure. Enjoy a range of benefits, including paid parental leave, health insurance, and a vibrant work environment.

• Develop novel machine learning pipelines and models to predict the behavior of other agents and plan the best course of action for the ego vehicle.
• Collaborate with engineers on Perception, Planning, Simulation, and Validation to solve the overall Autonomous Driving problem.
• Contribute to large-scale machine learning infrastructure to discover new solutions and push the boundaries of the field.
• Fluency in Python with a basic understanding of C++
• Experience with production Machine Learning pipelines: dataset creation, training frameworks, metrics pipelines
• Experience with training and deploying transformer-based model architectures
• Experience in Planning and / or Prediction using Reinforcement Learning techniques
• PhD degree in computer science or related field or master's degree and 5+ years of professional experience in a relevant field
• Top tier publications (NeurIPS, ICML, CVPR)