Staff Software Engineer, AI/ML jobs in United States
cer-icon
Apply on Employer Site
company-logo

Google · 1 day ago

Staff Software Engineer, AI/ML

Google is a leading technology company seeking a Staff Software Engineer specializing in AI/ML. In this role, you will build infrastructure for large-scale AI, collaborate with teams to develop TPU-based systems, and address engineering challenges while working closely with customers.

AppsArtificial Intelligence (AI)Cloud StorageSearch EngineSEO
check
Growth Opportunities
check
H1B Sponsor Likelynote

Responsibilities

Build the infrastructure and automation that makes large-scale AI possible
Partner with TPU Software and Hardware team and help design, develop and validate TPU based systems for standalone deployment. Build and integrate Cloud Compute software to bootstrap TPU AI Infrastructure
Solve unique engineering challenges associated with managing both Google-networked and fully air-gapped hardware environments
Work with customers, triage, debug, and resolve system issues encountered during customer trial

Qualification

C++Software testingLarge-scale infrastructureCloud compute platformsData structuresSoftware designCross-functional projectsTechnical leadership

Required

Bachelor's degree or equivalent practical experience
8 years of experience programming in C++
5 years of experience testing, and launching software products
5 years of experience building and developing large-scale infrastructure, distributed systems or networks, or experience with compute technologies, storage, or hardware architecture
3 years of experience with software design and architecture
Experience with Cloud compute platforms like Kubernetes, Cloud Functions

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 in a technical leadership role leading project teams and setting technical direction
3 years of experience working in a complex, matrixed organization involving cross-functional, or cross-business projects

Benefits

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 (statutory, where applicable); 5 days/event (discretionary)
Maternity Leave (Short-Term Disability + Baby Bonding): 28-30 weeks
Baby Bonding Leave: 18 weeks
Holidays: 13 paid days per year

Company

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

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

leader-logo
Sundar Pichai
CEO
linkedin
leader-logo
Thomas Kurian
CEO - Google Cloud
linkedin
Company data provided by crunchbase