9 months ago
Base Salary
$204k - $259k/yr
Responsibilities
- Develop advanced AI/ML infrastructure for multi-billion-parameter foundation models used in realistic, multi-agent simulations.
- Drive architectural decisions and own large, complex systems that meet technical and business objectives.
- Design and scale distributed systems for planet-scale dataset generation, model training, and evaluation across the ML lifecycle.
- Collaborate with research engineering and realism modeling teams to use foundation models to improve simulation realism.
- Derive performance and system-level requirements for large ML systems and translate business goals into measurable technical deliverables.
Requirements
- At least 5 years of professional software engineering experience, including at least 3 years developing, designing, scaling, training, deploying, or optimizing large-scale machine learning systems.
- Experience with machine learning infrastructure tools such as DeepSpeed, PyTorch, TensorFlow, Ray, or similar frameworks.
- Strong understanding of state-of-the-art machine learning models and algorithms, including autoregressive transformers, and experience scaling large models with ML accelerator profiling tools.
- Experience driving ambiguous technical problems end-to-end and building infrastructure, libraries, tools, and pipelines for engineers and scientists.
- Strong verbal and written communication skills with the ability to explain complex technical concepts to broad audiences.
- Practical familiarity with autonomous driving, simulations, and ML accelerators is preferred.
Benefits
- Eligible for a discretionary annual bonus program, equity incentive plan, and company benefits program, subject to eligibility requirements.
- Full-time position available across US locations; remote availability and location-specific salary ranges are discussed during the hiring process.
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.