🛠 Tech Stack

💼 About This Role

You'll design and implement generative world action models for autonomous driving at Waymo, a leader in self-driving technology. Your work will directly impact safe vehicle operation across dozens of cities. Collaborate with Alphabet research teams on cutting-edge ML solutions.

🎯 What You'll Do

• Design and implement generative world action models for embodied agents.
• Develop and maintain scalable data pipelines from multiple sources.
• Design and implement evaluation strategies for world action models.
• Analyze model behaviors like scaling efficacy and architecture design.

📋 Requirements

• PhD in Computer Science or equivalent deep learning research experience.
• 5+ years of experience with Deep Learning and Generative Models.
• Strong coding skills in Python and familiarity with JAX, Tensorflow, Pytorch.
• Experience in large-scale distributed training and parallelism.

✨ Nice to Have

• Publications at top-tier conferences like CVPR/ICCV/NeurIPS.
• Contributions to high-impact industry AI projects.
• Experience training large world models or world-action models.

🎁 Benefits & Perks

• 💰 Competitive salary ($213k-$263k)
• 📈 Discretionary annual bonus
• 📊 Equity incentive plan
• 🎁 Generous company benefits