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 9 mins ago
M
Software System Test Engineer
ManpowerGroup
Santa Clarita, CA
- $55 to $65 USD / hour
We are seeking an experienced Software Systems Test Engineer to support R&D and engineering initiatives. The ideal candidate will have strong Python programming skills, hands-on system and…
Added 9 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 10 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 10 mins ago
RF Engineer, Amazon Leo Government, Amazon Leo Government
Amazon
Northridge, CA
…specific customers (such as government agencies). The RF/Microwave Engineer will play a pivotal role in the team delivering innovative and fast-paced analysis and design…
Added 10 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 10 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 10 mins ago
Operations AI Engineer - Early Career
Lockheed Martin
Palmdale, CA
Perform exploratory research and analysis to identify novel and meaningful patterns in data and uses statistical methods to reject or accept proposed hypotheses about relationships or…
Added 10 mins ago
Software Engineer, Machine Learning - Ads Infra and Ranking
Meta
Menlo Park, CA
Software Engineer, Machine Learning - Ads Infra and Ranking Responsibilities:
Added 10 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 10 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 11 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…
Added 11 mins ago
Industrial Engineer /Principal Industrial Engineer
Northrop Grumman
Sunnyvale, CA
Northrop Grumman Mission Systems is looking for you to join our team as an Industrial Engineer / Principal Industrial Engineer based out of Sunnyvale, CA .
Added 11 mins ago
Mechanical Engineer II (Thermal Analysis)
Lockheed Martin
Sunnyvale, CA
…community of engineers looking for an enthusiastic Aeronautical Engineer to join the tight-knit group. The Fleet Ballistic Missile program has been of Lockheed Martin’s premier Lines of…
Added 11 mins ago
TI
Analog Design Eng Intern
Texas Instruments
Santa Clara, CA
- $33 to $47.5 USD / hour
Partnering with business teams and system engineering to develop mutually agreeable design specifications
Added 11 mins ago
TI
Analog Design Verification Eng Intern
Texas Instruments
Santa Clara, CA
- $33 to $47.5 USD / hour
Partnering with business teams and system engineering to develop mutually agreeable design specifications
Added 11 mins ago
TI
Digital Design Verification Engineering Intern
Texas Instruments
Santa Clara, CA
- $33 to $47.5 USD / hour
Partnering with business teams and system engineering to develop mutually agreeable design specifications
Added 11 mins ago
Staff NPI Engineer - Johnson and Johnson MedTech, Circulatory Restoration
J&J Family of Companies
Santa Clara, CA
Johnson & Johnson is hiring for a Staff NPI Engineer to join our team located in Santa Clara, CA.
Added 11 mins ago
Engineer/Principal Engineer Aeronautical - Hardware Avionics
Northrop Grumman
San Diego, CA
Experience with system redundancy and associated safety analysis
Added 11 mins ago
Principal / Sr Principal Engineer - Product Support (DMSMS)
Northrop Grumman
San Diego, CA · Palmdale, CA
Bachelor's degree in a STEM field with a focus on data analysis, statistics, or system modeling.
Added 12 mins ago
Specialist, Test Engineer
L3Harris
San Diego, CA
We are seeking an experienced Test Engineer to support the design verification, characterization, and integration of automated RF test systems and subsystems (ATS). The ideal candidate will…
Added 12 mins ago
Engineer VII
General Atomics
San Diego, CA
…expanding portfolio of specialized products and integrated system solutions support critical fleet, space systems and satellites, missile defense, power and energy, and process and…
Added 12 mins ago
Principal / Senior Principal Software Engineer
Northrop Grumman
Los Angeles, CA
We are looking for you to join our team as a Principal Software Engineer or Senior Principal Software Engineer based out of Woodland Hills, CA . As a Software Engineer at Northrop Grumman…
Added 12 mins ago
Principal / Senior Principal Software Engineer
Northrop Grumman
Los Angeles, CA
We are looking for you to join our team as a Principal or Senior Principal Software Engineer based out of Woodland Hills, CA . As a Software Engineer at Northrop Grumman Mission Systems you…
Added 12 mins ago
Lead Manufacturing Engineer
Winchester Interconnect
Garden Grove, CA
Job Title: Senior Manufacturing & Process Engineer
Added 12 mins ago
Maritime Software Integration and DevOps Engineer
BOOZ, ALLEN & HAMILTON, INC.
Oxnard, CA
As a Maritime Software Integration and DevOps Engineer, you'll bring your object oriented software solutioning and DevOps expertise to work on projects for the U.S. Navy. You'll support…

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