Build Zoox Training framework for all ML teams; scalable, reliable, efficient.
Design, implement, and operate an ML platform for training, validation, serving, and monitoring.
Collaborate with ML researchers, software/data engineers to define requirements.

🎯 Requirements

2+ years of ML infrastructure experience.
Experience with PyTorch, DeepSpeed, JAX, or Ray.
Experience with AWS cloud.

🎁 Benefits

Paid time off (sick leave, vacation, bereavement).
Unpaid time off.
Zoox Stock Appreciation Rights and Amazon RSUs.
Health insurance, long-term care, disability, and life insurance.

Apply on employer's website

This employer gathers applications via their own applicant tracking system.

You will be redirected to an external application form.

Share job

Meet JobCopilot: Your Personal AI Job Hunter

Automatically Apply to Engineering Jobs. Just set your preferences and Job Copilot will do the rest — finding, filtering, and applying while you focus on what matters.

Activate JobCopilot