Senior Staff ML Engineer, Driver Understanding and Evaluation
Core
Building and operating scalable ML/data systems, simulation workflows, and evaluation tools to improve the Waymo Driver's performance using expert human judgments and advanced ML models.
Role type
Senior Staff ML Engineer (Evaluation & Strategy)
Builds
Scalable ML systems, simulation workflows, and evaluation tools for the Waymo Driver
Domain
Autonomous driving, Embodied AI, Reinforcement Learning
Deliverable
production ML models
Required skills
Reinforcement Learning, Deep Learning, Generative Models (LLM/VLM), Sequence Modeling, Python, Distributed Training, Data Processing, Technical Leadership, Cross-functional Alignment
Preferred skills
Autonomous Vehicles domain experience, Complex Simulation environments, RLHF techniques, Large-scale simulation platform integration, AI system metrics design
Technologies
JAX, TensorFlow
Responsibilities
Grow end-to-end strategy for ML-based evaluation metrics; Architect and build systems for training/fine-tuning large-scale generative models; Lead design of novel RL algorithms and reward functions; Develop Deep Learning and Generative AI solutions for behavior analysis; Monitor and assimilate best practices for RLHF data collection; Provide technical mentorship and thought leadership; Guide and align multiple teams on evaluation strategy
Seniority
Senior Staff, hands-on IC with people management