Sign up to access all features of our service.
  • Job search
  • Favorites
  • Create a CV
    New
  • Salaries
  • Subscriptions

Tech Lead, TPU AI Infrastructure

$207k - $300k

Google

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.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.

Vacancy posted 1 day ago
Similar jobs that could be interesting for youBased on the Tech Lead, TPU AI Infrastructure in Sunnyvale, CA vacancy
  • $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... 
    Suggested
    Worldwide

    Google

    Sunnyvale, CA
    3 days ago
  • $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... 
    Suggested
    Worldwide

    Google

    Sunnyvale, CA
    1 day ago
  • $207k - $301k

     ...Software Engineer Manager, TPU Silicon Validation Google, Sunnyvale, CA, USA...  ...Software Engineering Manager, you will lead a group focused on building the software...  ...edge chip and hardware systems. The AI and Infrastructure team is redefining what’s possible. We... 
    Suggested
    Worldwide

    Google Inc.

    Sunnyvale, CA
    2 days ago
  • $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,... 
    Suggested
    Shift work

    Google

    Sunnyvale, CA
    1 day ago
  • $207k - $300k

    Focus on Tensor Processing Unit (TPU) fleet efficiency analysis and performance optimization...  ...in a technical leadership role leading project teams and setting technical...  ...Gemini and other open-source models.The AI and Infrastructure team is redefining what’s possible. We... 
    Suggested
    Worldwide

    Google

    Sunnyvale, CA
    17 hours ago
  •  ...are the most immediate and profound opportunity for AI to drive positive change in the physical world. Safer...  ..., Softbank, Fidelity, T. Rowe Price, and other leading investorsAbout the RoleEvaluation Infrastructure plays a critical role at Nuro, directly enabling L4... 
    Temporary work
    Immediate start
    Flexible hours

    Nuro

    Mountain View, CA
    3 days ago
  • $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... 

    Google

    Sunnyvale, CA
    2 days ago
  • $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

    Google

    Sunnyvale, CA
    4 days ago
  • $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

    Google

    Sunnyvale, CA
    1 day ago
  • $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

    Google

    Sunnyvale, CA
    21 hours ago
  • $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

    Google

    Sunnyvale, CA
    21 hours ago
  •  ...Mind Mind Robotics is building Physical AI for real-world industrial deployment, starting...  ..., reliably and safely, depends on the infrastructure that ships software to them, validates...  ...it's out there. We're looking for a Tech Lead to own fleet deployment, testing infrastructure... 
    Remote work

    Garuda Ventures

    Palo Alto, CA
    1 day ago
  • $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

    Google

    Sunnyvale, CA
    1 day ago
  • $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

    Google

    Sunnyvale, CA
    4 days ago
  • $193.93k - $291.15k

     ...and profound opportunity for AI to drive positive change in the...  ..., T. Rowe Price, and other leading investors.About the RoleAs the...  ...ownership of the networking infrastructure, telemetry, and routing that...  ...track record of acting as a Tech Lead on complex software projects... 
    Immediate start
    Remote work
    Flexible hours

    Nuro

    Mountain View, CA
    1 day ago
  • $183k - $275k

     ...immediate and profound opportunity for AI to drive positive change in the physical...  ...Softbank, Fidelity, T. Rowe Price, and other leading investorsAbout the RoleNuro leverages...  ...it reaches the road. You will own the infrastructure that makes this possible.Our Performance... 
    Temporary work
    Immediate start
    Flexible hours

    Nuro

    Mountain View, CA
    4 days ago
  • $160.36k - $240.54k

     ...immediate and profound opportunity for AI to drive positive change in the physical...  ...Softbank, Fidelity, T. Rowe Price, and other leading investorsAbout the RoleNuro is seeking a...  ...Engineer with expertise in large-scale infrastructure, workload orchestration, and data... 
    Immediate start
    Flexible hours

    Nuro

    Mountain View, CA
    4 days ago
  • $168k - $264.5k

     ...amazing people. Today, we’re tapping into the unlimited potential of AI to define the next era of computing. An era in which our GPU...  ...you can make a lasting impact on the world.Join NVIDIA’s CAD Infrastructure team and be part of something outstanding! We are building a... 
    Full time

    Nvidia

    Santa Clara, CA
    1 day ago
  • $160.36k - $240.54k

     ...immediate and profound opportunity for AI to drive positive change in the physical...  ...Softbank, Fidelity, T. Rowe Price, and other leading investors.About the RoleOur software...  ...Data Platform, Simulation, and Technical Infrastructure.Data Platform: The Data Platform serves... 
    Immediate start
    Flexible hours

    Nuro

    Mountain View, CA
    1 day ago
  • $180k - $215k

     ...has developed an artificial intelligence (AI) powered technology stack purpose-built...  ...for a Senior Software Engineer to build infrastructure, tools, and systems that support our Planning...  ...used across the Planning team.Lead infrastructure projects with broad team-wide... 
    Visa sponsorship

    Kodiak Robotics

    Mountain View, CA
    1 day ago
  • $193.93k - $352.29k

     ...immediate and profound opportunity for AI to drive positive change in the physical...  ...Softbank, Fidelity, T. Rowe Price, and other leading investors.About the TeamFrontier models...  ...a month. What is not fungible is the infrastructure that decides whether an autonomous system... 
    Immediate start
    Flexible hours

    Nuro

    Mountain View, CA
    3 days ago
  • $160.36k - $240.54k

     ...are the most immediate and profound opportunity for AI to drive positive change in the physical world....  ...Google, Softbank, Fidelity, T. Rowe Price, and other leading investors.About the RoleThe Autonomy ML Infrastructure team is responsible for building & improving the core... 
    Work experience placement
    Immediate start
    Flexible hours

    Nuro

    Mountain View, CA
    4 days ago
  • $215k - $352k

     ...based in Mountain View, CA.The Enterprise Infrastructure Engineering (EIE) team builds and...  ...supporting internal applications like LangSmith, AI integrations, and enterprise data...  ...high compliance across global interfaces.Lead Azure governance for tenants, subscriptions... 
    For contractors
    Work at office
    Remote work
    Flexible hours

    Linkedin

    Mountain View, CA
    2 days ago
  • $146.3k - $306.4k

    Job Description Mentors teams and leads the architecture of highly scalable, interdependent...  ...security vulnerabilities. Ensure cloud infrastructure is in compliance with industry standards...  ...to life‑saving care. And with AI embedded across our products and services... 
    Temporary work
    Flexible hours

    Oracle

    Santa Clara, CA
    4 days ago
  • $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

    Google

    Sunnyvale, CA
    2 days ago
  • $276.1k - $350k

    We are looking for a Senior Principal Tech Lead Manager for AWS Transform, a frontier agent for migration...  ...drive the technical evolution of autonomous AI agent systems that modernize mission-critical enterprise infrastructure and very large codebases. We help our... 
    Local area
    Flexible hours

    Amazon

    Santa Clara, CA
    3 days ago
  • $258k - $387k

     ...immediate and profound opportunity for AI to drive positive change in the physical...  ...Softbank, Fidelity, T. Rowe Price, and other leading investors.About the RoleAs a Principal...  ...technical direction of Nuro’s onboard infrastructure. We are looking for a technical leader who... 
    Immediate start
    Flexible hours

    Nuro

    Mountain View, CA
    4 days ago
  • $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 work
    Worldwide

    Google

    Sunnyvale, CA
    3 days ago
  • $175k - $287k

     ...LinkedIn operates one of the largest privately managed compute infrastructures in the world outside the public cloud providers. As a Staff...  ...data and analytics pipelines processing petabytes daily, and the AI/ML infrastructure driving recommendations, search relevance,... 
    For contractors
    Work experience placement
    Work at office
    Flexible hours

    Linkedin

    Mountain View, CA
    3 days ago
  • $198k - $326k

     ...improve machine learning models at scale. Our AI systems power recommendations, search,...  ...team engineers robust, highly scalable infrastructure that delivers continuous, real-time insights...  ...and production environments.Lead the diagnosis and resolution of complex,... 
    For contractors
    Work at office
    Flexible hours
    Shift work

    Linkedin

    Sunnyvale, CA
    21 hours ago

Do you want to receive more vacancies?

Subscribe and receive similar vacancies to Tech Lead, TPU AI Infrastructure. Be the first to apply!