Senior Software Engineer, Simulation ML Infrastructure
Core
Building scalable AI/ML infrastructure to accelerate the development of ultra-realistic multi-agent simulations for testing and training the Waymo Driver autonomous vehicle system.
Role type
Senior IC machine learning infrastructure engineer (simulation)
Builds
Scalable AI/ML infrastructure for multi-billion parameter foundation models in ML accelerator-friendly simulations
Domain
Autonomous driving, simulation, large-scale distributed systems
Deliverable
production ML models
Required skills
massive model scaling, ML accelerators, large-scale distributed systems, data engineering, model development, architectural design, system-level requirements translation
Preferred skills
DeepSpeed, PyTorch, TensorFlow, Ray, autoregressive transformers, ML accelerator profiling tools, autonomous driving domain knowledge
Technologies
DeepSpeed, PyTorch, TensorFlow, Ray
Responsibilities
Design and scale large distributed systems covering the ML lifecycle; drive architectural decisions for complex systems; collaborate with core realism modeling teams to improve sim realism; derive performance and system-level requirements for large ML systems
Seniority
Senior, hands-on IC