Senior Software Engineer (Machine Learning)

TL;DR

Senior Software Engineer (Machine Learning): Developing and improving pre-trained models for autonomous driving with an accent on large-scale reinforcement learning and model efficiency. Focus on building scalable systems across diverse environments and enhancing model intelligence.

Location: Onsite in Mountain View, California, United States

Salary Range: $204,000—$259,000 USD

Company

Waymo is an autonomous driving technology company focused on building the Waymo Driver to improve mobility and safety.

What you will do

• Report into the Head of ML Frameworks & Efficiency
• Develop the core training system for adapting RL techniques to unprecedented scales and heterogeneous environments
• Collaborate with teams to integrate the latest rollout strategies, policies, and RL algorithms
• Improve the end-to-end RL training pipeline for efficient and scalable learners
• Build evaluations, analyze experimental results, and iterate quickly to improve model performance
• Stay current with the latest research in RL and related models

Requirements

• B.S. in Computer Science, Math, or 8+ years equivalent real-world experience
• Proficient in distributed systems design with an understanding of ML efficiency
• Experience with ML frameworks, including TensorFlow, JAX, XLA
• Solid programming skills in Python and C++
• Practical familiarity with profiling tools to uncover performance bottlenecks