PhD/MS Software Engineer Intern - ML Systems & RL

Waymo LLC; Mountain View (CA); Hybrid; USD 96,000 - 117,000; Full time

Benefits offered by this job

Housing/relocation bonus

Medical, dental, and vision insurance

Free meals on site

Shuttle access

Job summary

Waymo LLC is seeking a research‑oriented software intern in Mountain View to advance RL/ML infrastructure for the LPM model. You will work with SIML on scalable ML frameworks, profiling performance, and implementing efficient training pipelines.

The role targets students pursuing a PhD or Master’s in CS/EE/ML/Robotics, with strong Python skills and experience in ML/DL/RL. Hybrid onsite work and company benefits are provided.

Qualifications

• Currently pursuing a PhD/Master degree in Computer Science, Electrical Engineering, Machine Learning, Robotics, or a related technical field.
• Fundamental understanding of machine learning, deep learning and reinforcement learning
• Familiar with modern frameworks (JAX, Flax, PyTorch, or TensorFlow)
• Strong software development skills with Python

Responsibilities

• Understand the problem the current LPM model is trying to solve along with its details
• Improve/extend the current RL infrastructure to support LPM model training needs
• Profiling and understanding slowness, maintain and improve the training efficiency
• Identify promising new RL technologies and integrate to our infrastructure

Skills

Python programming

Strong math foundation

Academic research mindset

Education

PhD or Masters degree in CS/EE/ML/Robotics

Tools

JAX

Flax

PyTorch

TensorFlow

Job description

Waymo LLC is seeking a research‑oriented software intern in Mountain View to advance RL/ML infrastructure for the LPM model. You will work with SIML on scalable ML frameworks, profiling performance, and implementing efficient training pipelines.

The role targets students pursuing a PhD or Master’s in CS/EE/ML/Robotics, with strong Python skills and experience in ML/DL/RL. Hybrid onsite work and company benefits are provided.