Senior ML Compiler Engineer, Silicon

2 Weeks ago • 5-8 Years • Artificial Intelligence • Research & Development

Job Summary

Job Description

As a Senior ML Compiler Engineer at Google, you will develop Machine Learning (ML) compilers for the Tensor TPU to accelerate Generative AI (GenAI) and other complex machine learning models running on custom hardware accelerators. Responsibilities include building compilers and tools that map ML models to the hardware ISA, evaluating trade-offs of parallelization strategies (performance, power, energy, memory), collaborating with ML researchers and hardware engineers, and productizing ML workloads on the Pixel Edge TPU. This role requires a deep understanding of compiler technology and experience optimizing for performance and power efficiency in hardware-accelerated systems. The position involves working on projects critical to Google's needs with opportunities to switch teams and projects as the business evolves.
Must have:
  • 5+ years compiler experience
  • 3+ years software product experience
  • Bachelor's degree in CS or related field
  • Build ML compilers for Tensor TPU
  • Evaluate parallelization strategies
  • Collaborate with researchers and engineers
Good to have:
  • Master's or PhD in CS
  • Power/performance optimization experience
  • Domain-specific ML compiler experience
  • Hardware parallelism knowledge

Job Details

Minimum qualifications:

  • Bachelor's degree in Computer Science, a related technical field, or equivalent practical experience.
  • 5 years of experience with Compilers
  • 3 years of experience testing, maintaining, or launching software products, with 1 year of experience with software design and architecture.

Preferred qualifications:

  • Master's degree or PhD in Computer Science, or a related technical field.
  • Experience in power and performance optimizations.
  • Experience with domain-specific compilers for machine learning.
  • Knowledge of hardware that provides a degree of parallelism.

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.

As a Software Engineer you will work on developing Machine Learning (ML) compilers for the Tensor TPU to accelerate Generative AI (GenAI)and other complex machine learning models running on custom hardware accelerators. Along with your technical expertise, you will manage project priorities, deadlines and deliverables.

Google's mission is to organize the world's information and make it universally accessible and useful. Our team combines the best of Google AI, Software, and Hardware to create radically helpful experiences. We research, design, and develop new technologies and hardware to make computing faster, seamless, and more powerful. We aim to make people's lives better through technology.

Responsibilities

  • Build compilers and tools that map ML models (with a particular focus on always-on computing use cases) to the hardware Instruction Set Architecture (ISA).
  • Evaluate various trade-offs of different parallelization strategies such as performance, power, energy and memory consumption.
  • Collaborate with machine learning researchers to improve the domain-specific compiler.
  • Collaborate with hardware engineers to evolve future accelerators.
  • Productize various ML workloads on the Pixel Edge TPU including Gemini Nano, Camera, Imaging and Speech ML models.

Similar Jobs

BULKHEAD - Senior Gameplay Engineer

BULKHEAD

Derby, England, United Kingdom (On-Site)
1 Month ago
Gaming Innovation Group  - Senior Platform DevOps Engineer

Gaming Innovation Group

St. Julian's, Malta (Hybrid)
1 Month ago
AGBO - Gameplay / Tools Programmer

AGBO

Raleigh, North Carolina, United States (On-Site)
3 Weeks ago
OKX - Sales Manager (Russian Speaker)

OKX

Dubai, Dubai, United Arab Emirates (On-Site)
9 Hours ago
NVIDIA - Senior Software Engineer

NVIDIA

Tel Aviv-Yafo, Tel Aviv District, Israel (On-Site)
3 Months ago
The Walt Disney Company - Senior Data Scientist - NLP/LLM

The Walt Disney Company

Glendale, California, United States (On-Site)
2 Weeks ago
Hitachi - Senior AI Data Scientist

Hitachi

Chennai, Tamil Nadu, India (On-Site)
6 Months ago
SmileGate - AI Data Curation Specialist

SmileGate

Seongnam-si, Gyeonggi-do, South Korea (On-Site)
2 Weeks ago
Canva - Senior Computer Vision Engineer - Photo AI

Canva

Prague, Czechia (Remote)
3 Months ago
Tencent - IaaS Product Solution Architect

Tencent

(On-Site)
3 Weeks ago

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

Applike Group - Acquisition Manager - Mobile Gaming

Applike Group

Hamburg, Hamburg, Germany (Hybrid)
3 Months ago
Every matrix - Senior Sales Manager

Every matrix

Yerevan, Yerevan, Armenia (Hybrid)
3 Months ago
anydesk - Operations Engineer / Linux Administrator

anydesk

Stuttgart, Baden-Württemberg, Germany (On-Site)
7 Hours ago
Google - Software Engineer, Black Community Inclusion

