Software Engineer - Learned Trajectory Machine Learning Engineer jobs in United States
cer-icon
Apply on Employer Site
company-logo

Zoox · 11 hours ago

Software Engineer - Learned Trajectory Machine Learning Engineer

positionFoster City, CA
timeFull-time
remoteOnsite
seniorityMid, Senior Level
money$189K/yr - $270K/yr

Zoox is developing the first ground-up, fully autonomous vehicle fleet and the supporting ecosystem required to bring this technology to market. As a Learned Trajectory Machine Learning Engineer, you will be responsible for developing deep learned models that produce trajectories for autonomous vehicles, collaborating closely with various teams to enhance vehicle behavior and performance in complex urban environments.

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

Responsibilities

You will develop new deep learning models that use imitation learning and reinforcement learning to generate driving plans for our autonomous vehicle
You will also work on techniques to estimate the quality of those driving plans along the dimensions of safety, progress, comfort etc
You will leverage 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, and Simulation to solve the overall Autonomous Driving problem in complex urban environments

Qualification

Deep learning modelsReinforcement learningTransformer-based modelsMachine Learning pipelinesC++PythonProgrammingAlgorithm designMathematics skillsConference publicationsRobotics experience

Required

BS, MS, or PhD degree in computer science or related field
Experience with training and deploying transformer-based model architectures and reinforcement learning
Experience with production Machine Learning pipelines: dataset creation, training frameworks, metrics pipelines
Fluency in C++ or Fluency in Python with a basic understanding of C++
Extensive experience with programming and algorithm design
Strong mathematics skills

Preferred

Conference or Journal publications in Machine Learning or Robotics related venues
Prior experience with Prediction and/or autonomous vehicles or robotics in general

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

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