Find work in the United States.

A simpler way to find your next job.

Jobs in the United States

25+ jobs

Save this search

Added just now
Senior Engineer, Waste Treatment Design
Micron Idaho Semiconductor Manufacturing (Triton) LLC
Boise, ID
As a Waste Treatment Design Engineer, you will help shape the systems that safely manage industrial waste streams across major semiconductor construction projects. You will coordinate…
Added just now
Senior Engineer, Reclamation and ZLD
Micron Idaho Semiconductor Manufacturing (Triton) LLC
Boise, ID
As a Water Reclamation Engineer, you will lead the design and integration of systems that recover, reuse, and responsibly manage industrial water streams for major semiconductor…
Added just now
Senior Engineer, Process Mechanical Engineer
Micron Idaho Semiconductor Manufacturing (Triton) LLC
Boise, ID
As a Process Mechanical Engineer, you will provide technical leadership and oversight of third-party process mechanical design for major construction projects. You will manage design…
Added 1 min ago
UG
Full Stack Software Engineer
UnitedHealth Group
Eden Prairie, MN
- $72,800 to $130,000 USD / year
To support this mission, OSIT has initiated a multi‑year modernization program aimed at updating and enhancing enterprise technology systems in accordance with modern design standards
Added 9 mins ago
TT
Lead Product Engineer
True Talent Solutions Partners
New York, NY
- $225,000 to $275,000 USD / year
- Own and improve the frontend codebase, establishing reusable design system components and delivering new features.
Added 11 mins ago
Deputy Director
City of New York
Queens, NY
…construction management, construction technology, sustainable design, urban planning, urban studies, city planning, transportation planning, business administration, or public…
Added 12 mins ago
A
Software Dev Engineer II - AMZ27051.1
Amazon
East Palo Alto, CA
- $165,200 to $223,600 USD / year
Design, develop, implement, test, and document embedded or distributed software applications, tools, systems and services. Translate functional requirements into robust, efficient,…
Added 13 mins ago
C
Automation Engineer
CBRE
Tracy, CA
- $115,606.4 USD / year
…We’re looking for a skilled and forward-thinking Ctrl Systems Engineer / Automation Engineer (AE) to join our growing team as we work to service one of the world’s largest online…
Added 13 mins ago
Opto-Mechanical Engineer, Amazon Leo for Government
Amazon
NORTHRIDGE, CA
…customers (such as government agencies). The Opto-Mechanical Engineer will play an important role in the team delivering innovative and fast-paced design and analysis for both space and…
Added 13 mins ago
RF Engineer, Amazon Leo Government, Amazon Leo Government
Amazon
Northridge, CA
The RF/Microwave Engineer will focus on the design of the various RF and Microwave circuit blocks in the range of Ka, L, and S bands. They will be expected to lead a variety of tasks, from…
Added 13 mins ago
ES
Material Handler I
Empirical Systems Aerospace, Inc.
San Luis Obispo, CA
- $20 to $25 USD / hour
…cost-effective solutions to the industry. We specialize in cutting-edge system design, rapid product development, precision manufacturing, comprehensive testing, and technology…
Added 13 mins ago
Industrial Facility Security Officer (FSO)
Empirical Systems Aerospace, Inc.
San Luis Obispo, CA
…cost-effective solutions to the industry. We specialize in cutting-edge system design, rapid product development, precision manufacturing, comprehensive testing, and technology…
Added 13 mins ago
Deputy Supply Chain Director
Empirical Systems Aerospace, Inc.
San Luis Obispo, CA
…cost-effective solutions to the industry. We specialize in cutting-edge system design, rapid product development, precision manufacturing, comprehensive testing, and technology…
Added 13 mins ago
Additive Manufacturing Supervisor
Empirical Systems Aerospace, Inc.
San Luis Obispo, CA
…cost-effective solutions to the industry. We specialize in cutting-edge system design, rapid product development, precision manufacturing, comprehensive testing, and technology…
Added 13 mins ago
ES
Production Technician II
Empirical Systems Aerospace, Inc.
San Luis Obispo, CA
- $26 to $29 USD / hour
…cost-effective solutions to the industry. We specialize in cutting-edge system design, rapid product development, precision manufacturing, comprehensive testing, and technology…
Added 13 mins ago
M
Restraints Engineer
ManpowerGroup
Foster City, CA
- $80 to $90 USD / hour
Our client is seeking a Restraints Engineer to join their team. As a Restraints Engineer, you will be part of the Vehicle Engineering team supporting occupant protection and restraint…
Added 14 mins ago
Senior DevOps Engineer
Abbott
Pleasanton, CA
We are seeking a highly skilled and experienced Senior DevOps Engineer to design, implement, and optimize scalable DevOps solutions that support the build, release, and deployment of modern…
Added 14 mins ago
Engineer V
General Atomics
Poway, CA
Work with internal project team and customer team members to define network system design requirements.
Added 14 mins ago
Robotics Engineer, Logistics Automation
Meta
Fremont, CA
Meta is seeking a Robotics Engineer to join our Infra Robotics team. This team implements advanced robotics technologies to enable efficient and safe data center operations.The logistics…
Added 14 mins ago
Operations AI Engineer - Early Career
Lockheed Martin
Palmdale, CA
Design and implement scalable computing solutions (GPU, distributed, cloud) for large‑scale data processing.
Added 14 mins ago
Software Engineer, Machine Learning - Ads Infra and Ranking
Meta
Menlo Park, CA
Design offline evaluations and online experiments that assess ad relevance, advertiser outcomes, and system performance.
Added 14 mins ago
Spacecraft Thermal Engineer (Early Career) - Millennium Space Systems
The Boeing Company
El Segundo, CA
Thermal Engineers support the Spacecraft Development Program Managers and Chief Systems Engineer through the development of all spacecraft thermal system configurations, designs, analyses,…
Added 14 mins ago
Senior Site Installation Engineer
Trinzic
El Segundo, CA
Sr. Engineer and Subject Matter Expert to PTES PMO Government Customer for all SATCOM, Site Development, System Integration, and project acceptance documentation issues
Added 14 mins ago
Senior Staff Software Engineer, Generative Retail Ad Formats and LLM Applications
Google
Mountain View, CA
…including information retrieval, distributed computing, large-scale system design, networking and data storage, security, artificial intelligence, natural language processing, UI design and…
Added 14 mins ago
Staff Software Engineer, ML Compiler Performance Engineer Co-Design
Google
Mountain View, CA
…including information retrieval, distributed computing, large-scale system design, networking and data storage, security, artificial intelligence, natural language processing, UI design and…

