Machine Learning Engineer

📋 Description

• Build scalable systems for training and fine-tuning large-scale generative models to produce
• Lead the implementation and iteration of novel RL algorithms, reward functions, and training
• Develop cutting-edge Deep Learning models and Generative AI solutions (LLM/VLM) to enhance
• Oversee production and optimization of ML models to assess Waymo’s expansive fleet miles.
• Monitor and adopt best practices to develop RLHF-based data collection and evaluation systems.
• Collaborate with teams across Waymo to deliver on key strategic efforts.

🎯 Requirements

• M.S. or Ph.D. in Computer Science, ML, AI, or related field, or equivalent practical experience.
• 5+ years of hands-on ML development with a strong focus on Reinforcement Learning.
• Expertise in deep learning, sequence modeling, and generative models.
• Strong publication record or history of impactful project delivery in RL or related areas.
• Proficiency in Python and ML frameworks (e.g., JAX, TensorFlow).
• Experience with large-scale distributed training and data processing.

🎁 Benefits

• Salary range £120,000 – £130,000 GBP
• Discretionary annual bonus, equity incentive plan, and generous company benefits
• Opportunities to work with cutting-edge ML, RL, and AI tooling