The opportunity
Waymo is an autonomous driving technology company with the mission to be the world's most trusted driver. Since its start as the Google Self-Driving Car Project in 2009, Waymo has focused on building the Waymo Driver—The World's Most Experienced Driver™—to improve access to…
What you'll do
Design and build distributed evaluation platforms for large-scale ML evaluation workloads.
Profile evaluation platforms, identify performance bottlenecks (CPU, memory,: I/O, network), and implement optimizations to improve inference speed and resource utilization.
Collaborate with ML engineers to understand evaluation requirements and: scenarios, and improve DevX of the evaluation infrastructure.
Improve runtime goodput of ML inference workload and efficiency of metrics: computations, ensuring scalability and reliability across distributed environments.
Implement and maintain advanced ML infrastructure tools, including ML Pathways, JAX, Flume and TensorFlow.
B.S. in Computer Science, Math, or 3+ years equivalent real-world experience.
What they're looking for
- Proficient in distributed systems design with an understanding of ML efficiency.
- Experience with ML frameworks, including TensorFlow, JAX, XLA.
- Solid programming skills in Python and C++.
- Practical familiarity with profiling tools to uncover performance bottlenecks.