Apply on Employer Site

Google · 16 hours ago

Staff Software Engineer, AI/ML Infrastructure

Sunnyvale, CA

Full-time

Hybrid

Lead/Staff

$197K/yr - $291K/yr

8+ years exp

Google is a leading technology company that develops next-generation technologies impacting billions of users. The role involves working on integrated AI infrastructure systems, collaborating with cross-functional leaders, and delivering high-quality projects that enhance Google's ML infrastructure.

AppsArtificial Intelligence (AI)Cloud StorageSearch EngineSEO

Growth Opportunities

H1B Sponsor Likely

Responsibilities

Partner with TPU hardware and software team and help design, develop and validate TPU based systems for standalone deployment

Build and integrate Cloud Compute software to bootstrap TPU AI Infrastructure

Drive high-quality, timely delivery of projects within your team

Collaborate with cross-functional leaders to shape the future of our ML infrastructure

Provide technical guidance and mentorship, fostering the growth of team members

Qualification

ML infrastructureSoftware designArchitectureC++Speech/audio technologyReinforcement learningData structuresAlgorithmsAccelerators (TPU/GPU)Cloud computeTeam collaborationMentorship

Required

Bachelor's degree or equivalent practical experience

8 years of experience in software development

5 years of experience testing, and launching software products, and 3 years of experience with software design and architecture

5 years of experience with one or more of the following: Speech/audio (e.g., technology duplicating and responding to the human voice), reinforcement learning (e.g., sequential decision making), ML infrastructure, or specialization in another ML field

5 years of experience with ML design and ML infrastructure (e.g., model deployment, model evaluation, data processing, debugging, fine tuning)

Preferred

Master's degree or PhD in Engineering, Computer Science, or a related technical field

8 years of experience with data structures and algorithms

3 years of experience working in a complex, matrixed organization involving cross-functional, or cross-business projects

Experience in software development, with proficiency in languages such as C++

Experience in Accelerators (i.e., TPU or GPU) Software/Hardware co-design engineering domain

Experience in machine learning infrastructure, Stock-Keeping Units (SKUs)/Bill of Materials (BOM) design, cloud compute, networking, storage technologies

Benefits

Bonus

Equity

Benefits

Company

Google

Glassdoor4.4

Google specializes in internet-related services and products, including search, advertising, and software. It is a sub-organization of Alphabet.

Founded in 1998

Mountain View, California, USA

10001+ employees

https://www.google.com

H1B Sponsorship

Google has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)

Distribution of Different Job Fields Receiving Sponsorship

Represents job field similar to this job

Trends of Total Sponsorships

2025 (8763)

2024 (8872)

2023 (9682)

2022 (11626)

2021 (9109)

2020 (9785)

Funding

Current Stage

Public Company

Total Funding

$26.1M

Key Investors

Andy Bechtolsheim

2004-08-19IPO

1999-06-07Series Unknown· $25M

1998-11-01Angel· $1M

Leadership Team

Sundar Pichai

CEO

Thomas Kurian

CEO - Google Cloud

Recent News

The Indian Express

Who is Ragini Das? Rejected by Google 12 years ago, she now leads its startup arm in India

2025-10-08

Tech.eu

Profits up at Google UK, but nearly 400 jobs cut

2025-10-04