Power and Performance Architect, TPU

In this role, you’ll work to shape the future of AI/ML hardware acceleration. You will have an opportunity to drive cutting-edge TPU (Tensor Processing Unit) technology that powers Google's most demanding AI/ML applications. You’ll be part of a team that pushes boundaries, developing custom silicon solutions that power the future of Google's TPU. You'll contribute to the innovation behind products loved by millions worldwide, and leverage your design and verification expertise to verify complex digital designs, with a specific focus on TPU architecture and its integration within AI/ML-driven systems.

As a Power and Performance Architect, you will be a technical lead responsible for defining and driving the power architecture roadmap for Google’s next-generation TPU SOCs. You will focuses on bridging the gap between high-level architectural concepts and silicon execution, ensuring that sophisticated power management features are successfully implemented to meet demands.

The AI and Infrastructure team is redefining what’s possible. We empower Google customers with breakthrough capabilities and insights by delivering AI and Infrastructure at unparalleled scale, efficiency, reliability and velocity. Our customers include Googlers, Google Cloud customers, and billions of Google users worldwide.

We're the driving force behind Google's groundbreaking innovations, empowering the development of our cutting-edge AI models, delivering unparalleled computing power to global services, and providing the essential platforms that enable developers to build the future. From software to hardware our teams are shaping the future of world-leading hyperscale computing, with key teams working on the development of our TPUs, Vertex AI for Google Cloud, Google Global Networking, Data Center operations, systems research, and much more.

Individual pay is determined by factors including job-related skills, experience, and relevant education or training.

US: $240000 - $334000 (USD) + 25% bonus target + bonus + equity + benefits

Learn more about benefits at Google.

Responsibilities

• Lead the definition of power architecture for the next generation of TPU SOCs, optimizing for performance-per-watt across machine learning (ML) workloads.
• Bridge the architecture-to-execution gap by partnering with SOC implementation, IP providers, and hardware and software validation teams.
• Drive the design and integration of power management features, including dynamic voltage and frequency scaling (DVFS), power gating, and thermal mitigation strategies.
• Collaborate with the software community and data center teams to provide technology roadmaps that align silicon capabilities with system-level power constraints.
• Provide technical leadership and mentorship across multidisciplinary teams to anticipate future data center architectures and define SOC requirements.

Minimum qualifications:

• Bachelor's degree in Electrical Engineering, Computer Engineering, Computer Science, or a related field, or equivalent practical experience.
  
• 15 years of experience in the design or definition of computer chips (such as SOC, CPU, GPU, or hardware accelerators).
• Experience with performance analysis or performance modeling.
• Experience with power analysis, power modeling, or power delivery systems.

Preferred qualifications:

• Master's degree or PhD in Electrical Engineering, Computer Engineering or Computer Science, with an emphasis on computer architecture.
  
• Experience taking silicon power features from architecture definition through to tape-out and post-silicon validation.
• Experience with machine learning (ML) accelerator architecture and workload characterization for power optimization.
• Experience implementing DVFS or AVS, multi-voltage domain designs, and cross-layer power policies.
• Knowledge of system software components (Linux kernel, drivers) and their impact on runtime power/thermal behavior.