Software Engineer (Learned Trajectory Machine Learning Engineer)

Join Zoox, a company dedicated to revolutionizing transportation through autonomous vehicles. As a Learned Trajectory Machine Learning Engineer, you will develop deep learning models for trajectory generation, collaborate with various teams, and leverage our machine learning infrastructure. Enjoy benefits such as paid parental leave, health insurance, and a vibrant work environment.

Responsibilities

• Develop deep learning models that use imitation learning and reinforcement learning to generate driving plans for autonomous vehicles.
• Collaborate closely with the Planner and Perception teams to advance overall vehicle behavior and ensure tight integration of behavior prediction and motion planning.
• Leverage large-scale machine learning infrastructure to discover new solutions and push the boundaries of the field.

Requirements

• Extensive experience with programming and algorithm design
• Experience with production Machine Learning pipelines: dataset creation, training frameworks, metrics pipelines

Strong mathematics skills

• BS, MS, or PhD degree in computer science or related field
• Experience with training and deploying transformer-based model architectures and reinforcement learning
• Fluency in C++ or Fluency in Python with a basic understanding of C++

Preferred

• Prior experience with Prediction and/or autonomous vehicles or robotics in general
• Conference or Journal publications in Machine Learning or Robotics related venues