GrepJob
Waymo

Staff Machine Learning Engineer

Waymo
Apply
about 4 hours ago
London, United KingdomStaff+
H1B Sponsor

Responsibilities

  • Build scalable systems for training and fine-tuning large-scale generative models.
  • Lead the implementation and iteration of novel RL algorithms and reward functions.
  • Develop cutting-edge Deep Learning models and Generative AI solutions.
  • Oversee the production and optimization of machine learning models for vehicle assessment.
  • Monitor best practices from Alphabet and the industry for RLHF-based data collection.
  • Collaborate with multiple teams and senior leadership on strategic efforts.

Requirements

  • M.S. or Ph.D. in Computer Science, Machine Learning, AI, or a related field.
  • 7+ years of hands-on experience in developing and applying Machine Learning models.
  • Expertise in deep learning, sequence modeling, and generative models.
  • Strong publication record or impactful project delivery in RL or related areas.
  • Proficiency in Python and standard ML frameworks like JAX and TensorFlow.
  • Experience with large-scale distributed training and data processing.
  • Proven ability to lead complex technical projects from conception to completion.

Benefits

  • Eligible for Waymo’s discretionary annual bonus program.
  • Participation in equity incentive plan.
  • Generous Company benefits program.

Tech Stack

PythonTensorFlow

Categories

AI & MLData Science