10 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
Waymo develops the Waymo Driver, a full-stack autonomous driving system, and operates the Waymo One robotaxi service that offers paid, driverless rides in select U.S. cities. The company monetizes through ride-hailing fares and partnerships integrating its technology on multiple vehicle platforms. Founded in 2009 as Google’s self-driving project, Waymo is headquartered in Mountain View and is an Alphabet subsidiary, with over 10 million rider-only trips and 100+ million autonomous miles on public roads.