Autonomy AI & Graph Systems Intern

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

Benefits offered by this job

Housing/relocation bonus

Medical, dental, vision insurance

Free meals (onsite)

Job summary

Waymo is offering a hybrid onsite internship focused on building an automated semantic extraction pipeline that converts large-scale simulation logs and vehicle trajectories into structured graphs. You will advance temporal graph models to capture multi‑agent interactions and develop a multi‑modal retrieval engine to accelerate scenario discovery and failure analysis.

Join a cross‑functional team to document architecture and roadmap plans.

Qualifications

• Currently enrolled in a graduate program (PhD or Master’s) in Computer Science, Artificial Intelligence, Robotics, Electrical Engineering, or related field.
• At least one academic term remaining.
• Strong software development experience in Python and/or C++ in a Linux environment.
• Hands-on experience with modern deep learning frameworks (PyTorch, JAX, TensorFlow).

Responsibilities

• Design and implement an automated semantic extraction pipeline for large-scale logs and trajectories.
• Develop temporal graph modeling for time-varying multi-agent interactions.
• Build a hybrid retrieval engine for fast scenario discovery and failure analysis.
• Create data exploration tools and dashboards for autonomy engineers and researchers.
• Collaborate cross-functionally to document architecture and roadmap recommendations.

Skills

Python

C++

Linux

Data structures

Education

Graduate student (PhD or Masters)

Tools

PyTorch

TensorFlow

JAX

Job description

Waymo is offering a hybrid onsite internship focused on building an automated semantic extraction pipeline that converts large-scale simulation logs and vehicle trajectories into structured graphs. You will advance temporal graph models to capture multi‑agent interactions and develop a multi‑modal retrieval engine to accelerate scenario discovery and failure analysis.

Join a cross‑functional team to document architecture and roadmap plans.