Senior Software Engineer, ML Platform
Zoox
7 months ago
Foster City, CA, USA
Senior / Mid Level
H1B Sponsor
Base Salary
$220k - $285k/yr
Responsibilities
- Build the Zoox Training framework for all ML teams.
- Develop the Off-vehicle inference service for foundational models.
- Lead the design and operation of a robust ML platform.
- Collaborate with cross-functional teams to define requirements.
- Provide technical guidance and mentorship to junior engineers.
Requirements
- 4+ years of ML infrastructure experience.
- Experience with large-scale distributed multi-node GPU model training.
- Familiarity with training frameworks like PyTorch and OSS frameworks like Ray.
- Experience with GPU-accelerated inference using TensorRT or similar frameworks.
- Experience working with cloud providers like AWS.
Benefits
- Comprehensive benefits package including health insurance and paid time off.
- Zoox Stock Appreciation Rights and Amazon Restricted Stock Units.
- Long-term and short-term disability insurance and life insurance.
Tech Stack
AWSPyTorch
Categories
AI & MLData Engineering