Machine Learning Engineer
Core
Building and operating scalable machine learning systems, simulation workflows, and insight tools to improve the evaluation and developer onboarding journeys for the Waymo Driver.
Role type
Senior Machine Learning Engineer (Reinforcement Learning & Generative AI)
Builds
Scalable ML systems, simulation workflows, and insight tools for the Waymo Driver evaluation and developer onboarding.
Domain
Autonomous driving, Reinforcement Learning, Generative AI
Deliverable
production ML models
Required skills
Reinforcement Learning, Deep Learning, Sequence Modeling, Generative Models (LLM/VLM), Python, JAX, TensorFlow, Large-scale distributed training, Data processing
Preferred skills
Autonomous vehicles domain experience, Robotics, Complex simulation environments, RLHF techniques, Metrics design for complex AI systems, Technical leadership
Technologies
JAX, TensorFlow, Python
Responsibilities
Build scalable systems for training and fine-tuning large-scale generative models to produce realistic driving behaviors; Lead implementation and iteration of novel RL algorithms and reward functions; Develop Deep Learning and Generative AI solutions to enhance human-led triaging and automate high-volume workflows; Oversee production and optimization of ML models assessing the vehicle fleet; Monitor and assimilate best practices to develop RLHF-based data collection and evaluation systems.
Seniority
Senior, hands-on IC with leadership responsibilities