Conduct applied world model research and development
Post-train Waymo's world model
Design compelling experiments by training and evaluating large generative models
Present results to peers and leadership
Leverage AI to write high quality code, unit tests, and documentation, and automate workflows

🎯 Requirements

Masters or PhD in deep learning in generative / world modeling
3+ years of experience with modern deep learning frameworks (e.g., JAX, TensorFlow 2, Pytorch)
Experience with diffusion / flow matching models
Experience with reinforcement learning
Experience training large foundational models

🎁 Benefits

Discretionary annual bonus program
Equity incentive plan
Company benefits program

Apply on employer's website

This employer gathers applications via their own applicant tracking system.

You will be redirected to an external application form.

Share job

Meet JobCopilot: Your Personal AI Job Hunter

Automatically Apply to Data Jobs. Just set your preferences and Job Copilot will do the rest — finding, filtering, and applying while you focus on what matters.

Activate JobCopilot