Machine Learning Engineer developing Foundation Models for ML Agents at Zoox. Collaborating with teams to validate driving performance and generate human-like driving behavior.
Responsibilities
You will develop new deep learning models that use imitation learning and reinforcement learning to generate driving plans for human-like agents.
You will work on novel techniques to estimate the quality of those driving plans along the dimensions of safety, progress, comfort and realism.
You will contribute to our large-scale machine learning infrastructure to discover new solutions and push the boundaries of the field
You will develop metrics and tools to analyze errors and understand improvements of our systems
You will collaborate with engineers on Perception, Planning, Simulation, and Validation to solve the overall Autonomous Driving problem.
Requirements
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.
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++
Benefits
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
life insurance
Job title
Senior Machine Learning Engineer – ML Agents, Planning
Master Thesis focusing on developing machine learning models for lithium - ion cell sorting at Fraunhofer LBF. Involvement in innovative projects addressing circular economy in battery recycling.
Machine Learning Engineer designing and implementing AI systems focused on Japanese language challenges at Woven by Toyota. Involves technical R&D, system design, and collaboration with cross - functional teams.
Principal Software Engineer leading MLOps within Analytics Platform at Sun Life. Focused on AWS and machine learning operations, collaborating across technical and business teams.
Machine Learning Engineer designing and optimizing deep learning models for safety - critical environments at Destinus. Shaping the future of high - speed, autonomous flight technologies.
Machine Learning Engineer optimizing personalization systems for Spotify's audio streaming service. Collaborating with cross - functional teams to enhance user experience and deliver recommendations.
Principal Machine Learning Engineer developing ML and GenAI solutions in a cloud - native environment at Flexera. Leading a high - impact team and driving operational excellence for ML infrastructure.
Senior ML Platform/Ops Engineer building ML systems for AI - powered learning at Preply. Productionizing machine learning with high reliability, performance, and observability in a hybrid environment.
Senior ML Platform/Ops Engineer building AI - powered ML pipelines for a dynamic Ed - Tech company. Collaborating with ML scientists and engineers to ensure reliable deployment and observability.
Machine Learning Engineer developing advanced Deep Learning models for autonomous driving technology at Mobileye. Collaborating in a high - end algorithmic engineering team on critical computer vision challenges.