Senior Staff Software Engineer, Model Post Training jobs in United States
cer-icon
Apply on Employer Site
company-logo

Waymo · 11 hours ago

Senior Staff Software Engineer, Model Post Training

Waymo is an autonomous driving technology company with the mission to be the world's most trusted driver. They are seeking highly motivated and talented experienced engineers to lead the post-training LLM for autonomous driving, working alongside a world-class team to develop and advance AI models.

Artificial Intelligence (AI)TransportationAutomotiveAutonomous VehiclesSensor
check
H1B Sponsor Likelynote

Responsibilities

Develop and apply methods to enhance the instruction following and CoT abilities of LLM models targeting autonomous driving applications
Collaborate deeply with research teams asking the right technical questions, translating research advances into production-ready systems, and shaping joint technical direction
Provide technical leadership to influence senior engineers and researchers across ML, infra, and data teams
Raise the technical bar for how Waymo trains, evaluates, and deploys LLM models in the autonomous driving technical stack

Qualification

Post-training methods for LLMDeep learning frameworksLarge-scale model trainingReinforcement learning techniquesProduction-ready systemsSafety-critical domainsCommunicationCollaboration skills

Required

Demonstrated experience developing post-training methods for LLM models
Extensive experience with deep learning frameworks (e.g. PyTorch, JAX) and large-scale model training
A track record of operating effectively under ambiguity, setting direction amid rapidly evolving research and technical constraints
The ability to translate research insights into production-ready post-training pipelines and systems
Experience applying large language models or foundation models in complex, safety-critical domains (e.g., autonomy, robotics, or other high-reliability systems)
Excellent communication and collaboration skills

Benefits

Waymo’s discretionary annual bonus program
Equity incentive plan
Generous Company benefits program

Company

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.

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 Stage
Total Funding
$27.1B
Key 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

leader-logo
Tekedra Mawakana
Co-Chief Executive Officer
linkedin
leader-logo
Annabel Chang
Director of U.S. State and Local Public Policy and Political Campaigns
linkedin
Company data provided by crunchbase