Hybrid opportunity at
ZooxSoftware Engineer, ML Platform (ML Training)
Zoox is designing and building an autonomous robotaxi fleet from the ground up to deliver modern urban mobility services. The company operates at the intersection of…
Career Tools
About This Role
Zoox is designing and building an autonomous robotaxi fleet from the ground up to deliver modern urban mobility services. The company operates at the intersection of machine learning, robotics, and design as it deploys vehicles onto public roads. This full-time position follows a hybrid work model. The ML Platform team at Zoox facilitates innovations in computer vision and machine learning to support autonomous driving functions. Joining the ML Training platform…
Job Description
Zoox is designing and building an autonomous robotaxi fleet from the ground up to deliver modern urban mobility services. The company operates at the intersection of machine learning, robotics, and design as it deploys vehicles onto public roads. This full-time position follows a hybrid work model.
The ML Platform team at Zoox facilitates innovations in computer vision and machine learning to support autonomous driving functions. Joining the ML Training platform team means working across multiple engineering groups, including perception, prediction, foundational models, simulation, planning, and data science. The group creates and manages foundational machine learning tools, distributed training infrastructure, deep learning frameworks, and model lifecycle management utilities. Practitioners in this position help accelerate the journey from initial concept to production-ready artificial intelligence.
This opportunity suits engineers experienced in machine learning infrastructure who want to scale deep learning systems for autonomous vehicles. Team members help push the limits of applied machine learning within the organization while expanding robotaxi deployment initiatives.
Responsibilities
- Develop the Zoox Training framework to serve all machine learning teams reliably and efficiently
- Design, build, and run a dependable machine learning platform supporting model training, validation, monitoring, and serving
- Partner with software engineers, machine learning researchers, and data engineers to establish requirements and architectural plans
Requirements
- Minimum of two years of professional machine learning infrastructure experience
- Background working with cloud computing providers such as AWS
- Experience utilizing training tools like JAX, PyTorch, DeepSpeed, or Ray
Qualifications
- Familiarity with creating large-scale, cost-effective distributed model training and compute infrastructure
- Background in designing model lifecycle management utilities and experimentation tools
Core Skills
Frequently Asked Questions
What is the remote work policy for this position?
This position operates on a hybrid remote schedule.
What employment type is this role?
This is a full-time position.
What is the salary for this role?
The compensation details are not specified in the job posting.
Where is the job located?
The specific location of the position is not stated in the source posting.
Sample Interview Questions
AI-generated questions tailored to this specific role — a preview of the full practice set.