Staff Machine Learning Infrastructure Engineer, Simulation

Lead ML infrastructure development for large foundation models in simulations.

• •Waymo is seeking an experienced Senior Machine Learning Infrastructure Engineer to lead the development of advanced AI/ML infrastructure for multi-billion parameter foundation models in ML accelerator-friendly simulations.
• •This role is crucial for accelerating the Simulator team in sustainably innovating and building state-of-the-art simulations.
• •Key Responsibilities Lead the development of advanced AI/ML infrastructure for multi-billion parameter foundation models.
• •Design and scale large distributed systems covering the ML lifecycle, supporting planet-scale dataset generation and model training.
• •Provide deep technical leadership on large-scale ML model architectures, especially for autonomous vehicle models.
• •Collaborate closely with core Google DeepMind and Waymo Realism Modeling teams.
• •Mentor junior engineers, growing their expertise and fostering a collaborative culture.
• •Requirements BS in Computer Science, Robotics, or similar technical field, or equivalent practical experience. 5+ years of professional software engineering experience, with at least 3 years in machine learning infrastructure.
• •Experience in developing, scaling, training, deploying, and optimizing large-scale machine learning systems.