2026 Intern, PhD, Machine Learning Engineer, Simulation jobs in United States
cer-icon
Apply on Employer Site
company-logo

Waymo · 3 months ago

2026 Intern, PhD, Machine Learning Engineer, Simulation

Waymo is an autonomous driving technology company focused on improving mobility and safety through its advanced driving solutions. They are seeking a PhD intern in Machine Learning Engineering to research and implement state-of-the-art generative models for simulations, collaborating with various teams to enhance the performance of the Waymo Driver.

Artificial Intelligence (AI)AutomotiveAutonomous VehiclesSensorTransportation
check
H1B Sponsor Likelynote

Responsibilities

Research, implement, and evaluate state-of-the-art generative models and advanced sampling techniques for ultra-realistic multi-agent simulations and full fidelity scenario generation
Collaborate with research and engineering teams to integrate models into Waymo's simulation and verification workflows, applying cutting-edge VL and ML efficiency technology
Contribute to projects that evaluate and improve world modeling of high fidelity sensor data for Waymo's E2E AI-first Driver, with opportunities to publish novel research

Qualification

Machine LearningDeep LearningReinforcement LearningPythonPyTorchTensorFlowC/C++High Performance Computing3D VisionAI ArchitectureDistributed Systems

Required

Currently pursuing a Ph.D. in Computer Science, Machine Learning, Electrical Engineering, or a related technical discipline
Strong publication record in top-tier AI/ML or VL conferences and demonstrated expertise in AI Architecture, Deep Learning, Reinforcement Learning, High Performance Computing, Compilers, and/or Computer Architecture
Hands-on experience with deep learning frameworks (e.g., PyTorch, TensorFlow, JAX) and ML coding, proficiency in Python and/or C/C++

Preferred

Experience with large-scale model training, including large diffusion models, world modeling, foundational models, ML modeling and system design and experience in 3D Vision, including 3D or 4D Gaussian Splatting, 3D Assets Generation, etc
Proficiency in ML frameworks like Jax, XLA, and/or TPU, and contributions to relevant open-source projects
Demonstrated expertise in research domains such as Deep Learning, Reinforcement Learning, Distributed systems, AI Architecture, and familiarity with 3DGS/NeRF technologies

Benefits

Company’s generous benefits programs

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
$11.1B
Key Investors
Alphabet
2024-07-23Series C· $5.6B
2021-06-16Series B· $2.5B
2020-05-12Series A· $750M

Leadership Team

leader-logo
Tekedra Mawakana
Co-Chief Executive Officer
linkedin
leader-logo
Elisa de Martel
Chief Financial Officer
linkedin
Company data provided by crunchbase