3 months ago
Responsibilities
- Lead development of advanced AI/ML infrastructure for multi-billion-parameter foundation models used in autonomous-driving simulations.
- Provide technical leadership on large-scale ML model architectures and own complex systems from architecture through implementation.
- Design and scale distributed systems supporting planet-scale dataset generation and model training across the ML lifecycle.
- Collaborate with cross-functional teams to define performance and system requirements and translate business goals into technical deliverables.
- Collaborate with Google DeepMind and Waymo teams to apply large models to improve simulation realism.
- Mentor junior engineers and foster a collaborative engineering culture.
Requirements
- Bachelor’s degree in Computer Science, Robotics, or a similar technical field, or equivalent practical experience.
- At least 5 years of professional software engineering experience, including at least 3 years developing, scaling, training, deploying, or optimizing large-scale machine learning infrastructure.
- Preferred: Master’s degree in Computer Science, Robotics, or a similar technical field, or equivalent practical experience.
- Preferred: At least 10 years of professional software engineering experience, including at least 5 years in machine learning infrastructure.
- Experience with machine learning infrastructure tools such as DeepSpeed, PyTorch, TensorFlow, or similar frameworks.
- Strong expertise in distributed training, gradient sharding, optimization strategies for scaling large models, and ML accelerator profiling.
- Deep understanding of state-of-the-art machine learning models such as autoregressive transformers and familiarity with custom kernels for diverse hardware compute.
- Practical familiarity with autonomous driving, simulations, and ML accelerators is preferred.
Benefits
- Full-time position with eligibility for a discretionary annual bonus program, equity incentive plan, and company benefits program, subject to eligibility requirements.
Tech Stack
PyTorchTensorFlow
Categories
About Waymo
On the journey to be the world's most trusted driver. With the Waymo Driver, we can improve mobility while saving thousands of lives. Download the Waymo One app and ride today. Waymo reaches out to candidates from official channels only (e.g. directly from @waymo.com email addresses, or through our recruiters or sourcers who are noted as such on LinkedIn). We do not contact candidates about career opportunities through instant messaging apps like Telegram, email addresses from domains other than waymo.com (such as Gmail addresses), direct messages on Twitter, Facebook, and Instagram, or text messages. Visit waymo.com to check out our official job listings. Need support? Send us a message – we're here to help.