Autonomy ML Engineer Intern (MS/PhD) — Hybrid

Waymo LLC; San Francisco, Northern (CA, KY); Hybrid; USD 96,000 - 117,000; Full time

Benefits offered by this job

Housing/relocation bonus

Medical, dental, and vision insurance

Intern events

Free meals

Shuttle access

Onsite gym

Job summary

Waymo LLC in San Francisco offers a hybrid onsite internship for MS/PhD students in Computer Science or ML, focusing on training and evaluating deep learning models for autonomous driving using JAX/Flax, PyTorch, or TensorFlow in a modern ML framework. Interns collaborate across teams and will contribute to evaluation pipelines and data workflows.

Strong C++/Python skills and ML fundamentals are essential. The role includes presenting results, writing production-quality code, and working with

Qualifications

• Currently enrolled in an MS or PhD program in Computer Science, Machine Learning or a related field, returning to the program after the internship.
• Hands-on experience training and evaluating deep learning models in a modern framework (JAX, PyTorch or TensorFlow).
• Strong programming skills in C++/Python and solid ML fundamentals.

Responsibilities

• Train and fine-tune a large multi-task transformer over driving-log sequences on TPUs.
• Design and run offline and end-to-end evaluations and build the datasets and pipelines needed.
• Land production-quality code in a shared codebase and communicate results through design docs and presentations.

Skills

C++/Python programming

Deep learning frameworks

Education

MS or PhD in Computer Science/ML

Tools

JAX/Flax

PyTorch

TensorFlow

TPUs/GPUs

Job description

Waymo LLC in San Francisco offers a hybrid onsite internship for MS/PhD students in Computer Science or ML, focusing on training and evaluating deep learning models for autonomous driving using JAX/Flax, PyTorch, or TensorFlow in a modern ML framework. Interns collaborate across teams and will contribute to evaluation pipelines and data workflows.

Strong C++/Python skills and ML fundamentals are essential. The role includes presenting results, writing production-quality code, and working with