ML Infrastructure Engineer

Waymo is seeking an engineer to design and improve distributed input data pipelines for large-scale machine learning training workloads. The role involves collaborating with researchers and ML engineers to resolve performance bottlenecks, optimizing runtime efficiency, and implementing advanced ML infrastructure tools such as ML Pathways, Grain, JAX, and TensorFlow.