Senior / Staff Machine Learning Engineer (ML Agents & Planning)

Join Zoox, a cutting-edge autonomous vehicle company, as a Senior/Staff Machine Learning Engineer. In this role, you will develop foundation models for ML agents and planning, collaborate with various teams to validate driving performance, and work on novel machine learning pipelines and models. You will also contribute to the machine learning infrastructure, develop metrics and tools for analysis, and collaborate with engineers on perception, planning, simulation, and validation. Enjoy a range of benefits including paid parental leave, health insurance, and a supportive work environment.

• Develop and implement 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 cross-functional teams to develop and validate driving performance, and contribute to large-scale machine learning infrastructure.
• Develop metrics and tools to analyze errors and understand improvements of the systems, and work on novel techniques to estimate the quality of driving plans.
• Experience with training and deploying transformer-based model architectures
• Experience in Planning and / or Prediction using Reinforcement Learning techniques
• Experience with production Machine Learning pipelines: dataset creation, training frameworks, metrics pipelines
• PhD degree in computer science or related field +1y of professional experience (top tier publications can remove the need for the year of experience) or, MSc +5y of professional experience in a relevant field
• Fluency in Python with a basic understanding of C++
• Top tier publications (NeurIPS, ICML, CVPR)