MS/PhD Machine Learning Driver Refinement Foundations (2027 Summer Internship)

Waymo LLC; San Francisco (CA); Hybrid; USD 34,000 - 48,000; Full time

Benefits offered by this job

Medical, dental, and vision insurance

Bonus and equity opportunities

401(k) retirement plans

Job summary

Waymo LLC is offering an internship to contribute to data-driven driving behavior improvements. You will mine and curate driving data, and work with large language models and foundation models to support reasoning, decision making, and multi-modality generation.

The role emphasizes building robust data generation pipelines, designing experiments for model training, and advancing data-centric AI techniques within the Planner organization. Hybrid work options are available.

Qualifications

• You are currently enrolled in a graduate program (Master’s or PhD) in Computer Science or a related technical field.
• You have strong problem solving and programming skills in Python and/or C++.
• You have experience with model evaluation and understanding metrics.
• Some understanding of SQL is preferred, not required.
• Previous Internship experience is a plus.

Responsibilities

• Mine and curate dataset samples from large corpus of driving data.
• Leverage large language models and world foundation models for reasoning, decision making and multi-modality generation.
• Design and implement robust data generation pipelines for data generation.
• Design and evaluate experiments for model training.
• Designing robust data generation pipelines for model training.

Skills

Python

C++

Problem solving

SQL basics

Education

Graduate student (MS/PhD)

PhD preferred

Tools

Jax

PyTorch

Job description

• The team is accountable for providing the common, consistent framework (and enabling expertise) for post-training and heuristic hill climbing across the entire Planner organization. We build fundamental platforms to assist Waymo engineers in delivering driving behavior improvements using data-driven and heuristic methods
• Waymo interns partner with leaders in the industry on projects that create impact to the company. We believe learning is a two-way street: applying your knowledge while providing you with opportunities to expand your skill-set. Interns are an important part of our culture and our recruiting pipeline. Join us at Waymo for a fun and rewarding internship!
• You will:
• Mine and curate dataset samples from large corpus of driving data
• Leverage large language models and world foundation models for reasoning, decision making and multi-modality generation
• Design and implement robust data generation pipelines for data generation
• Design and evaluate experiments for model training
• Designing robust data generation pipelines for model training

Benefits

• Medical, dental, and vision insurance for employees and dependents
• Employee assistance programs focused on mental health
• Personalized workplace adjustments for diverse needs and abilities, including physical, mental, and neurodivergent considerations
• Access to mental health apps
• Onsite wellness centers
• Medical advocacy program for transgender employees
• Second medical opinion for you and your loved ones
• Counseling services
• Support programs including menopause benefit
• Competitive compensation
• Regular bonus and equity performance grant opportunities
• Generous 401(k) and regional retirement plans
• 1-on-1 financial coaching
• Annual cross-company compensation review and pay equity analysis
• Fertility and growing family assistance
• Parental leave and baby bonding leave
• Elder care and support
• Survivor income benefitBackup childcare
• Caregiver leave
• Paid time off, including vacation, bereavement, sick leave, parental leave, disability, and holidays
• Jury duty leave
• Military leave
• Hybrid work model with remote work opportunities also available
• Educational reimbursement
• Peer learning and coaching platform
• Donation matching programs
• Employee resource groups
• Volunteer hours
• Internal community groups and local culture clubs
• Inspiring spaces to work, recharge, and collaborate with fellow Waymonauts
• On-site meals and snacks
• Fitness centers, massage programs, and ergonomic support
• On-demand fitness, wellbeing, and cooking classes
• Commuter benefits

• Experience in strong software engineering principles
• Good understanding of Jax/Pytorch
• Currently enrolled in a graduate program (Master’s or PhD) in Computer Science or a related technical field
• You have strong problem solving and programming skills in Python and/or C++
• Experience in model evaluation and understanding metrics
• Some understanding of SQL is preferred, not required
• PhD preferred
• Previous Internship experience