Skip to content
Senior Machine Learning Engineer – Runtime & Optimization
| Company | Waymo |
|---|
| Location | Mountain View, CA, USA |
|---|
| Salary | $204000 – $259000 |
|---|
| Type | Full-Time |
|---|
| Degrees | Master’s |
|---|
| Experience Level | Senior |
|---|
Requirements
- M.S. in CS, EE, Deep Learning or a related field
- 3+ years of experience on model optimization or efficient deep learning techniques
- Strong Python or C++ programming skills
- Solid experience with designing, training and debugging deep learning models to achieve the highest scores/accuracies.
Responsibilities
- Work across the entire ML framework/compiler stack (e.g. JAX, XLA, Triton, and CUDA), and system-efficient deep learning models.
- Deep dive into the whole stack of ML software stack, from custom ops, framework/ ML compiler, to low-level libraries.
- Apply model optimization and efficient deep learning techniques to models and optimized ML operator libraries.
Preferred Qualifications
- PhD in CS, EE, Deep Learning or a related field.
- Deep knowledge on system performance, GPU optimization or ML compiler.
- 5+ years of experience on model optimization or efficient deep learning techniques
