Machine Learning Engineer - Perception Offline Driving Intelligence jobs in United States
cer-icon
Apply on Employer Site
company-logo

Zoox · 1 day ago

Machine Learning Engineer - Perception Offline Driving Intelligence

Zoox is developing the first ground-up, fully autonomous vehicle fleet and the supporting ecosystem required to bring this technology to market. As an engineer in the Offline Driving Intelligence team, you will develop advanced multimodal large language models that enhance environmental understanding for their autonomous robotaxi platform, ensuring efficient hazard identification and driving restriction interpretation.

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

Responsibilities

Develop multimodal large language models that enhance our robotaxis' understanding of complex urban environments
Implement model architectures and sophisticated training techniques
Build large high quality datasets leveraging all the inputs from our sensor stack and the overall large scale data we have at Zoox
Drive end-to-end ML solutions from research to production, utilizing Zoox's extensive data pipelines and infrastructure to improve autonomous driving capabilities
Collaborate with perception, planning, safety, and systems teams to integrate your models into the vehicle's decision-making pipeline
Validate and optimize your solutions using real-world driving scenarios, directly contributing to the safety and reliability of Zoox's autonomous system

Qualification

Large language modelsMachine learning pipelinesPythonML libraries PyTorchML libraries NumPyAutonomous robotics systemsPublications in conferencesData preprocessingModel evaluation

Required

MS or PhD in Computer Science, Machine Learning, or related technical field
Demonstrated experience training and deploying large language models (LLMs)
Experience building and maintaining ML training pipelines, including data preprocessing, model training, and evaluation
Proficiency in Python and ML libraries (PyTorch, NumPy) demonstrated through professional or research projects
Experience training models with large scale data

Preferred

Publications in top-tier conferences (CVPR, ICCV, RSS, ICRA)
Experience with autonomous robotics systems

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 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 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
Company data provided by crunchbase