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 improve distributed input data pipelines for large-scale machine learning training workloads.
Collaborate with researchers and machine learning engineers to resolve: bottlenecks in data pipeline performance.
Improve runtime goodput of training workloads by optimizing input data: processing systems for scalability and reliability.
Implement and maintain advanced machine learning infrastructure tools: including ML Pathways, Grain, JAX, and TensorFlow.
Evaluate and integrate modern technologies to enhance the performance and: scalability of machine learning systems.
Promote best practices for distributed systems architecture and contribute to: technical leadership within the team.
What they're looking for
- Bachelor's degree in Computer Science, Mathematics, or a related field, and 5: years of progressive, post-baccalaureate experience in the job offered or in a Software Engineering-related occupation.
- years of experience in Machine learning frameworks and distributed data: processing libraries, including TensorFlow, JAX, ML Pathways, Apache Beam, etc.;
- years of experience in Python or C++ for machine learning libraries and distributed systems;
- years of experience in Technical leadership in architecting ML infrastructure: for large foundation models (LLMs);
