Technical Program Manager, AI2, TPU Software

Google LLC; Sunnyvale (CA); On-site; USD 192,000 - 278,000; Full time

Benefits offered by this job

Bonus target

Equity grants

Benefits

Job summary

Google LLC is seeking a Technical Program Manager for AI2, TPU Software in Sunnyvale. Lead complex, cross-team initiatives spanning TPU hardware, ML frameworks, and scalable inference. Manage requirements, risks, schedules, and stakeholder communications with engineers and executives.

You will drive strategy for production-grade ML infrastructure, coordinate with Jax and PyTorch teams, and expand support into multimodal and world models while automating onboarding pipelines.

Qualifications

• Bachelor's degree in a technical field or equivalent practical experience.
• 8+ years of experience in program management.
• Experience with AI/ML projects, tools, or infrastructure.

Responsibilities

• Develop and execute end-to-end program strategies for ML infrastructure and TPU ecosystem.
• Partner with cross-functional teams to define milestones and platform capabilities.
• Drive execution across workstreams, manage dependencies and risks, and report to leadership.
• Ensure production-readiness and scale for hardware/software deployments and developer adoption.
• Communicate technical tradeoffs and strategic recommendations to executives.

Skills

JAX

PyTorch

Qwen

GLM

vLLM

SGLang

Education

Bachelor's degree in a technical field

Job description

Technical Program Manager, AI2, TPU Software

Share Technical Program Manager, AI2, TPU Software

corporate_fare Google place Sunnyvale, CA, USA

• Bachelor's degree in a technical field, or equivalent practical experience.
  
• 8 years of experience in program management.
  
• Experience with artificial intelligence/machine learning projects, tools, or infrastructure.

Preferred qualifications:

• 8 years of experience managing cross-functional or cross-team projects.
• Experience with Jax and PyTorch frameworks, OSS models like Qwen and GLM, and inference stacks like vLLM and SGLang.
• Experience running production facing large scale software programs.

About the job

A problem isn’t truly solved until it’s solved for all. That’s why Googlers build products that help create opportunities for everyone, whether down the street or across the globe. As a Technical Program Manager at Google, you’ll use your technical expertise to lead complex, multi-disciplinary projects from start to finish. You’ll work with stakeholders to plan requirements, identify risks, manage project schedules, and communicate clearly with cross-functional partners across the company. You're equally comfortable explaining your team's analyses and recommendations to executives as you are discussing the technical tradeoffs in product development with engineers.

The Core ML Program Management Office (PMO) is looking for a strong Technical Program Manager (TPM) to drive the strategy and execution for the TPU ecosystem, scaling production-grade inference and training infrastructure across Jax and PyTorch frameworks. You will lead efforts to establish price-performance and total cost of ownership leadership for frontier AI models. Your work will focus on qualifying next-generation TPU hardware, advancing distributed orchestration for agentic workloads and tiered KV-cache management, and expanding support into multimodal, diffusion, physical AI, and world models. Additionally, you will foster developer velocity by automating model onboarding pipelines and optimizing specialized kernel libraries.

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.

About the job

A problem isn’t truly solved until it’s solved for all. That’s why Googlers build products that help create opportunities for everyone, whether down the street or across the globe. As a Technical Program Manager at Google, you’ll use your technical expertise to lead complex, multi-disciplinary projects from start to finish. You’ll work with stakeholders to plan requirements, identify risks, manage project schedules, and communicate clearly with cross-functional partners across the company. You're equally comfortable explaining your team's analyses and recommendations to executives as you are discussing the technical tradeoffs in product development with engineers.

The Core ML Program Management Office (PMO) is looking for a strong Technical Program Manager (TPM) to drive the strategy and execution for the TPU ecosystem, scaling production-grade inference and training infrastructure across Jax and PyTorch frameworks. You will lead efforts to establish price-performance and total cost of ownership leadership for frontier AI models. Your work will focus on qualifying next-generation TPU hardware, advancing distributed orchestration for agentic workloads and tiered KV-cache management, and expanding support into multimodal, diffusion, physical AI, and world models. Additionally, you will foster developer velocity by automating model onboarding pipelines and optimizing specialized kernel libraries.

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: $192000 - $278000 (USD) + 20% bonus target + equity + benefits

Learn more about benefits at Google .

• Develop and execute end-to-end program strategies for complex ML infrastructure, covering scope, resource planning, risk management, and roadmap delivery.
• Partner with cross-functional engineering, product, and research teams to define technical milestones and prioritize key platform capabilities.
• Drive execution across workstreams, proactively identifying dependencies, mitigating risks, and maintaining alignment on schedules and critical path items.
• Provide clear, concise updates and strategic recommendations to leadership and executive stakeholders to align on technical priorities and decisions.
• Ensure production readiness and quality standards for hardware/software deployments, supporting large-scale developer adoption and workload efficiency.

Google is proud to be an equal opportunity and affirmative action employer. We are committed to building a workforce that is representative of the users we serve, creating a culture of belonging, and providing an equal employment opportunity regardless of race, creed, color, religion, gender, sexual orientation, gender identity/expression, national origin, disability, age, genetic information, veteran status, marital status, pregnancy or related condition (including breastfeeding), expecting or parents-to-be, criminal histories consistent with legal requirements, or any other basis protected by law. See also Google's EEO Policy , Know your rights: workplace discrimination is illegal , Belonging at Google , and How we hire .

Google is a global company and, in order to facilitate efficient collaboration and communication globally, English proficiency is a requirement for all roles unless stated otherwise in the job posting.

To all recruitment agencies: Google does not accept agency resumes. Please do not forward resumes to our jobs alias, Google employees, or any other organization location. Google is not responsible for any fees related to unsolicited resumes.

Equity is granted exclusively and discretionarily by Alphabet Inc. on the basis of an agreement concluded between you and Alphabet Inc. Alphabet Inc. is your sole contractual partner with respect to equity grants. GSU grants are not guaranteed, are discretionary, are subject to approval by the Alphabet Inc. board of directors or its delegate, the terms of the relevant Alphabet Inc. stock plan, and your grant agreement. They have no impact on statutory payments. Current or past grants do not confer an acquired right.