Senior Machine Learning Engineer, Simulation Evaluation

Job description

Job responsibilities

• Lead the design, development and deployment of cutting-edge evaluation approaches to assess realism of state-of-the-art multimodel world models and generative systems for simulation use cases at Waymo.
• Architect and implement robust and scalable machine learning pipelines for tuning, evaluating, and deploying large-scale discriminator models for the purposes of simulator realism evaluation.
• Evaluate open-source and production-ready video generation techniques that measure realism (e.g. temporal stability, multi-modal consistency, geometric discrepancy, condition following, etc.)
• Apply vision language models to evaluate semantic understanding and controllability across our world simulation products.
• Collaborate with research teams across Waymo and Alphabet to integrate advancements in 4D world modeling and generative AI into production systems.
• Mentor engineers on the team and provide technical guidance on architecture and execution.

Job 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 a portfolio of shipped products or peer-reviewed publications.
• Proficient programming skills in Python and hands-on experience with modern machine learning frameworks such as Jax, Flax, or PyTorch.
• Experience designing and implementing evaluation frameworks for complex systems or machine learning models.