Senior Staff TPU Architecture & Co-Design Engineer

Google; Sunnyvale (CA); On-site; USD 240,000 - 333,000; Full time

Job summary

Google is seeking a Senior Staff Co-Design Engineer to shape TPU architecture and AI/ML hardware acceleration. You will drive cutting-edge TPU technology and develop custom silicon for high-performance AI workloads, collaborating across research, software and hardware teams.

You will bridge model innovation with hardware design, setting long-term architectural roadmaps for training and serving capabilities, and engaging with the global ML community to advance frontier models.

Qualifications

• Bachelor's degree or equivalent practical experience required.
• 12 years of experience in computer architecture or hardware-software co-design.
• Experience developing systems for performance modeling, simulation or analysis.

Responsibilities

• Define and drive the technical roadmap and architecture for hardware/software stack.
• Act as liaison between advanced research, software and hardware teams.
• Architect and oversee next-generation simulation frameworks and models.
• Advocate system-level performance analysis for distributed ML systems.
• Manage cross-functional partnerships to influence strategy and production.

Skills

Performance modeling

System analysis

Hardware-software co-design

Education

Bachelor's degree in Electrical/Computer Engineering or Computer Science

Tools

PyTorch

TensorFlow

Job description

Google is seeking a Senior Staff Co-Design Engineer to shape TPU architecture and AI/ML hardware acceleration. You will drive cutting-edge TPU technology and develop custom silicon for high-performance AI workloads, collaborating across research, software and hardware teams.

You will bridge model innovation with hardware design, setting long-term architectural roadmaps for training and serving capabilities, and engaging with the global ML community to advance frontier models.