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
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
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 CompanyTotal Funding
$26.1MKey Investors
Andy Bechtolsheim
2004-08-19IPO
1999-06-07Series Unknown· $25M
1998-11-01Angel· $1M
Recent News
Small Business Trends
2026-01-24
2026-01-24
Search Engine Journal
2026-01-24
Company data provided by crunchbase