Waymo
2027 Summer Intern, MS/PhD, Software Engineer, ML Frameworks and Efficiency
Mountain View, CA, USA +1 · Internship · Posted Oct 23, 2026
- Location
- Mountain View, CA, USA
Mountain View, California, United States - Education
- Currently pursuing a PhD or Master’s degree in Computer Science, Electrical Engineering, Machine Learning, Robotics, or a related technical field.
About the job
Improve Waymo’s reinforcement-learning infrastructure and training efficiency to support LPM model training.
What you'll do
- Understand the problem the current LPM model is trying to solve and its details.
- Improve and extend RL infrastructure, profile training slowdowns, and improve training efficiency.
- Identify promising RL technologies and integrate them into the infrastructure.
What they ask for
- Currently pursuing a PhD or Master’s degree in Computer Science, Electrical Engineering, Machine Learning, Robotics, or a related technical field.
- Fundamental understanding of machine learning, deep learning, and reinforcement learning.
- Familiarity with modern frameworks such as JAX, Flax, PyTorch, or TensorFlow.
- Strong software development skills with Python.
Nice to have
- Hands-on ML or RL project experience.
- Experience with large-scale ML system infrastructure or autonomous driving and related technologies.
Pay
$70 — $70 USD (base hourly pay, Masters); $85 — $85 USD (base hourly pay, PhD)
Tools
JAX, Flax, PyTorch, TensorFlow, Python
AI focus
Reinforcement learning infrastructure and training for the LPM model, including training efficiency and integration of RL technologies.
Model-written summary · the original posting is authoritative