Senior Software Engineer, AI/ML System Infrastructure

Google Inc.

Kirkland, Sunnyvale (WA, CA)

On-site

USD 174,000 - 252,000

Full time

14 days+

Get more replies from employers

Send a job-specific resume in minutes.

Benefits offered by this job

Health insurance
Retirement plan
Paid time off
Maternity leave
Baby bonding leave
Holidays

Job summary

Google seeks a Senior Software Engineer to drive TPU system control plane development and health management using hardware telemetry. You will build analytics to detect hardware issues and craft health rules for ICI, OCS, EMD, Tross, and related failure modes, integrating with TPU clusters and cloud infrastructure.

You will contribute to Paced project delivery, CI/CD, and health monitoring, collaborating across teams while supporting growth and innovation in AI/ML system infrastructure.

Qualifications

  • Bachelor's degree or equivalent practical experience.
  • 5 years of experience working with Go.
  • 3 years of experience with developing large-scale infrastructure, distributed systems or networks, or experience with compute technologies, storage or hardware architecture.
  • 3 years of experience in distributed computing.
  • 3 years of experience in infrastructure design.

Responsibilities

  • Build and develop systems to integrate in continuous running of TPU AI infrastructure.
  • Work cross-functionally to define the requirements for Diagnoser, and how the customer will benefit from it and define the Critical User Journeys (CUJs).
  • Influence and align the cross-functional teams on roadmap of PodCare and Diagnoser.
  • Deliver high quality code and timely project while working with cross-functional teams.
  • Contribute to CI/CD pipeline and integration and regression test for health monitoring system.

Skills

Go
Distributed systems
System architecture
Data structures & algorithms
C/C++

Education

Bachelor's degree or equivalent practical experience
Master's degree or PhD in Computer Science or related field

Tools

Tensor Processing Unit (TPU)

Job description

Senior Software Engineer, AI/ML System Infrastructure
Mid

Experience driving progress, solving problems, and mentoring more junior team members; deeper expertise and applied knowledge within relevant area.

X In accordance with Washington state law, we are highlighting our comprehensive benefits package, which is available to all eligible US based employees. Benefits for this role include:

  • Health, dental, vision, life, disability insurance
  • Retirement Benefits: 401(k) with company match
  • Paid Time Off: 20 days of vacation per year, accruing at a rate of 6.15 hours per pay period for the first five years of employment
  • Sick Time: 40 hours/year (increased to 69 hours/year for Seattle) including 5 discretionary sick days per instance
  • Maternity Leave (Short-Term Disability + Baby Bonding): 28-30 weeks
  • Baby Bonding Leave: 18 weeks
  • Holidays: 13 paid days per year

Note: By applying to this position you will have an opportunity to share your preferred working location from the following: Sunnyvale, CA, USA; Kirkland, WA, USA.

  • Bachelor's degree or equivalent practical experience.
  • 5 years of experience working with Go.
  • 3 years of experience with developing large-scale infrastructure, distributed systems or networks, or experience with compute technologies, storage or hardware architecture.
  • 3 years of experience in distributed computing.
  • 3 years of experience in infrastructure design.

3 years of experience in system architecture.

Preferred qualifications:
  • Master's degree or PhD in Computer Science or related technical field.
  • 5 years of experience with data structures and algorithms.
  • 5 years of experience with C and C++.
  • 1 year of experience in a technical leadership role.
About the job

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.

As the Senior Software Engineer, you will drive software development for Tensor Processing Unit (TPU) system control planes. You will design and implement health management systems that rely on hardware telemetry. You will build analytics for detecting hardware problems and develop different health rules specializing in detecting ICI, OCS, EMD, Tross, and out of band entities failure modes. You will build algorithms to generate correlated failures and suggest actions for repair workflows, integrating with both TPU cluster and cloud infrastructure. You will also build the Diagnoser for the specialized health rules to maintain and operate these TPU clusters.

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: $174000 - $252000 (USD) + 15% bonus target + equity + benefits

  • Build and develop systems to integrate in continuous running of TPU AI infrastructure.
  • Work cross-functionally to define the requirements for Diagnoser, and how the customer will benefit from it and define the Critical User Journeys (CUJs).
  • Influence and align the cross-functional teams on roadmap of PodCare and Diagnoser.
  • Deliver high quality code and timely project while working with cross-functional teams.
  • Contribute to CI/CD pipeline and integration and regression test for health monitoring system.

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.

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.

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

Similar jobs worth comparing

Senior Software Engineer, AI/ML System Infrastructure
Senior Software Engineer, AI/ML System Infrastructure

Google • United States

On-site
USD 174,000 - 252,000
Health insurance
Dental insurance
Vision insurance
+8
Senior Software Engineer, AI/ML System Infrastructure
Senior Software Engineer, AI/ML System Infrastructure

Google • Sunnyvale (CA)

On-site
USD 174,000 - 252,000
Health insurance
Dental insurance
Vision insurance
+6
Senior Software Engineer, AI/ML System Infrastructure
Senior Software Engineer, AI/ML System Infrastructure

Google • Kirkland (WA)

On-site
USD 174,000 - 252,000
Health insurance
Retirement benefits
Paid time off (PTO)
+4
Senior Software Engineer, AI/TPU Infrastructure Firmware, On-prem
Senior Software Engineer, AI/TPU Infrastructure Firmware, On-prem

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

On-site
USD 174,000 - 252,000
Health insurance
Dental insurance
Vision insurance
+8
Tech Lead, TPU AI Infrastructure
Tech Lead, TPU AI Infrastructure

Google Inc. • Sunnyvale (CA)

On-site
USD 207,000 - 300,000
Equity
Benefits
Staff Software Engineer, AI/ML Infrastructure, TPU Supercomputers
Staff Software Engineer, AI/ML Infrastructure, TPU Supercomputers

Google • Sunnyvale (CA)

On-site
USD 207,000 - 301,000
20% bonus target
Equity
Comprehensive benefits package
Staff Software Engineer, ML Systems Co-Design
Staff Software Engineer, ML Systems Co-Design

Google • United States

On-site
USD 207,000 - 300,000
Health, dental, vision, life, and/ or
401(k) with company match
PTO 20 days per year
+4
Senior Hardware Engineering Manager, Emergent AI Infrastructure
Senior Hardware Engineering Manager, Emergent AI Infrastructure

Google • United States

On-site
USD 236,000 - 329,000
Health insurance
401(k) with match
Paid time off
+1
Software Engineer III, Infrastructure, Google Cloud AI
Software Engineer III, Infrastructure, Google Cloud AI

Google • United States

On-site
USD 147,000 - 210,000
Health insurance
Dental insurance
Vision insurance
+8
Staff Software Engineer, ML Systems Co-Design
Staff Software Engineer, ML Systems Co-Design

Google Inc. • Sunnyvale (CA)

On-site
USD 207,000 - 300,000
Health, dental, vision, life, and(dis)
401(k) with company match
Paid Time Off: 20 days/year
+4