Machine Learning Engineer - ML Agents and Planning

Experteer Overview In this role you will develop foundation models for ML Agents and planning systems to enable generalization in autonomous driving. You will collaborate with Planner, Simulation and Validation teams to build and validate driving performance. You’ll create imitation learning and reinforcement learning pipelines to shape ego vehicle decisions, and contribute to large-scale ML infrastructure. You solve safety, progress, comfort, and realism metrics, driving impactful advances in autonomous driving. Compensation / Benefits * Develop deep learning models using imitation learning and reinforcement learning to generate driving plans for human-like agents. * Design and apply methods to estimate plan quality across safety, progress, comfort, and realism. * Contribute to large-scale ML infrastructure to explore new solutions and push methodological boundaries. * Create metrics and tools to analyze errors and understand improvements in the system. * Collaborate with Perception, Planning, Simulation, and Validation engineers to address the autonomous driving problem. Tasks * PhD in computer science or related field, or master’s degree with 5+ years of relevant experience * Experience in Planning and/or Prediction using Reinforcement Learning techniques * Experience with training and deploying transformer-based model architectures * Experience with production ML pipelines: dataset creation, training frameworks, metrics pipelines * Fluency in Python with a basic understanding of C++ Key requirements * paid time off * health insurance * long-term disability insurance * short-term disability insurance * life insurance * Stock Appreciation Rights (Zoox and Amazon RSUs)

Planning, Simulation, and Validation engineers to address the autonomous driving problem. Tasks * PhD in computer science or related field, or master’s degree with 5+ years of relevant experience * Experience in Planning and/or Prediction using Reinforcement Learning techniques * Experience with training and deploying transformer-based model architectures * Experience with production ML pipelines: dataset creation, training frameworks, metrics pipelines * Fluency in Python with a basic understanding of C++ Key requirements * paid time off * health insurance * long-term disability insurance * short-term disability insurance * life insurance * Stock Appreciation Rights (Zoox and Amazon RSUs)