Zoox · 2 weeks ago
Senior Machine Learning Engineer - Motion Planning & Model Introspection
Zoox is a company that designs, builds, and maintains software for machine learning-based motion planners. They are seeking a Senior Machine Learning Engineer to lead initiatives in introspection techniques and collaborate with various engineering teams to improve behavior in motion planning models.
Artificial Intelligence (AI)Autonomous VehiclesMachine LearningRoboticsTransportation
Responsibilities
Lead new initiatives to introspect the output of machine learning motion planning models. You will both utilize existing cutting edge techniques as well as develop novel introspection techniques
Design new architecture and implement new tools used to analyze machine learning model behavior
Collaborate with engineers on Perception, Prediction, Planning, Machine Learning Model, and Visualization teams to enable development of behavior improvements
Qualification
Required
MS or PhD in Computer Science, Machine Learning, or a related field
5+ years of industry experience in Machine Learning–experience building and maintaining ML training pipelines, including data preprocessing, model training, and evaluation
Proficiency in Python and ML libraries (PyTorch, NumPy, Jax)
Strong understanding of Imitation and Reinforcement Learning concepts and motion planning search techniques
Familiarity with common learned model introspection techniques
Preferred
Familiarity with autonomous vehicles or robotics
Solid understanding of LLM concepts
Publications in top-tier conferences (CVPR, ICCV, RSS, ICRA)
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
Company
Zoox
Zoox is an AI robotics company that provides mobility-as-a-service and self-driving car services. It is a sub-organization of Amazon.
H1B Sponsorship
Zoox has a track record of offering H1B sponsorships. Please note that this does not
guarantee sponsorship for this specific role. Below presents additional info for your
reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (293)
2024 (297)
2023 (209)
2022 (204)
2021 (131)
2020 (83)
Funding
Current Stage
Late StageTotal Funding
$955MKey Investors
Grok Ventures
2020-06-26Acquired
2019-10-21Convertible Note· $200M
2018-07-08Series B· $465M
Leadership Team
Recent News
Channel NewsAsia
2025-12-24
Clean Technica
2025-12-13
Company data provided by crunchbase