Google · 7 hours ago
Staff Software Engineer, Machine Learning Infrastructure
Mountain View, CA, USA
Full-time
Onsite
Lead/Staff
$197K/yr - $291K/yr
8+ years expGoogle is a leading technology company that develops next-generation technologies to enhance user interaction with information. They are seeking a Staff Software Engineer to work on critical projects for Google Cloud, focusing on machine learning infrastructure and performance optimization.
AppsArtificial Intelligence (AI)Cloud StorageSearch EngineSEO
Responsibilities
Design and implement new quantization features to keep pace with the rapidly evolving JAX ecosystem
Research and implement experimental quantization algorithms to lead customer adoption rather than just reacting to requests
Debug and optimize low-level performance issues. Use XLA profiling tools to analyze TPU/GPU execution traces, identify bottlenecks, and ensure the lowest-cost implementation of algorithms
Collaborate with the multiple performance teams to align roadmaps and resolve upstream compiler issues that affect quantization quality
Write high-quality documentation, tutorials, and examples for the open-source community to lower the barrier to entry for new users
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
Experience with developing and applying quantization algorithms
Experience with performance analysis and optimization
Experience with JAX
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
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
Small Business Trends
2026-01-24
2026-01-24
Search Engine Journal
2026-01-24
Company data provided by crunchbase