Lead technical development of machine learning algorithms for the onboard Planner, advancing imitation learning, reinforcement learning, and model scaling; integrate foundation models; create metrics, data, and training frameworks; mentor ML developers; and collaborate across perception, simulation, system design, and safety teams to solve autonomous driving challenges in complex urban environments.
The Autonomy Behaviors team is responsible for producing our vehicle’s driving trajectory. As a Principal ML Engineer, you will lead the technical development of machine learning algorithms for our next-generation ML Planner. Your impact will range from delivering innovations for immediate milestones to pioneering exploratory projects. You will collaborate closely with teams specializing in Foundation Models, Perception, Simulation, and Safety Validation, in influencing our autonomy stack. Your role will look at problems in a way that crosses team boundaries to prototype new approaches that influence the long term technical direction of multiple organizations within the company.
In this role, you will:
- Set the technical direction of ML development for the future of our Planner
- Push the frontiers of imitation learning, reinforcement learning, and model scaling to develop new ML models for our onboard Planner
- Identify and drive force-multipliers for ML development, including metrics, data, and training frameworks
- Develop new methods to leverage Zoox’s Foundation Models into the ML Planner
- Provide technical mentorship to the broader group of ML developers at Zoox
- Collaborate with engineers across Autonomy Software, Simulation, System Design and Mission Assurance, and other departments to solve the overall Autonomous Driving problem in complex urban environments
Qualifications
- BS, MS, or PhD degree in computer science or related field
- Proven track record of technical leadership in large-scale AI organizations
- Significant experience with training and deploying Deep Learning models to production
- Fluency in C++ or Fluency in Python with a basic understanding of C++
- Extensive experience with programming and algorithm design-Strong mathematics skills, 10+ years of experience
Bonus Qualifications
- Conference or Journal publications in Machine Learning or Robotics related venues
- Prior experience with Prediction and Planning for autonomous vehicles
About Zoox
Zoox is developing the first ground-up, fully autonomous vehicle fleet and the supporting ecosystem required to bring this technology to market. Sitting at the intersection of robotics, machine learning, and design, Zoox aims to provide the next generation of mobility-as-a-service in urban environments. We’re looking for top talent that shares our passion and wants to be part of a fast-moving and highly execution-oriented team.
Follow us on LinkedIn
Accommodations
If you need an accommodation to participate in the application or interview process please reach out to [email protected] or your assigned recruiter.
A Final Note:
You do not need to match every listed expectation to apply for this position. Here at Zoox, we know that diverse perspectives foster the innovation we need to be successful, and we are committed to building a team that encompasses a variety of backgrounds, experiences, and skills.
Zoox Foster City, California, USA Office
4000 E 3rd Ave, Foster City, CA, United States, 94404
Zoox Foster City, California, USA Office
1149 Chess Drive, Foster City, CA, United States, 94404
Zoox Fremont, California, USA Office
47540 Kato Road, Fremont, CA, United States, 94538
Zoox San Francisco, California, USA Office
60 Broadway St, San Francisco, CA, United States, 94111
Similar Jobs
Aerospace • Artificial Intelligence • Machine Learning • Robotics • Software
Designs and automates security controls across AWS infrastructure, containers, CI/CD systems, and platform services. Builds hardened container images, integrates vulnerability and configuration scanning, manages secrets and identity controls, automates compliance checks, and supports audit evidence. Diagnoses security weaknesses and drives remediation with engineering teams. At the Staff level, establishes reusable security patterns and promotes adoption across multiple teams while balancing security, reliability, and developer productivity.
Top Skills:
AWSBashCi/CdContainerizationGoInfrastructure As CodePython
Aerospace • Artificial Intelligence • Machine Learning • Robotics • Software
Lead the establishment and maturation of the SRE function across cloud infrastructure and platform services. Define reliability targets, build observability, lead incident response and root-cause analysis, improve resilience through automation and testing, and develop operational tooling. Partner with engineering teams on reliability requirements, establish incident practices, mentor engineers, and manage the short- and long-term SRE roadmap.
Top Skills:
AWSGoKubernetesPython
Aerospace • Artificial Intelligence • Machine Learning • Robotics • Software
Lead the establishment and maturation of the SRE function across cloud infrastructure and platform services. Define reliability targets, build observability and automation, lead incident response and root-cause analysis, improve resilience and recovery, and reduce operational toil. Partner with product and platform teams on reliability-focused design, establish incident practices, manage the SRE roadmap, and mentor engineers across Cloud Engineering and Reliability.
Top Skills:
AWSCloud-Native ObservabilityGoInfrastructure As CodeKubernetesPython
What you need to know about the San Francisco Tech Scene
San Francisco and the surrounding Bay Area attracts more startup funding than any other region in the world. Home to Stanford University and UC Berkeley, leading VC firms and several of the world’s most valuable companies, the Bay Area is the place to go for anyone looking to make it big in the tech industry. That said, San Francisco has a lot to offer beyond technology thanks to a thriving art and music scene, excellent food and a short drive to several of the country’s most beautiful recreational areas.
Key Facts About San Francisco Tech
- Number of Tech Workers: 365,500; 13.9% of overall workforce (2024 CompTIA survey)
- Major Tech Employers: Google, Apple, Salesforce, Meta
- Key Industries: Artificial intelligence, cloud computing, fintech, consumer technology, software
- Funding Landscape: $50.5 billion in venture capital funding in 2024 (Pitchbook)
- Notable Investors: Sequoia Capital, Andreessen Horowitz, Bessemer Venture Partners, Greylock Partners, Khosla Ventures, Kleiner Perkins
- Research Centers and Universities: Stanford University; University of California, Berkeley; University of San Francisco; Santa Clara University; Ames Research Center; Center for AI Safety; California Institute for Regenerative Medicine
