Requirements:

• BS in Computer Science, Robotics, similar technical field of study, or equivalent practical experience
• 5+ years of professional software engineering experience, with at least 3 years in machine learning infrastructure

Nice to haves:

• MS in Computer Science, Robotics, similar technical field of study, or equivalent practical experience
• 10+ years of professional software engineering experience, with at least 5 years in machine learning infrastructure
• Solid experience in the development and optimization of machine learning infrastructure tools like DeepSpeed, PyTorch, TensorFlow, or similar frameworks
• Strong expertise in distributed training techniques, including gradient sharding and optimization strategies for scaling large models
• Deep understanding of state-of-the-art machine learning models such as auto-regressive transformers
• Practical familiarity in Autonomous Driving, Simulations, and ML accelerators is a huge plus

What you'll be doing:

• 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.
• Design and scale large distributed systems covering the ML lifecycle, supporting planet-scale dataset generation and model training.
• Mentor junior engineers, growing their expertise and fostering a collaborative culture.

Benefits & Perks:

• Expected base salary range: £155,000 - £163,000 GBP
• Participation in Waymo’s discretionary annual bonus program
• Equity incentive plan
• Generous Company benefits program, subject to eligibility requirements