Tech Lead, TPU AI Infrastructure
$207k - $300kDefine 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.Minimum qualifications:Bachelor's degree or equivalent practical experience.8 years of experience with software development in one or more programming languages (e.g., Python, C, C++, Java, JavaScript).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.Preferred qualifications:Experience with Temporal orchestration workflows or similar workflow engines.Expertise in Kubernetes, ArgoCD, and container orchestration.Familiarity with hardware/software integration, hardware lifecycle, or bare-metal provisioning (PXE).Excellent communication skills with the ability to manage stakeholder relationships and drive cross-functional alignment.Google's software engineers develop the next-generation technologies that change how billions of users connect, explore, and interact with information and one another. Our products need to handle information at massive scale, and extend well beyond web search. We're looking for engineers who bring fresh ideas from all areas, including information retrieval, distributed computing, large-scale system design, networking and data storage, security, artificial intelligence, natural language processing, UI design and mobile; the list goes on and is growing every day. As a software engineer, you will work on a specific project critical to Google’s needs with opportunities to switch teams and projects as you and our fast-paced business grow and evolve. We need our engineers to be versatile, display leadership qualities and be enthusiastic to take on new problems across the full-stack as we continue to push technology forward.Our team is building the future of at-scale computing by delivering a groundbreaking, on-premises AI supercomputer—the next evolution of Google's AI accelerator hardware. This is a rare opportunity to join a specialized team creating novel solutions for customers with sophisticated security and data locality requirements. You will be at the heart of building the next generation of AI hardware and the critical software that brings it to life, making a Google-scale impact on the future of AI.We manage complex infrastructure, integrating with Kubernetes, Temporal, and custom orchestration tools to ensure our compute nodes meet rigorous health standards. We are evolving our infrastructure toward automated node repair, self-healing frameworks, and advanced fleet-wide observability to reduce manual operational overhead and build robust platform boundaries.Google Cloud accelerates every organization’s ability to digitally transform its business and industry. We deliver enterprise-grade solutions that leverage Google’s cutting-edge technology, and tools that help developers build more sustainably. Customers in more than 200 countries and territories turn to Google Cloud as their trusted partner to enable growth and solve their most critical business problems.Individual pay is determined by factors including job-related skills, experience, and relevant education or training. US: $207000 - $300000 (USD) + 20% bonus target + equity + benefitsLearn more about benefits at Google.Bachelor's degree or equivalent practical experience.8 years of experience with software development in one or more programming languages (e.g., Python, C, C++, Java, JavaScript).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.
$207k - $300k
Lead the technical strategy and architecture of next-generation control planes and software infrastructure for highly-available ML network fabrics.Design and implement agentic SDLC, leveraging... ...and hardware across the full stack.The AI and Infrastructure team is redefining...SuggestedWorldwide$207k - $300k
Lead and manage the team responsible for TPU development and Silicon Validation by setting clear priorities, providing coaching and performance feedback... ...our cutting-edge chip and hardware systems.The AI and Infrastructure team is redefining what’s possible. We empower...SuggestedWorldwide$207k - $300k
...optimizing Generative AI performance across heterogeneous... ...decision making), ML infrastructure, or specialization in... ....Track record of leading and delivering ML projects... ...(e.g., GPU / Pixel TPU / NPUs / CPU) on Android... ...mission of bringing AI to tech, healthcare, finance,...SuggestedShift work$134k - $265k
Position Summary Domain Tech Lead — Infrastructure Patching Automation (Mainframe & AS/400)Join our AI & Engineering team by transforming technology platforms, driving innovation, and helping make a significant impact on our clients' success. You’ll collaborate...SuggestedLocal area$174k - $252k
....3 years of experience with developing large-scale infrastructure, distributed systems or networks, or experience with... ...infrastructure. Contribute to projects enabling AI networking and high-performance fabrics for TPU and GPU workloads.Our Platforms Infrastructure Engineering...Suggested$192k - $278k
...impact on hardware performance.Lead the development of... ...ll work to shape the future of AI/ML hardware acceleration. You... ...opportunity to drive cutting-edge TPU (Tensor Processing Unit) technology... ...efficient accelerators.The AI and Infrastructure team is redefining what’s...Worldwide$116k - $166k
...for digital designs within the TPU.Write performant, and power-... ...ll work to shape the future of AI/ML hardware acceleration. You... ...collaborative environment.The AI and Infrastructure team is redefining what’s... ...shaping the future of world-leading hyperscale computing, with key...Worldwide$163k - $236k
Lead advanced SoC integration projects from concept to final delivery... ...work to shape the future of AI/ML hardware acceleration. You... ...have an opportunity to drive TPU (Tensor Processing Unit) technology... ...project schedules.The AI and Infrastructure team is redefining what’s...Worldwide$138k - $197k
...ll work to shape the future of AI/ML hardware acceleration. You... ...opportunity to drive cutting-edge TPU (Tensor Processing Unit)... ...driven systems.As an SoC Test Infrastructure Engineer, you will drive the post... ...shaping the future of world-leading hyperscale computing, with key...Worldwide$147k - $210k
...system development code.Participate in, or lead design reviews with peers and... ...the architecture built by the Technical Infrastructure team to keep it running. From developing... ...best and fastest experience possible.The AI and Infrastructure team is redefining what...Worldwide$147k - $210k
...or more specialized ML areas, utilize ML infrastructure, and contribute to model optimization... ...continue to push technology forward.The AI and Infrastructure team is redefining what... ...our teams are shaping the future of world-leading hyperscale computing, with key teams...Worldwide$240k - $333k
...flows and "Layout-Aware" design.Lead full custom digital circuit... ...and maximize yield for massive TPU-scale chips.Minimum qualifications... ...goals for the intersection of AI-driven layout and traditional... ...processes.The AI and Infrastructure team is redefining what’s possible...Worldwide$207k - $301k
Lead, mentor, and develop a team of high-performing engineers who design and model the... ...goals, strategy, and roadmap for modeling infrastructure, exercising strong technical judgment.Actively... ...evolving demands of both networking and AI/ML landscapes. Effectively working with...Remote workWorldwide$207k - $300k
...experience building and developing large-scale infrastructure, distributed systems or networks, or... ...in a technical leadership role leading project teams and setting technical direction... ...continue to push technology forward.The AI and Infrastructure team is redefining what...Worldwide$147k - $210k
...system development code.Participate in, or lead design reviews with peers and... ...Design and implement changes to make Cloud AI products more enterprise-ready (customer... ...of experience with developing large-scale infrastructure, distributed systems or networks, or experience...Worldwide$207k - $301k
...product ordering and platform engineering.Lead complex, cross-functional projects from... ...experience building and developing large-scale infrastructure, distributed systems, or large-scale... ...network infrastructure.As we scale into the AI era, demand financial analysis (FA) is...Worldwide$174k - $252k
...system development code.Participate in, or lead design reviews with peers and... ...years of experience developing large-scale infrastructure, distributed systems or networks, or experience... ...maintain, and enhance software solutions.The AI and Infrastructure team is redefining...Worldwide$240k - $333k
Lead, mentor and manage a team of RTL Design and DV Engineers developing DRAM subsystems... ...integration.As Technical Manager for TPU DRAM, you will develop and validate... ...manufacturers and third party IP providers.The AI and Infrastructure team is redefining what’s possible. We...Worldwide$207k - $301k
...experience building and developing large-scale infrastructure, distributed systems or networks, or... ...in a technical leadership role leading project teams and setting technical direction... ...maintain, and enhance software solutions.The AI and Infrastructure team works on the...Worldwide$207k - $301k
..., open-ended problems, designing robust infrastructure solutions that reduce operational complexity... ...in a technical leadership role leading project teams and setting technical direction... ...context of implementing agentic workflows or AI transformations.Google's software...$207k - $300k
...experience building and developing large-scale infrastructure, distributed systems or networks, or... ...in a technical leadership role leading project teams and setting technical direction... ...maintain, and enhance software solutions.The AI and Infrastructure team is redefining...Worldwide$207k - $300k
...maintain and improve switch software.Work on projects enabling AI networking infrastructure, enabling high performing fabrics for Tensor Processing... ...to hardware our teams are shaping the future of world-leading hyperscale computing, with key teams working on the development...Worldwide$174k - $252k
...experience building and developing large-scale infrastructure or distributed systems.1 year of... ...push technology forward.The Google Cloud AI Research team addresses AI challenges motivated... ...Google Cloud’s mission of bringing AI to tech, healthcare, finance, retail and many...$147k - $210k
...system development code.Participate in, or lead design reviews with peers and... ...of experience with developing large-scale infrastructure, distributed systems or networks, or experience... ...platforms or frameworks.Experience with AI Algorithms.Preferred qualifications:Master...$240k - $333k
...high-frequency jitter profiles.Lead the debugging of RTL and... ...methodologies, and roadmaps for AI accelerators, High-Performance... ...opportunity to drive cutting-edge TPU (Tensor Processing Unit)... ...scale AI data centers.The AI and Infrastructure team is redefining what’s possible...WorldwideShift work$262k - $364k
Technical Leadership and Mentorship: Lead and coach a distributed engineering team, fostering... ...roadmaps bridging RDMA, storage, and AI/ML, staying ahead of training and... ...with Storage Systems or Machine Learning Infrastructure.Google's software engineers develop the next...Remote workWorldwide$262k - $365k
...products.Experience integrating generative AI tools or LLM interfaces into workflows.... ...in a technical leadership role leading project teams and setting technical direction... ...Network traffic engineering is a critical infrastructure service for Google. In this role, you will...Worldwide$159k - $231k
...correlation studies, and root-cause analyses. Lead the triage and resolution of power... ...Power Integrity Engineer within Platforms Infrastructure Engineering, you will play a pivotal role... ...will directly shape our next generation AI hardware for AL/ML, server, networking, and...Worldwide$126k - $180k
...scale, multi-node environments.Experience using or integrating AI/LLMs (such as Gemini) into daily engineering or troubleshooting... ...security, reliability and scalability, running the full stack from infrastructure to applications to devices and hardware. Our teams are...Internship$174k - $252k
...specialized ML areas, leverage ML infrastructure, and demonstrate expertise in... ...LiteRT, Google's on-device AI framework for first- and third... ...various accelerators (GPU/Pixel TPU/NPUs/CPU) on Android, Chrome,... ...qualifications:Experience leading and delivering successful ML projects...
Do you want to receive more vacancies?
Subscribe and receive similar vacancies to Tech Lead, TPU AI Infrastructure. Be the first to apply!
- design technology Sunnyvale, TX
- manufacturing engineering tech Sunnyvale, TX
- electronics engineering technology Sunnyvale, TX
- technology risk Sunnyvale, TX
- patient tech Sunnyvale, TX
- process technology Sunnyvale, TX
- part time remote tech Sunnyvale, TX
- technology solutions professional Sunnyvale, TX
- mobility tech Sunnyvale, TX
- data tech Sunnyvale, TX
