Machine Learning Engineer - ML Agents and Planning

Role overview

Responsibilities

• check_circle You will develop new deep learning models that use imitation learning and reinforcement learning to generate driving plans for human-like agents.
• check_circle You will work on novel techniques to estimate the quality of those driving plans along the dimensions of safety, progress, comfort and realism.
• check_circle You will contribute to our large-scale machine learning infrastructure to discover new solutions and push the boundaries of the field
• check_circle You will develop metrics and tools to analyze errors and understand improvements of our systems
• check_circle You will collaborate with engineers on Perception, Planning,Simulation, and Validation to solve the overall Autonomous Driving problem.

Basic qualifications

• PhD degree in computer science or related field or master's degree and 5+ years of professional experience in a relevant field.
• Experience in Planning and / or Prediction using Reinforcement Learning techniques
• Experience with training and deploying transformer-based model architectures
• Experience with production Machine Learning pipelines: dataset creation, training frameworks, metrics pipelines
• Fluency in Python with a basic understanding of C++

Preferred qualifications

• Top tier publications (NeurIPS, ICML, CVPR)

Benefits

• check_circle $189,000 - $270,000 a year
• check_circle Zoox also offers a comprehensive package of benefits, including paid time off (e.g. sick leave, vacation, bereavement), unpaid time off, Zoox Stock Appreciation Rights, Amazon RSUs, health insurance, long-term care insurance, long-term and short-term disability insurance, and life insurance.