Staff ML Engineer, Generative Model Performance & Efficiency

, you will report to a Senior Staff Engineering Manager You will: Analyze model architectures and identify bottlenecks in training and inference performance (e.g., memory bandwidth, compute, communication). Apply and develop techniques such as quantization (e.g., FP8, INT4), pruning, knowledge distillation, and efficient attention mechanisms. Optimize model code for specific hardware accelerators (TPUs, GPUs)…

Waymo has 373 open roles right now, 7 of them remote. Most are in Engineering, Data & Research and Operations. The median advertised salary across its roles with a published range is $238K. See all its roles