Google

Staff Software Engineer, ML Compiler Performance Engineer Co-Design

Mountain View, CA, United States

Apply About employer

Applications open in a new tab.

Job details

Posting language: English
Posted: Nov 10, 2026

Original job description

Staff Software Engineer, ML Compiler Performance Engineer Co-Design

Share Staff Software Engineer, ML Compiler Performance Engineer Co-Design

corporate_fare Google place Mountain View, CA, USA

Advanced

Experience owning outcomes and decision making, solving ambiguous problems and influencing stakeholders; deep expertise in domain.

Share Staff Software Engineer, ML Compiler Performance Engineer Co-Design

info_outline

XIn most instances, this position requires in-person interviews as part of the hiring process.

Minimum qualifications:

Bachelor's degree or equivalent practical experience.
8 years of experience programming in C++ or Python.
5 years of experience testing, and launching software products.
5 years of experience with performance, large-scale systems data analysis, visualization tools, or debugging.
3 years of experience with software design and architecture.

Preferred qualifications:

Experience architecting, validating, or extending high-performance simulation tools, emulation frameworks, or analytical performance modeling pipelines (e.g., roofline models, cycle-accurate, or compiler-aware simulators).
Technical proficiency ML workload performance modeling on pre-silicon systems, familiarity with high level compiler intermediate representations (e.g., MLIR dialects, XLA, HLO), ML execution frameworks (JAX, PyTorch, PyTorch/XLA), or accelerator kernel programming (Pallas, Triton, CUDA).
Ability to lead complex, multi-quarter technical initiatives across distinct organizational boundaries (e.g., hardware design, compilers, AI research, and cloud infrastructure).

About the job

Google's software engineers develop the next-generation technologies that change how billions of users connect, explore, and interact with information and one another. Our products need to handle information at massive scale, and extend well beyond web search. We're looking for engineers who bring fresh ideas from all areas, including information retrieval, distributed computing, large-scale system design, networking and data storage, security, artificial intelligence, natural language processing, UI design and mobile; the list goes on and is growing every day. As a software engineer, you will work on a specific project critical to Google’s needs with opportunities to switch teams and projects as you and our fast-paced business grow and evolve. We need our engineers to be versatile, display leadership qualities and be enthusiastic to take on new problems across the full-stack as we continue to push technology forward.

With your technical expertise you will manage project priorities, deadlines, and deliverables. You will design, develop, test, deploy, maintain, and enhance software solutions.

