Skip to content
Staff Machine Learning Engineer – Runtime & Optimization
| Company | Waymo |
|---|
| Location | Mountain View, CA, USA |
|---|
| Salary | $238000 – $302000 |
|---|
| Type | Full-Time |
|---|
| Degrees | Master’s |
|---|
| Experience Level | Senior, Expert or higher |
|---|
Requirements
- M.S. in CS, EE, Deep Learning or a related field
- 2+ years of experience as a technical lead, including writing project plans, engaging with customer teams, mentoring, responsible for goals & execution, reporting status
- 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
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.
Preferred Qualifications
- PhD in CS, EE, Deep Learning or a related field
- 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