Google

São Paulo, State Of São Paulo, Brazil (On-Site)
1 Week ago
NVIDIA - Senior Mechanical Engineer

NVIDIA

Yokne'am Illit, North District, Israel (On-Site)
1 Week ago
Palo Alto Networks - Sr. Technical Support Engineer, Focused Services

Palo Alto Networks

Plano, Texas, United States (On-Site)
2 Days ago
Microsoft - Digital Solution Area Specialist

Microsoft

Sydney, New South Wales, Australia (On-Site)
1 Week ago
Google - Principal Staff Software Engineer, AI and Data Infrastructure

Google

London, England, United Kingdom (On-Site)
2 Days ago
Fennec Labs - Mid/Senior Unity Engineer

Fennec Labs

Stafford, England, United Kingdom (On-Site)
8 Months ago
Google - Staff Software Engineer, Google Cloud

Google

Cambridge, Massachusetts, United States (On-Site)
2 Days ago

Get notifed when new similar jobs are uploaded

Jobs in Bengaluru, Karnataka, India

SafeSend - QA Engineer L1

SafeSend

Bengaluru, Karnataka, India (On-Site)
6 Months ago
Boomi - Senior Specialist - Accounts Receivable

Boomi

Hyderabad, Telangana, India (On-Site)
1 Day ago
PwC - Senior Associate - RPA

PwC

Pune, Maharashtra, India (On-Site)
7 Months ago
Fi - Product Designer

Fi

Bengaluru, Karnataka, India (On-Site)
5 Years ago
Sumo Logic - Senior Product Manager II

Sumo Logic

Noida, Uttar Pradesh, India (On-Site)
1 Day ago
Cognite - Senior Financial Data Analytics

Cognite

Bengaluru, Karnataka, India (Hybrid)
1 Month ago
AI Dash - SDE 1 - Frontend

AI Dash

Bengaluru, Karnataka, India (Hybrid)
1 Month ago
ION - UK Holdings - India - 855

ION

Mumbai, Maharashtra, India (On-Site)
6 Months ago
Accenture - Trust & Safety New Associate

Accenture

Navi Mumbai, Maharashtra, India (On-Site)
1 Week ago
Cyara - Software Engineer - Node.JS + React.JS

Cyara

Hyderabad, Telangana, India (Hybrid)
1 Day ago

Get notifed when new similar jobs are uploaded

Artificial Intelligence Jobs

NVIDIA - Senior High-Performance LLM Training Engineer

NVIDIA

Santa Clara, California, United States (Hybrid)
4 Weeks ago
DEVOTEAM - Data Driven | MLOps Engineer

DEVOTEAM

Lisbon, Lisbon, Portugal (Remote)
6 Months ago
Google - Senior Machine Learning Physical Design Engineer

Google

Bengaluru, Karnataka, India (On-Site)
2 Days ago
HP - Machine Learning Engineer

HP

Palo Alto, California, United States (On-Site)
7 Months ago
Google - Student Researcher, BS/MS, Winter/Summer 2025

Google

Mountain View, California, United States (On-Site)
5 Months ago
Google - Software Engineer III, AI/ML, Google Ads

Google

Mountain View, California, United States (On-Site)
2 Weeks ago
ByteDance - Research Scientist, Responsible AI

ByteDance

San Jose, California, United States (On-Site)
1 Month ago
Microsoft - Applied Scientist: Microsoft AI – PhD – Redmond

Microsoft

Redmond, Washington, United States (On-Site)
2 Weeks ago
Google - Staff Research Scientist, Google Cloud AI

Google

Sunnyvale, California, United States (On-Site)
2 Weeks ago
Inworld AI - Staff C++ Engineer

Inworld AI

Mountain View, California, United States (On-Site)
1 Month ago

Get notifed when new similar jobs are uploaded

About The Company

A problem isn't truly solved until it's solved for all. Googlers build products that help create opportunities for everyone, whether down the street or across the globe. Bring your insight, imagination and a healthy disregard for the impossible. Bring everything that makes you unique. Together, we can build for everyone.

Mountain View, California, United States (On-Site)

Mountain View, California, United States (On-Site)

Bengaluru, Karnataka, India (On-Site)

Bengaluru, Karnataka, India (On-Site)

Bengaluru, Karnataka, India (On-Site)

Bengaluru, Karnataka, India (On-Site)

Bengaluru, Karnataka, India (On-Site)

View All Jobs

Get notified when new jobs are added by Google

Level Up Your Career in Game Development!

Transform Your Passion into Profession with Our Comprehensive Courses for Aspiring Game Developers.

Job Common Plug