3 months ago
Base Salary
$213k - $263k/yr
Responsibilities
- Design, develop, and deploy evaluation approaches for assessing the realism of multimodal world models and generative systems.
- Architect and implement scalable machine learning pipelines for tuning, evaluating, and deploying large-scale discriminator models for simulator realism evaluation.
- Evaluate video-generation techniques for temporal stability, multimodal consistency, geometric discrepancy, and condition following.
- Apply vision-language models to assess semantic understanding and controllability across simulation products.
- Collaborate with Waymo and Alphabet research teams to integrate 4D world modeling and generative AI advances into production systems.
- Mentor engineers and provide technical guidance on architecture and execution.
Requirements
- Bachelor's, Master's, or PhD in computer science, machine learning, robotics, or a related field.
- Five or more years of experience in machine learning engineering or applied deep learning, supported by shipped products or peer-reviewed publications.
- Proficiency in Python and hands-on experience with JAX, Flax, or PyTorch.
- Experience designing and implementing evaluation frameworks for complex systems or machine learning models.
- Preferred experience training large-scale generative models, including diffusion models, flow matching, or vision-language models.
- A PhD and success delivering machine learning products focused on 3D generative models, world models, or video generation is preferred.
- Preferred experience simulating camera, lidar, and radar sensor data or modeling semantic scenes.
- Experience with autonomous systems, robotics software, or autonomous vehicle simulations is preferred.
- Preferred experience training and optimizing large-scale models on GPU or TPU clusters for efficient production serving.
- Professional C++ experience for high-performance production systems is preferred.
Benefits
- Full-time position with eligibility for a discretionary annual bonus program, equity incentive plan, and company benefits program.
- Remote work may be available depending on the role and location.
Categories
About Waymo
Waymo develops the Waymo Driver, a full-stack autonomous driving system, and operates the Waymo One robotaxi service that offers paid, driverless rides in select U.S. cities. The company monetizes through ride-hailing fares and partnerships integrating its technology on multiple vehicle platforms. Founded in 2009 as Google’s self-driving project, Waymo is headquartered in Mountain View and is an Alphabet subsidiary, with over 10 million rider-only trips and 100+ million autonomous miles on public roads.