8 months ago
Responsibilities
- Build scalable systems for training and fine-tuning large-scale generative models that produce realistic driving behaviors and evaluate driving scenarios.
- Lead implementation and iteration of reinforcement learning algorithms, reward functions, and training paradigms for high-fidelity driving behavior generation.
- Lead development of deep learning, generative AI, LLM, and VLM solutions for human-led triage automation and anomaly detection in self-driving behavior.
- Oversee production and optimization of machine learning models used to assess Waymo’s vehicle fleet.
- Develop a reinforcement learning from human preference data collection and evaluation system based on industry and Alphabet best practices.
- Collaborate with Prediction, Planning, Research, technical leads, and senior leadership on strategic efforts.
Requirements
- M.S. or Ph.D. in Computer Science, Machine Learning, Artificial Intelligence, or a related technical field, or equivalent practical experience.
- At least 7 years of hands-on experience developing and applying machine learning models, with significant reinforcement learning experience.
- Expertise in deep learning, sequence modeling, and generative models.
- Strong publication record or impactful project delivery in reinforcement learning or related areas.
- Proficiency in Python and standard machine learning frameworks such as JAX and TensorFlow.
- Experience with large-scale distributed training and data processing.
- Ability to lead complex and ambiguous technical projects from conception through completion.
- Preferred: 10+ years of relevant machine learning or reinforcement learning research and application experience.
- Preferred: experience in autonomous vehicles, robotics, or complex simulation environments.
- Preferred: understanding of reinforcement learning techniques for fine-tuning large models using human feedback or preferences.
- Preferred: familiarity with large-scale simulation platforms and their integration with machine learning training workflows.
- Preferred: experience designing and using metrics to evaluate complex AI systems.
- Preferred: technical leadership experience influencing senior stakeholders and driving innovation across team boundaries.
- Preferred: excellent communication skills for explaining complex technical concepts clearly.
Benefits
- Eligible for a discretionary annual bonus program, equity incentive plan, and company benefits program, subject to eligibility requirements.
- Full-time position.
Tech Stack
PythonTensorFlow
Categories
AI ResearchML Engineering
About Waymo
On the journey to be the world's most trusted driver. With the Waymo Driver, we can improve mobility while saving thousands of lives. Download the Waymo One app and ride today. Waymo reaches out to candidates from official channels only (e.g. directly from @waymo.com email addresses, or through our recruiters or sourcers who are noted as such on LinkedIn). We do not contact candidates about career opportunities through instant messaging apps like Telegram, email addresses from domains other than waymo.com (such as Gmail addresses), direct messages on Twitter, Facebook, and Instagram, or text messages. Visit waymo.com to check out our official job listings. Need support? Send us a message – we're here to help.