about 12 hours ago
Foster City, CA, USAMid Level / Senior
H1B Sponsor
Base Salary
$182k - $257k/yr
Responsibilities
- Build the Zoox Training framework for all ML teams.
- Design and implement a robust ML platform for model training and validation.
- Collaborate with ML researchers and engineers to define requirements and architectural decisions.
Requirements
- 2+ years of ML infrastructure experience.
- Experience with training frameworks like PyTorch, DeepSpeed, JAX, or Ray.
- Experience working with cloud providers like AWS.