Software Engineer
Core
Design and improve distributed input data pipelines for large-scale machine learning training workloads to optimize performance and scalability.
Role type
Senior IC machine learning infrastructure engineer
Builds
Distributed data pipelines and ML infrastructure tools for Waymo's autonomous driving systems
Domain
Autonomous driving / Machine Learning Infrastructure
Deliverable
production ML models
Required skills
Distributed data processing, ML frameworks (TensorFlow, JAX, ML Pathways), Python, C++, Distributed systems architecture, Technical leadership, Productionizing foundation models, Computational efficiency optimization
Preferred skills
Experience with Apache Beam, Large foundation models (LLMs), Multi-host distributed ML runtimes
Technologies
TensorFlow, JAX, ML Pathways, Grain, Apache Beam
Responsibilities
Design distributed input data pipelines, Collaborate with researchers to resolve bottlenecks, Optimize runtime goodput of training workloads, Implement and maintain ML infrastructure tools, Evaluate and integrate modern technologies, Promote best practices for distributed systems architecture
Seniority
Senior, hands-on IC with technical leadership
