Senior Staff Software Engineer, TPU Performance

Google Cloud is seeking a Senior Staff Software Engineer to join their ML, Systems, & Cloud AI (MSCA) organization, focusing on TPU Performance. This role is critical in developing next-generation technologies for Google's ML infrastructure. The position involves working on advanced machine learning systems, particularly focusing on TPU (Tensor Processing Unit) performance optimization and architecture. The role combines software engineering expertise with machine learning infrastructure development, requiring deep understanding of high-performance computing and ML frameworks.

The successful candidate will work on cutting-edge projects involving large language models, ML framework optimization, and hardware-software co-design. They will be part of a team that designs and implements infrastructure used by billions of users through Google services and Google Cloud. The position offers the opportunity to work on challenging problems in ML acceleration, system architecture, and performance optimization.

This role at Google offers competitive compensation including a base salary range of $248,000-$349,000 plus bonus, equity, and comprehensive benefits. The position is based in Sunnyvale, CA, and requires expertise in performance modeling, ML optimization, and large-scale distributed systems. The role combines technical leadership with hands-on development, making it ideal for experienced engineers passionate about advancing ML infrastructure and performance optimization.

Google provides a collaborative environment where engineers can work on impactful projects while having opportunities for growth and development. The position requires both technical depth in ML systems and the ability to work across teams to drive innovation in ML acceleration and performance optimization.