Posted in

Software Engineer – Inference – Machine Learning Infrastructure

Software Engineer – Inference – Machine Learning Infrastructure

CompanyWaymo
LocationMountain View, CA, USA
Salary$170000 – $216000
TypeFull-Time
DegreesBachelor’s, Master’s
Experience LevelMid 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