Skip to content

Software Engineer – Inference – Machine Learning Infrastructure
Company | Waymo |
---|
Location | Mountain View, CA, USA |
---|
Salary | $170000 – $216000 |
---|
Type | Full-Time |
---|
Degrees | Bachelor’s, Master’s |
---|
Experience Level | Mid Level, Senior |
---|
Requirements
- B.S. in Computer Science, Math, or equivalent real-world experience
- 3+ years of experience with machine learning fundamentals and experience with popular ML frameworks such as JAX, PyTorch, or TensorFlow
- Experienced in software development projects using C++ or Python and deploying on cloud platforms, for example AWS, Azure, or GCP
- Experience with distributed systems and multi-threaded and stream-based programming models
Responsibilities
- Report into our Head of ML Inference
- Focus on developing large-scale and efficient inference solutions for Waymo
Preferred Qualifications
- MS in Computer Science, Math or related field
- Practical familiarity using ML accelerator profiling tools to uncover performance bottlenecks
- 3+ years of experience building scalable inference platforms for real-time and batch processing
- Familiarity with ML distillation concepts and experience with distillation frameworks