Software Engineer
Waymo, PERM - N/A +1
What you will 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 you need
- 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.