The MARC (Model Architecture and Realizable-performance Co-design) team is Google’s proactive, ahead-of-silicon co-design engine. We operate at the critical intersection of frontier ML workloads and multi-year TPU silicon roadmaps. Given rapid 6-month model breakthroughs and complex Compound AI Systems, our mandate is to focus on the 12+ month time window between Hardware Architecture Freeze and Physical Silicon Pilot.

In this role, you will define and own the strategy for enabling open-source ML models to achieve TPU performance with internal models at launch. You will establish a principled, scalable methodology to analyze mapping gaps across open-source model families.

The AI and Infrastructure team is redefining what’s possible. We empower Google customers with breakthrough capabilities and insights by delivering AI and Infrastructure at unparalleled scale, efficiency, reliability and velocity. Our customers include Googlers, Google Cloud customers, and billions of Google users worldwide.

We're the driving force behind Google's groundbreaking innovations, empowering the development of our cutting-edge AI models, delivering unparalleled computing power to global services, and providing the essential platforms that enable developers to build the future. From software to hardware our teams are shaping the future of world-leading hyperscale computing, with key teams working on the development of our TPUs, Vertex AI for Google Cloud, Google Global Networking, Data Center operations, systems research, and much more.

Individual pay is determined by factors including job-related skills, experience, and relevant education or training.

US: $207000 - $300000 (USD) + 20% bonus target + equity + benefits

Learn more aboutbenefits at Google (https://www.google.com/about/careers/applications/benefits/) .

Responsibilities

Deliver certified performance benchmarks six months pre-pilot as primary targets for pricing, capacity planning, and kernel optimization.
Adapt emerging open-source models into TPU-native paired variants, including sparsity optimization, to demonstrate differentiated TPU performance.
Ingest and optimize multi-turn agentic workflows, reasoning loops, and prompt and decode topologies prior to silicon tape-out.
Author robust reference implementations using PyTorch, JAX, or Pallas that meet pre-pilot goals under compiler constraints.
Partner with director and executive technical leadership across TPU Hardware, XLA Compiler, and Cloud AI Systems to steer multi-year roadmaps.

Information collected and processed as part of your Google Careers profile, and any job applications you choose to submit is subject to Google'sApplicant and Candidate Privacy Policy (./privacy-policy) .

Google is proud to be an equal opportunity and affirmative action employer. We are committed to building a workforce that is representative of the users we serve, creating a culture of belonging, and providing an equal employment opportunity regardless of race, creed, color, religion, gender, sexual orientation, gender identity/expression, national origin, disability, age, genetic information, veteran status, marital status, pregnancy or related condition (including breastfeeding), expecting or parents-to-be, criminal histories consistent with legal requirements, or any other basis protected by law. See alsoGoogle's EEO Policy (https://www.google.com/about/careers/applications/eeo/) ,Know your rights: workplace discrimination is illegal (https://careers.google.com/jobs/dist/legal/EEOC_KnowYourRights_10_20.pdf) ,Belonging at Google (https://about.google/belonging/) , andHow we hire (https://careers.google.com/how-we-hire/) .

If you have a need that requires accommodation, please let us know by completing ourAccommodations for Applicants form (https://goo.gl/forms/aBt6Pu71i1kzpLHe2) .

Google is a global company and, in order to facilitate efficient collaboration and communication globally, English proficiency is a requirement for all roles unless stated otherwise in the job posting.

To all recruitment agencies: Google does not accept agency resumes. Please do not forward resumes to our jobs alias, Google employees, or any other organization location. Google is not responsible for any fees related to unsolicited resumes.

Equity is granted exclusively and discretionarily by Alphabet Inc. on the basis of an agreement concluded between you and Alphabet Inc. Alphabet Inc. is your sole contractual partner with respect to equity grants. GSU grants are not guaranteed, are discretionary, are subject to approval by the Alphabet Inc. board of directors or its delegate, the terms of the relevant Alphabet Inc. stock plan, and your grant agreement. They have no impact on statutory payments. Current or past grants do not confer an acquired right.

Google is proud to be an equal opportunity workplace and is an affirmative action employer. We are committed to equal employment opportunity regardless of race, color, ancestry, religion, sex, national origin, sexual orientation, age, citizenship, marital status, disability, gender identity or Veteran status. We also consider qualified applicants regardless of criminal histories, consistent with legal requirements. See also https://careers.google.com/eeo/ and https://careers.google.com/jobs/dist/legal/OFCCP_EEO_Post.pdf If you have a need that requires accommodation, please let us know by completing our Accommodations for Applicants form: https://goo.gl/forms/aBt6Pu71i1kzpLHe2.

Apply About employer

Report this job