Filtr uses AI to scan 1000+ jobs and finds postings that perfectly matches your resume

Waymo is looking to solve the problem of improving access to mobility while saving thousands of lives lost to traffic crashes by building the Waymo Driver—The World's Most Experienced Driver™, and the Simulation ML Infrastructure team is focused on building scalable AI/ML infrastructure to accelerate the Simulator team in sustainably innovating and building state of the art simulations of realistic environments for the testing and training of the Waymo Driver.

Requirements

• 5+ years of professional software engineering experience, with at least 3 years in machine learning infrastructure such as developing, designing, scaling, training, deploying, and optimizing large-scale machine learning systems from data to model.
• Solid experience in the development and optimization of machine learning infrastructure tools like DeepSpeed, PyTorch, TensorFlow, Ray, or similar frameworks.
• Strong understanding of state-of-the-art machine learning models and algorithms such as autoregressive transformers and familiarity scaling large models across ML accelerator profiling tools to uncover performance bottlenecks.
• Practical familiarity in Autonomous Driving, Simulations, and ML accelerators is a plus.
• Experience with massive model scaling, ML accelerators, and large-scale distributed systems.
• Experience with designing and scaling AI/ML infrastructure for multi-billion parameter foundation models in ML accelerator-friendly simulations.
• Familiarity with large foundation models trained on massive datasets to model the real world.

Responsibilities

• Be part of a world-class, high-performing research engineering team to advance the state of the art of ultra realistic multi-agent simulations using foundation models.
• Collaborate closely with the core Waymo Realism Modeling team in London and Waymo Oxford to use large foundation models to improve sim realism.
• Work at the intersection of data engineering, model development, and simulations, and drive architectural decisions.
• Own large, complex systems, driving architectures and designs that meet technical and business objectives.
• Design and scale large distributed systems covering the ML lifecycle, supporting planet-scale dataset generation, model training, and evaluation.
• Collaborate cross-functionally to derive performance and system-level requirements for large ML systems.
• Translate product/business goals into measurable technical deliverables, ensuring system component alignment.

Other

• Excellent communication skills, both verbal and written, with the ability to translate complex technical concepts for a broad audience.
• Strong leadership skills with experience driving ambiguous problems end-to-end, with a willingness and independence to pick up whatever knowledge to get the job done.
• Passionate about building infrastructure, libraries, tools, and pipelines for engineers and scientists.
• Ability to work in a world-class, high-performing research engineering team.
• Eligibility to participate in Waymo’s discretionary annual bonus program, equity incentive plan, and generous Company benefits program, subject to eligibility requirements.