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

Google · 21 hours ago

Staff Software Engineer, AI/ML Infrastructure

Google is a leading technology company that develops next-generation technologies to change how users connect and interact with information. They are seeking a Staff Software Engineer to work on AI infrastructure systems, collaborating with hardware and software teams to design and validate TPU-based systems and integrate Cloud Compute software.

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

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

Software developmentMachine learning infrastructureC++Data structuresAlgorithmsSpeech/audio technologyReinforcement learningTPU/GPU co-designCloud computeNetworking technologiesStorage technologiesSoftware designArchitectureModel deploymentModel evaluationData processingDebuggingFine tuningEnthusiasm for new problemsCross-functional collaborationMentorshipLeadership qualities

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

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