Zoox

Software Engineer, ML Platform (ML Training)

Zoox
Apply
2 months ago
Foster City, CA, USAMid Level
H1B Sponsor

Base Salary

$182k - $257k/yr

Responsibilities

  • Build and maintain Zoox’s scalable, reliable, and efficient ML training framework.
  • Design, implement, and operate infrastructure for ML model training, validation, serving, and monitoring.
  • Collaborate with ML researchers, software engineers, data engineers, and hardware engineering teams on requirements and architectural decisions.

Requirements

  • At least 2 years of ML infrastructure experience.
  • Experience with ML training frameworks such as PyTorch, DeepSpeed, JAX, or Ray.
  • Experience working with cloud providers such as AWS.
Zoox

About Zoox

1,001-5,000 employees

Zoox is transforming mobility-as-a-service by developing a fully autonomous, purpose-built fleet designed for AI to drive and humans to enjoy.

Contact me