Waymo is an autonomous driving technology company that seeks an experienced Senior Machine Learning Infrastructure Engineer to lead the development of advanced AI/ML infrastructure for multi-billion parameter foundation models in ML accelerator-friendly simulations.
Requirements
- BS in Computer Science, Robotics, similar technical field of study, or equivalent practical experience
- 5+ years of professional software engineering experience, with at least 3 years in machine learning infrastructure
- Solid experience in the development and optimization of machine learning infrastructure tools like DeepSpeed, PyTorch, TensorFlow, or similar frameworks
- Strong expertise in distributed training techniques, including gradient sharding and optimization strategies
- Deep understanding of state-of-the-art machine learning models such as auto-regressive transformers
Benefits
- Discretionary annual bonus program
- Equity incentive plan
- Generous Company benefits program