Jobs in the United States

25+ jobs

Save this search

Added 5 mins ago
Senior Databricks Data Engineer
Securiport
Reston, VA
…analysis, certification, and compliance. The Senior Databricks Data Engineer will lead the design, development, and optimization of data pipelines and data platforms within a…
Added 7 mins ago
VT
Systems Administrator
VivSoft Technologies
Remote, United States
- $65,000 to $75,000 USD / year
Red Hat Certified Engineer (RHCE) · Active Public Trust or security clearance · DHS, DoW or civilian agency experience · Ansible, Puppet or PowerShell DSC · Azure Government or AWS GovCloud…
Added 12 mins ago
Mission Engineer
BAE Systems
Sterling, VA
We are looking for a Missions Engineer experienced in delivering comprehensive multi-source analysis, reporting, and intelligence production supporting a new Enterprise lab environment…
Added 12 mins ago
AI/ML Engineer
BAE Systems
Sterling, VA
We are looking for an AI/ML Engineer to support the design, development, and deployment of intelligent systems that can learn from data, make predictions, classify objects, and generate…
Added 21 mins ago
C
AI Engineer
Claro
Remote, United States
- $100,000 to $110,000 USD / year
The AI Engineer is responsible for building and deploying AI agents (Microsoft Copilot and Claude) and private retrieval-augmented generation (RAG) applications. This role requires…
Added 24 mins ago
Lead AI Engineer - NBA
Humana
Jefferson City, MO
The Lead AI Engineer owns the design and delivery of the NBA platform's agentic AI capabilities. This is a hands-on engineering role first - you are expected to be a significant individual…
Added 26 mins ago
Summer 2027 AI Design Intern for Facility Design
The Haskell Company
Jacksonville, FL
Added 29 mins ago
ML Software Engineer Intern - AI/CAD Integration
Falcomm
Atlanta, GA
…energy-efficient power amplifier products. As a ML Software Engineer Intern in AI/CAD Integration, you will collaborate with our team of experts to develop and deploy AI/ML models that…
Added 29 mins ago
AI/CAD Software Engineer
Falcomm
Atlanta, GA
…Amplifier (PA) products. We are seeking a full-time AI/CAD Software Engineer to join our team. You will lead the development of intelligent tools that enhance RF workflows, automate complex…
Added 32 mins ago
Senior ML/AI Engineer
GuruLink
Remote, United States
The Senior ML/AI Engineer will take models from prototype into production and keep them healthy, on a platform whose output is used daily by care teams working with real patients. The data…
Added 32 mins ago
Lead ML/AI Engineer
GuruLink
Remote, United States
The Lead ML/AI Engineer is the technical lead for the platform that turns model predictions into real-world actions for patients and care teams. The data science group runs a growing set of…
Added 40 mins ago
P
Intern - Operations Data Science
PODS
Clearwater, FL
- $25 to $30 USD / hour
Do real data science. Prepare and validate operational data, engineer features, and run model and optimization experiments with senior data scientists.
Added 47 mins ago
Staff Data Scientist
John Deere
IL · Deer Creek
Use PAAW datasets to engineer features that capture machine, operational, and environmental context, and associate that context with technology usage to explain how and where each…
Added 56 mins ago
Senior Machine Learning Engineer
VODA.ai
Remote, United States
VODA.ai is seeking a full-time Senior Machine Learning Engineer. In this role, you will own the reliability, scalability, and automation of our machine-learning pipelines. The role spans…
Added 58 mins ago
Sr. Process Engineer, R1 Body AI/ML
Rivian
Normal, IL
Added 59 mins ago
UG
Sr. AI/ML Engineer - Remote
UnitedHealth Group
Richardson, TX
- $120,100 to $214,500 USD / year
The Optum Health Technology team is seeking a Senior AI/ML Engineer with a solid data science background to build and scale cutting-edge AI-powered solutions that transform value-based care…
Added 1 hour ago
Senior Silicon Validation Engineer
Google
Sunnyvale, CA
Added 1 hour ago
Staff Software Engineer, ML Compiler Performance Engineer Co-Design
Google
Mountain View, CA
…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…
Added 1 hour ago
Staff AI/ML Software Engineer – Navigation R&D
DePuy Synthes Products Inc. Legal Entity
Boston, MA · Warsaw, IN
We are searching for the best talent for a Staff AI/ML Software Engineer – Navigation R&D. This is a hybrid role to be based out of Boston, MA or Warsaw, IN.
Added 1 hour ago
UG
Advisory Services Analyst, Consultant Development Program - AI/ML Engineering Track - Multiple Locations
UnitedHealth Group
Eden Prairie, MN
- $60,200 to $107,400 USD / year
Added 1 hour ago
T
Staff Software Engineer (Applied AI)
Traba
New York City, NY · San Francisco, CA
- $240,000 to $300,000 USD / year
Added 1 hour ago
Senior Talent Partner - US
Runware
Remote, United States
You don't need to be an engineer, but you should be genuinely comfortable in technical environments.
Added 1 hour ago
Intern - Operations Data Science
PODS Enterprises, LLC
Clearwater, FL
Do real data science. Prepare and validate operational data, engineer features, and run model and optimization experiments with senior data scientists.
Added 1 hour ago
Sr. Engineer - Dev Experience Sensor
CrowdStrike, Inc.
Redmond, WA
This is a (Sr/Sr Staff-lvl) Software Tools Engineer role under the End-to-End engineering team that supports validation of customer workflows, involving real sensor, cloud and UI. The tools…
Added 2 hours ago
Sr. Software Development Engineer, AWS Transform
Amazon Development Center U.S., Inc.
Boston, MA

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

Original job description

In 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).

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 about benefits at Google.

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.

Apply About employer

More about this job

Skills

ML
C++
Python
Data analysis
Debugging
PyTorch
Design hardware
Research
Distributed computing
Data storage
Natural language processing
UI design
Leadership
Open source model
Google Cloud

Report this job