Google Cloud is seeking a Senior Staff Software Engineer to lead GPU performance optimization efforts at Google scale. This role sits within the ML, Systems, & Cloud AI (MSCA) organization, which is responsible for the hardware, software, and infrastructure powering Google's services and Cloud offerings. The position focuses on optimizing GPU performance for machine learning workloads, working with cutting-edge GPU architectures, and developing solutions that impact billions of users.
The ideal candidate will have deep expertise in GPU architecture, compiler optimization, and machine learning infrastructure. They will work on critical projects involving OpenXLA, MLIR, and Triton, while collaborating with teams across Google to improve model performance and influence future GPU hardware designs. This role offers the opportunity to work with the latest GPU technologies and shape the future of accelerated computing at Google.
The position offers competitive compensation including a base salary range of $248,000-$349,000 plus bonus, equity, and comprehensive benefits. The role can be based in several major tech hubs including Sunnyvale, Kirkland, Mountain View, New York, or Austin, offering flexibility in location while working on some of the most challenging and impactful problems in GPU optimization and machine learning infrastructure.
This is a unique opportunity to work at the intersection of hardware and software, optimizing performance for Google's most critical AI and machine learning workloads while collaborating with world-class engineers and researchers. The role requires both deep technical expertise and the ability to lead and influence technical strategy across multiple teams and projects.