Staff ML Engineer, Perception Research
WaymoAutonomous Vehicles company
Mountain View, United States$251,000 - $310,000 USDLead
Dragoneer Investment Group
DST Global
Sequoia Capital
Andreessen Horowitz
Alphabet
Mubadala Capital
Data & AI
About the role
TL;DR
Develops and deploys advanced ML models for 3D perception in autonomous driving.
- •Waymo is seeking a Staff ML Engineer to join the AI Foundations team, focusing on developing cutting-edge machine learning solutions for autonomous driving.
- •This role involves training and deploying state-of-the-art Multimodal LLMs and World models for 3D Perception.
- •Key Responsibilities Conduct comprehensive experimentation to train and deploy state-of-the-art Multimodal LLMs and World models for 3D Perception.
- •Partner effectively with engineering and research teams to deploy new models and implement efficient workflows.
- •Apply and develop techniques such as quantization, pruning, knowledge distillation, and efficient attention mechanisms.
- •Develop and maintain scalable data pipelines for Training & Eval.
- •Design and implement evaluation frameworks for perception models.
- •Requirements PhD or Masters in Computer Science, Machine Learning, Robotics, or a similar technical field, with 4+ years of industry or post-doc research experience in Reinforcement Learning or Foundation Models.
- •Proficiency in implementing model training flows in a scalable, distributed and performant manner.
- •Proficiency in JAX, Flax, and potentially TensorFlow/PyTorch.
- •Hands-on experience with optimizing the training and inference of Transformer architectures.
Required skills
JAXLLMsComputer VisionPython
Nice-to-have skills
TensorFlowPyTorch
Domain expertise
automotivedeeptechai
Benefits & perks
Health insurance, Dental insurance, Vision insurance, Life insurance, Disability insurance, 401(k) with company match, Paid Time Off, Sick Time, Maternity Leave, Baby Bonding Leave, Holidays
Tech stack
PythonJAXTensorFlowPyTorchComputer VisionLLMs