Staff Tech Lead Manager, Machine Learning, Simulator Evaluation
Core
Lead a top-tier applied ML team to build evaluation frameworks, infrastructure, and AI-powered tooling for World Model understanding and evaluation within Waymo's autonomous driving simulator.
Role type
Staff Tech Lead Manager, Machine Learning
Builds
Eval frameworks, infrastructure, and AI-powered tooling for World Model understanding and evaluation
Domain
Autonomous driving, Simulation, Generative AI
Deliverable
production ML models
Required skills
Technical leadership, ML system deployment, Python/C++, ML frameworks (PyTorch/JAX/Tensorflow), deep learning, transformers, large-scale data processing, evaluation framework design
Preferred skills
Generative model training/evaluation (RL, diffusion, world models), GPU/TPU cluster management, MLOps, AI-powered product development
Technologies
PyTorch, JAX, Tensorflow
Responsibilities
Lead applied ML team, drive technical direction, architect ML-driven metrics, chart technical direction for World Model evaluation, build AI-powered internal tools, partner with cross-functional teams
Seniority
Staff, hands-on IC with management