Tech Lead, AI Infra & TPU Systems

Google Inc.

Sunnyvale (CA)

On-site

USD 207,000 - 300,000

Full time

3 days ago
Be an early applicant

Get more replies from employers

Send a job-specific resume in minutes.

Benefits offered by this job

Equity
Benefits

Job summary

Google is building the future of at-scale computing with an on-premises AI supercomputer. You will own core platform architecture, coordinate with leadership, and drive reliability across fleet health and observability in a security-conscious environment.

As a Share Tech Lead, TPU AI Infrastructure, you will mentor engineers, shape cross-functional strategies, and ensure robust delivery of critical AI hardware and software systems at Google.

Qualifications

  • Bachelor's degree or equivalent practical experience.
  • 5 years of experience building and developing large-scale infrastructure, distributed systems or networks, or experience with compute technologies, storage, or hardware architecture.
  • 5 years of experience testing, and launching software products.
  • 3 years of experience with 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 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

Large-scale infrastructure
Distributed systems
Compute technologies
Storage
Hardware architecture
Software design & architecture
Temporal workflows
Kubernetes
ArgoCD
Container orchestration

Education

Bachelor's degree or equivalent

Tools

Kubernetes
ArgoCD
Container orchestration

Job description

Google is building the future of at-scale computing with an on-premises AI supercomputer. You will own core platform architecture, coordinate with leadership, and drive reliability across fleet health and observability in a security-conscious environment.

As a Share Tech Lead, TPU AI Infrastructure, you will mentor engineers, shape cross-functional strategies, and ensure robust delivery of critical AI hardware and software systems at Google.

Get your free, confidential resume review.
or drag and drop your file here.
Similar jobs

Similar jobs worth comparing

Tech Lead: AI Infra & On-Prem TPU Systems
Tech Lead: AI Infra & On-Prem TPU Systems

Socket.dev • Sunnyvale (CA)

On-site
USD 207,000 - 300,000
Tech Lead — On-Prem AI Infrastructure & Orchestration
Tech Lead — On-Prem AI Infrastructure & Orchestration

Google • Sunnyvale (CA)

On-site
USD 207,000 - 300,000
Bonus target
Equity
Benefits
Senior AI/ML Infrastructure Engineer for TPU Health
Senior AI/ML Infrastructure Engineer for TPU Health

Google • United States

On-site
USD 174,000 - 252,000
Health insurance
Dental insurance
Vision insurance
+8
Senior AI/ML Infra Engineer — TPU Health Systems
Senior AI/ML Infra Engineer — TPU Health Systems

Google • Kirkland (WA)

On-site
USD 174,000 - 252,000
Health insurance
Retirement benefits
Paid time off (PTO)
+4
Senior Hardware Engineering Leader — AI Infrastructure
Senior Hardware Engineering Leader — AI Infrastructure

Google Inc. • Sunnyvale (CA), Kirkland (WA)

On-site
USD 236,000 - 329,000
Health Insurance
Dental Insurance
Vision Insurance
+8
Senior AI/ML Infrastructure Engineer (TPU Health)
Senior AI/ML Infrastructure Engineer (TPU Health)

Google Inc. • Kirkland (WA), Sunnyvale (CA)

On-site
USD 174,000 - 252,000
Health insurance
Retirement plan
Paid time off
+3
Principal AI Systems Engineer, Hyper-scale Infra
Principal AI Systems Engineer, Hyper-scale Infra

Google Inc. • Sunnyvale (CA)

On-site
USD 364,000 - 505,000
Principal Engineer, AI Infra & Bare-Metal Orchestration
Principal Engineer, AI Infra & Bare-Metal Orchestration

Socket.dev • Sunnyvale (CA)

On-site
USD 364,000 - 505,000
Equity
Bonus program
Benefits
Tech Lead, Internal Platforms & AI Systems
Tech Lead, Internal Platforms & AI Systems

Google • United States

On-site
USD 207,000 - 300,000
Staff AI/ML Infra Architect for TPU Supercomputers
Staff AI/ML Infra Architect for TPU Supercomputers

Google • Sunnyvale (CA)

On-site
USD 207,000 - 301,000
20% bonus target
Equity
Comprehensive benefits package