Machine Learning Engineer Intern, Autonomy Behavior jobs in United States
cer-icon
Apply on Employer Site
company-logo

Zoox · 2 hours ago

Machine Learning Engineer Intern, Autonomy Behavior

positionFoster City, CA
timeInternship
remoteOnsite
seniorityIntern
money$5,500/mo - $9,500/mo
workcardStart in 2026 Spring

Zoox is developing the first ground-up, fully autonomous vehicle fleet and the supporting ecosystem required to bring this technology to market. As a Machine Learning Engineer Intern in the Autonomy Behavior group, you will work on enhancing planning and control systems for autonomous vehicles, contributing to various teams focused on improving ML models and performance.

Artificial Intelligence (AI)Autonomous VehiclesMachine LearningRoboticsTransportation
check
H1B Sponsor Likelynote

Responsibilities

Explore ways to improve introspection for the ML planner as well as improve model performance
Develop a hybrid planning system that combines a learned planner component with a search based supervisor system
Explore novel approaches to behavior modeling, including research at the intersection of LLM models and motion planning systems
Drive performance improvements through rigorous experimentation
Develop innovative methods for analyzing and explaining the ML Planner's decisions

Qualification

PythonC++PyTorchJAXML pipelinesNeural Network designMachine learningPublication experienceML visualization

Required

Currently working towards a B.S., M.S., Ph.D., or advanced degree in a relevant engineering program
Must be returning to school to continue your education upon completing this internship
Good academic standing
Able to commit to a 12-week internship beginning in May or June of 2026
At least one previous industry internship, co-op, or project completed in a relevant area
Ability to relocate to the Bay Area, California for the duration of the internship
Interns at Zoox may not use any proprietary information they are working on as part of their thesis, any published work with their university, or to be distributed to anyone outside of Zoox
Advanced understanding of Python or C++
Strong experience with PyTorch and/or JAX
Experience with production ML pipelines: dataset creation, labeling, training, metrics
Experience with Neural Network design and implementation
Strong track record in machine learning for autonomous driving, robotics, or LLMs

Preferred

Publication at top-tier conferences (e.g., NeurIPS, ICML, CoRL, CVPR, ICCV, ECCV, IROS, ...)
Familiarity with applying ML visualization and introspection techniques

Benefits

Medical insurance
A housing stipend
Relocation assistance will be offered based on eligibility

Company

Zoox

twittertwittertwitter
Glassdoor3.8
company-logo
Zoox is an AI robotics company that provides mobility-as-a-service and self-driving car services. It is a sub-organization of Amazon.
dateFounded in 2014
location
Foster City, California, USA
people-size1001-5000 employees
web-link
https://www.zoox.com/

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 Stage
Total Funding
$955M
Key Investors
Grok Ventures
2020-06-26Acquired
2019-10-21Convertible Note· $200M
2018-07-08Series B· $465M

Leadership Team

leader-logo
Nelson Pedreiro
Sr. Vice President, Hardware
linkedin
Z
Zheng Gao
Director of Hardware Engineering
linkedin

Recent News

Channel NewsAsia
Amazon's Zoox to recall 332 US vehicles over software error
2025-12-24
AfroTech
Two Black Woman Leaders Are At The Forefront Of The Robotaxi Revolution As CEOs Of Major Vehicle Companies Waymo And Zoox
2025-12-17
Clean Technica
Zoox Becomes Robotaxi Partner for T-Mobile Arena
2025-12-13
Company data provided by crunchbase