Senior Staff Software Engineer, Performance Co-Design Simulator Architect jobs in United States
cer-icon
Apply on Employer Site
company-logo

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
check
Growth Opportunities
check
H1B Sponsor Likelynote

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

Software developmentPerformance analysisSystem architectureML performance modelingOpen-source compiler frameworksComputer architectureVersatilityLeadership qualitiesProblem-solving

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 Company
Total Funding
$26.1M
Key Investors
Kleiner Perkins,Sequoia CapitalAndy 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