Staff Machine Learning Infrastructure Engineer, Simulation

Waymo is an autonomous driving technology company with the mission to be the world's most trusted driver. They are seeking an experienced Senior Machine Learning Infrastructure Engineer to lead the development of advanced AI/ML infrastructure for simulations. The role involves collaborating with various teams, providing technical leadership, and mentoring junior engineers to innovate in realistic simulations for autonomous driving.

Responsibilities:

• Be part of a world-class, high-performing research engineering team to advance the state of the art of ultra realistic multi-agent simulations using foundation models
• Collaborate closely with the core Google DeepMind and Waymo Realism Modeling teams in London, and Waymo Oxford to use the large models to improve sim realism
• Provide deep technical leadership on large-scale ML model architectures, especially for autonomous vehicle models. Work at the intersection of data engineering, model development, and deployment, and provide guidance on architectural decisions and technical directions. Own large, complex systems, driving architectures that meet technical and business objectives
• Design and scale large distributed systems covering the ML lifecycle, supporting planet-scale dataset generation and model training
• Collaborate cross-functionally to derive performance and system-level requirements for large ML systems. Translate product/business goals into measurable technical deliverables, ensuring system component alignment
• Mentor junior engineers, growing their expertise and fostering a collaborative culture