Tech Lead, TPU AI Infra: Architect On-Prem AI Supercomputer

Google; Kirkland (WA); On-site; USD 207,000 - 300,000; Full time

Benefits offered by this job

Health insurance

401(k) with company match

Paid Time Off: 20 days per year

Sick Time: 40 hours/year

Maternity Leave (Short-Term Disability

Baby Bonding Leave: 18 weeks

Holidays: 13 paid days

Job summary

Google is seeking a Tech Lead for TPU AI Infrastructure to guide the on-prem AI hardware software stack and drive the technical roadmap across a distributed system. You will influence cross-functional teams, own architecture decisions, and shape platform capabilities to meet demanding performance and reliability requirements.

The role requires deep expertise in scalable infrastructure, wear many hats, and collaborate with leadership to deliver high-impact solutions while maintaining strong

Qualifications

• Bachelor's degree or equivalent practical experience.
• 8 years of software development experience in one or more languages.
• 5 years building large-scale infrastructure or distributed systems.
• 5 years testing and launching software products.
• 3 years of software design and architecture.

Responsibilities

• Define the technical roadmap and architecture for inventory and fleet management, focusing on Temporal orchestration, automated node re-bootstrapping, and self-service remediation.
• Own architectural and design decisions for the platform, collaborating with leadership and cross-functional teams to prioritize efforts, resolve roadblocks, and manage technical debt.
• Establish and maintain machine health Service Level Objectives (SLOs), driving continuous improvements in fleet-wide observability and environmental monitoring.
• Foster a culture of engineering excellence by establishing shared responsibility models with tenant teams and enforcing robust system-level access guardrails.
• Coach and mentor engineers on the team, guiding their technical development and helping them grow their impact.

Skills

Python

C

C++

Java

JavaScript

Education

Bachelor's degree or equivalent practical experience

Tools

Kubernetes

ArgoCD

Temporal

Container orchestration

PXE

Job description

Google is seeking a Tech Lead for TPU AI Infrastructure to guide the on-prem AI hardware software stack and drive the technical roadmap across a distributed system. You will influence cross-functional teams, own architecture decisions, and shape platform capabilities to meet demanding performance and reliability requirements.

The role requires deep expertise in scalable infrastructure, wear many hats, and collaborate with leadership to deliver high-impact solutions while maintaining strong