Google · 4 hours ago
Senior Staff Software Engineer, Performance Co-Design Simulator Architect
Google is a leading technology company that develops next-generation technologies for billions of users. They are seeking a Senior Staff Software Engineer to manage project priorities, develop software solutions, and enable hardware-software co-design for optimized machine learning systems.
AppsArtificial Intelligence (AI)Cloud StorageSearch EngineSEO
Responsibilities
Establish and maintain cross-organizational alignment on TPU hardware, TPU compiler, TPU systems hardware, software co-design simulator compiler technology for state-of-the-art tech stack, velocity, quality of results and verifiability
Develop compiler infrastructure for co-design simulator technologies to continuously keep up with evolving new system architecture choices and new ML workloads as well as supporting simulations at different abstraction levels
Explore and define future ML accelerator system and chip architecture with objective and data-driven ground truth while strategizing collaboration across partner organizations to maximize adoption and impact
Enable the cost effective peak performance of future ML systems with full-stack ML hardware, software co-design
Establish understanding of the latest business-critical ML models (large-language models, large embedding models etc.) to inform optimizations of model architecture, software systems and hardware architecture
Qualification
Required
Bachelor's degree or equivalent practical experience
8 years of experience in software development
7 years of experience with performance, large-scale systems data analysis, visualization tools, or debugging
5 years of experience with design and architecture; and testing/launching software products
Preferred
Master's degree or PhD in Engineering, Computer Science, or a related technical field
Experience in constructing custom IR dialects and leveraging open-source compiler frameworks (MLIR, XLA) to solve mapping problems
Experience architecting custom lowering, transformations, and legalization passes specifically designed for rapid architecture exploration
Knowledge of computer architecture, TPU, LLM or other accelerators, while focusing on ML performance modeling and optimizations
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.
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
Recent News
Small Business Trends
2026-01-24
2026-01-24
Search Engine Journal
2026-01-24
Company data provided by crunchbase