Zoox · 5 days ago
Principal Software Engineer, ML Infrastructure
Foster City, CA
Full-time
Onsite
Lead/Staff
$324K/yr - $470K/yrZoox is on a mission to reimagine transportation by building autonomous robotaxis. They are seeking a Principal Software Engineer to lead the development of their ML Infrastructure, enabling innovations in machine learning and computer vision for autonomous driving.
Artificial Intelligence (AI)Autonomous VehiclesMachine LearningRoboticsTransportation
Responsibilities
Develop and execute a strategic vision for ML Infrastructure that will unlock innovation in autonomous driving and enhance our rider experience
Lead the design and implementation of cutting-edge infrastructure spanning all stages of an ML lifecycle from data preparation to training to evaluation, deployment, and serving
Collaborate closely with cross-functional teams, including ML researchers, software engineers, data engineers, and hardware engineers, to define requirements and align on architectural decisions
Enable the engineers in the team to grow their careers by providing technical guidance and mentorship
Qualification
Required
Experience building and managing large-scale ML infrastructure that powers the development of large-scale ML models
Excellent leadership skills with a demonstrated ability to lead high-performing engineering teams
Strong experience with training frameworks like PyTorch, JAX, etc., leveraging GPUs efficiently for distributed model training
Experience with GPU-accelerated inference using TensorRT, Ray Serve, or similar frameworks
Proficient in Python and/or C++
Preferred
Experience enabling the development and deployment of large-scale Foundation models
Experience working on large-scale data infrastructure and big data processing frameworks like Apache Spark
Experience working in the AV domain supporting Perception, Prediction, Planner et al
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
Zoox is an AI robotics company that provides mobility-as-a-service and self-driving car services. It is a sub-organization of Amazon.
Founded in 2014
1001-5000 employees
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 StageTotal Funding
$955MKey Investors
Grok Ventures
2020-06-26Acquired
2019-10-21Convertible Note· $200M
2018-07-08Series B· $465M
Leadership Team
Nelson Pedreiro
Sr. Vice President, Hardware
Z
Zheng Gao
Director of Hardware Engineering
Recent News
Channel NewsAsia
2025-12-24
Clean Technica
2025-12-13
Company data provided by crunchbase