Staff Machine Learning Engineer
Waymo
about 2 months ago
London, United Kingdom
Staff+
H1B Sponsor
Responsibilities
- Build scalable systems for training and fine-tuning large-scale generative models.
- Lead the implementation and iteration of novel reinforcement learning algorithms.
- Develop cutting-edge deep learning models and generative AI solutions.
- Oversee the production and optimization of machine learning models for vehicle assessment.
- Monitor and assimilate best practices for developing RLHF-based data collection systems.
- 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.
- Demonstrated 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