Google · 7 hours ago
Staff Software Engineer, AI/ML, Data Infrastructure, Workspace
Sunnyvale, CA
Full-time
Onsite
Lead/Staff
$197K/yr - $291K/yr
8+ years expGoogle is a leading technology company that develops next-generation technologies impacting billions of users. The Staff Software Engineer will manage project priorities and deliverables, designing, developing, testing, and maintaining software solutions, particularly focusing on machine learning infrastructure to enhance Google Workspace products.
Artificial Intelligence (AI)Cloud ComputingAppsMarketingCloud StorageSearch EngineSEO
Responsibilities
Define and drive the implementation of technical roadmap and best practices for ML infrastructure within WASP. Build and support infrastructure for large-scale data ingestion and transformation for ML model consumption. Design, build, test, and maintain scalable, reliable, and efficient infrastructure and platforms for the machine learning life-cycle
Develop and improve ML pipelines, feature engineering and storage systems, and model management tools. Enhance and integrate with Google's ML platforms and technologies underlying data processing frameworks like Flume. Ensure the performance, reliability, and resource efficiency of ML training and serving systems
Develop frameworks and tools to improve ML model development velocity, quality, and maintainability
Collaborate with ML engineers, product engineers, and Site Reliability Engineers to deliver anti-abuse solutions
Troubleshoot and resolve issues in large-scale ML production environments
Qualification
Required
Bachelor's degree or equivalent practical experience
8 years of experience in software development
5 years of experience leading ML design and optimizing ML infrastructure (e.g., model deployment, model evaluation, data processing, debugging, fine tuning)
5 years of experience with one or more of the following: ML infrastructure or specialization in another ML field
5 years of experience testing, and launching software products, and 3 years of experience with software design and architecture
Experience in distributed and large-scale databases used for ML and AI training and evaluation etc
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
3 years of experience in a technical leadership role leading project teams, setting technical direction, and designing and architecting software solutions
Benefits
Bonus
Equity
Benefits
Company
Google specializes in internet-related services and products, including search, advertising, and software. It is a sub-organization of Alphabet.
10001+ employees
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
Kleiner Perkins,Sequoia CapitalAndy 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
Small Business Trends
2026-01-24
2026-01-24
Search Engine Journal
2026-01-24
Company data provided by crunchbase