Waymo · 4 months ago
Staff Software Engineer, Training Efficiency
Mountain View, California
Full-time
Onsite
Lead/Staff
$238K/yr - $302K/yr
8+ years expWaymo is an autonomous driving technology company with the mission to be the world's most trusted driver. The Staff Software Engineer in Training Efficiency will design and improve distributed input data pipelines for large-scale ML training workloads, collaborating with researchers and ML engineers to enhance performance and scalability of ML systems.
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 8+ 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
$27.1BKey Investors
Alphabet,Dragoneer Investment Group,DST Global,Sequoia CapitalAlphabetFidelity,Perry Creek Capital,T. Rowe Price
2026-02-02Series D· $16B
2024-07-23Series C· $5.6B
2021-06-16Series B· $2.5B
Leadership Team
Tekedra Mawakana
Co-Chief Executive Officer
Annabel Chang
Director of U.S. State and Local Public Policy and Political Campaigns
Recent News
2026-02-08
2026-02-08
2026-02-08
Company data provided by crunchbase