Waymo · 3 months ago
Senior Software Engineer, Training Efficiency
Mountain View, CA
Full-time
Hybrid
Senior Level
$238K/yr - $302K/yr
5+ years expWaymo is an autonomous driving technology company focused on building the Waymo Driver to improve mobility and save lives. They are seeking a Senior Software Engineer to enhance the runtime efficiency of input data pipelines for large-scale ML training workloads, collaborating with researchers and ML engineers to optimize performance.
Artificial Intelligence (AI)AutomotiveAutonomous VehiclesSensorTransportation
Responsibilities
Design, and improve distributed input data pipelines for large-scale ML training workloads
Collaborate with researchers and ML engineers to resolve bottlenecks in data pipeline performance
Improve runtime goodput of ML training workload, including optimizing input data processing systems, ensuring scalability and reliability across distributed environments
Implement and maintain advanced ML infrastructure tools, including ML Pathways, Grain, JAX, and TensorFlow
Evaluate and integrate modern technologies to enhance the performance and scalability of ML systems
Promote best practices for distributed systems architecture and contribute to technical leadership within the team
Qualification
Required
B.S. in Computer Science, Math, or 5+ years equivalent real-world experience
Proficient in distributed systems design with an understanding of ML data pipeline optimization
Experience with ML frameworks, including TensorFlow and JAX
Hands-on experience libraries like Grain or tf.data service
Solid programming skills in Python and C++
Practical familiarity with profiling tools to uncover performance bottlenecks
Preferred
MS in Computer Science, Math
Familiarity with distributed dataflow frameworks like ML Pathways
Benefits
Waymo’s discretionary annual bonus program
Equity incentive plan
Generous Company benefits program
Company
Waymo
Waymo is a mobility technology company that improves transportation by developing self-driving solutions for travelers and daily commuters. It is a sub-organization of Alphabet.
1001-5000 employees
H1B Sponsorship
Waymo has a track record of offering H1B sponsorships. Please note that this does not
guarantee sponsorship for this specific role. Below presents additional info for your
reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (231)
2024 (175)
2023 (268)
2022 (306)
2021 (298)
2020 (317)
Funding
Current Stage
Late StageTotal Funding
$11.1BKey Investors
Alphabet
2024-07-23Series C· $5.6B
2021-06-16Series B· $2.5B
2020-05-12Series A· $750M
Leadership Team
Tekedra Mawakana
Co-Chief Executive Officer
Elisa de Martel
Chief Financial Officer
Recent News
2026-01-12
UK Investor Magazine
2026-01-11
Company data provided by crunchbase