Waymo is seeking to improve the performance of its machine learning workloads on both cloud and car.
Requirements
- M.S. in CS, EE, Deep Learning or a related field
- 5+ years of experience developing solutions in ML systems or ML software stack (Pytorch/JAX/TF, runtime libraries, ML compiler)
- Deep understanding of ML system architecture, performance analysis and tools
- Strong Python or C++ programming skills
- Familiarity with the HW architecture of ML hardware accelerators (e.g., GPU/TPU)
- Deep knowledge of model optimization or efficient deep learning techniques for foundation models or LLM
- Experience with GPU HW or TPU HW and related system software
Responsibilities
- Lead the collaboration with the world-class Waymo ML scientists in perception, planner, research and simulation. Identify opportunities in both systems and models to make ML workloads faster.
- Lead projects from proposals through execution by developing junior engineers.
- Analyze and improve ML system workloads on both cloud and self-driving cars.
- Apply model optimization, efficient deep learning techniques and ML software improvements to Waymo's ML systems.
Other
- 2+ years of experience as a technical lead, including writing project plans, engaging with customer teams, mentoring, responsible for goals & execution, reporting status
- M.S. in CS, EE, Deep Learning or a related field
- PhD in CS, EE, Deep Learning or a related field (preferred)
