Staff Software Engineer, ML Performance, GPUs

Google is seeking a Staff Software Engineer to join their ML, Systems, & Cloud AI organization, focusing on ML Performance and GPUs. This role is critical in optimizing Large Language Models (LLMs) and GPU infrastructure for Google's cutting-edge AI initiatives, including Gemini, Search, and Cloud services.

The position involves deep technical work with GPU programming, performance analysis, and machine learning systems. You'll be responsible for analyzing and optimizing LLM performance, maintaining benchmarks, and working closely with product teams to solve complex ML model performance challenges.

The role requires extensive experience in software development (8+ years), ML infrastructure optimization (5+ years), and a strong background in performance analysis and GPU programming. You'll be working in a global environment, contributing to Google's next-generation technologies that impact billions of users.

The compensation package is competitive, ranging from $197,000 to $291,000 base salary, plus bonus, equity, and comprehensive benefits. The position offers the opportunity to work from several locations including Kirkland, WA, Sunnyvale, CA, or Seattle, WA.

This is an excellent opportunity for a senior technical leader who wants to make a significant impact on Google's AI infrastructure and performance optimization, working with cutting-edge technology and some of the world's largest machine learning models.