FREE ACCESS
5,000–10,000 jobs/day
See all jobs on JobTailor
Search thousands of fresh jobs every day.
Discover
- Fresh listings
- Fast filters
- No subscription required
Create a free account and start exploring right away.
Software Engineer, ML Platform – ML Training
ZooxSoftware Engineer developing the ML platform for autonomous robotaxis at Zoox. Collaborating with multiple teams to enhance ML infrastructure and innovation.
Posted 7/21/2026full-timeFoster City • California • 🇺🇸 United StatesJuniorMid-Level💰 $182,000 - $257,000 per yearWebsite
Core Competencies
Role fitCore Competencies
Use this summary to align your resume positioning with the role.
Demonstrates expertise in building and operating machine learning infrastructure, with a focus on training frameworks and cloud services. Collaborates effectively with cross-functional teams to align on architectural decisions and optimize model lifecycle management.
Highest-signal resume keywords
Machine Learning Infrastructure ExperiencePyTorchAWSDistributed Model TrainingModel Lifecycle Management
ATS Keywords
Tailor your resumeApplicant Tracking System Keywords
Tip: use these terms in your resume and cover letter to boost ATS matches.
Hard Skills
Machine LearningModel Training FrameworksDeepSpeedJAXRay
Soft Skills
CollaborationCross-Functional Teamwork
Industry Keywords
ML PlatformModel ValidationModel ServingModel MonitoringCost-Efficient Infrastructure
Tech Stack
Tools & technologiesAWSCloudPyTorchRay
About the role
Key responsibilities & impact- Build the Zoox Training framework leveraged by all ML teams within Zoox.
- Design, implement, and operate 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.
Requirements
What you’ll need- 2+ years of ML infrastructure experience.
- Experience with training frameworks like PyTorch, DeepSpeed, JAX, Ray, etc.
- Experience working with cloud providers like AWS.
- Experience with building large-scale, cost-efficient distributed model training and ML compute infrastructure (bonus).
- Experience with building model lifecycle management tools and experimentation (bonus)
Benefits
Comp & perks- 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