Senior Software Engineer, ML Platform jobs in United States
cer-icon
Apply on Employer Site
company-logo

Zoox · 4 hours ago

Senior Software Engineer, ML Platform

positionFoster City, CA
timeFull-time
remoteOnsite
seniorityMid, Senior Level
money$189K/yr - $290K/yr
date4+ years exp

Zoox is on a mission to reimagine transportation by building autonomous robotaxis. The Senior Software Engineer in the ML Platform team will focus on developing scalable ML tools and infrastructure to support various ML use cases, enhancing the efficiency of autonomous driving technologies.

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

Responsibilities

Build the Zoox Training framework leveraged by all ML teams within Zoox. This framework needs to be highly scalable, reliable, and efficient
Build the Off-vehicle inference service powering our Foundational models and the models that improve our rider experiences
Lead the design, implementation, and operation of a robust and efficient ML platform to enable the training, validation, serving, and monitoring of ML models
Collaborate closely with cross-functional teams, including ML researchers, software engineers, and data engineers, to define requirements and align on architectural decisions
Enable the junior engineers in the team to grow their careers by providing technical guidance and mentorship

Qualification

ML infrastructure experienceLarge-scale distributed trainingPyTorchGPU-accelerated inferenceCloud providers (AWS)Technical guidanceCollaboration

Required

4+ years of ML infrastructure experience
Experience building large-scale distributed multi-node GPU model training and/or high throughput, low latency serving use cases
Experience with training frameworks like PyTorch and OSS frameworks like Ray
Experience with GPU-accelerated inference using TensorRT, Ray Serve, or a similar framework
Experience working with cloud providers like AWS

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