Waymo is looking to develop machine learning solutions to address open problems in autonomous driving, with the goal of safely operating Waymo vehicles in numerous cities and under all driving conditions. This involves creating advanced AI models to simulate and understand complex driving scenarios.
Requirements
- Familiarity with major ML Frameworks (JAX, Tensorflow, Pytorch)
- Strong coding skills in Python
- 3+ years of experience with Deep Learning and Generative Models, and Reinforcement Learning
- Publications at top-tier conferences such as CVPR/ICCV/ECCV/ICLR/ICML/NeurIPS/IROS/CoRL etc.
- Experience in generative models for domains such as world models, trajectories, traffic, video; using techniques such as autoregressive model and/or diffusion / flow-matching
Responsibilities
- Design and implement generative behavior modeling solutions to simulate future world states / observations for embodied agents (autonomous vehicles)
- Develop and maintain reinforcement learning algorithms, including effective reward signals
- Study and analyze different behaviors of this model, such as scaling efficacy, downstream quality implications, model architecture design ablations, etc
- Conduct cutting-edge research and potentially communicating research findings to the wider academic community via technical reports and/or publications
Other
- Currently enrolled in a PhD program in computer science, statistics, applied mathematics, physics, or a related technical field of study
- Communicate to both technical and non-technical audiences regarding research motivations and research findings
- Substantial involvement in and contributions to high impact industry AI projects
- This will be a hybrid onsite internship position.
- We are committed to equal employment opportunity regardless of race, color, ancestry, religion, sex, national origin, sexual orientation, age, citizenship, marital status, disability, gender identity or Veteran status.
