Senior ML Compiler Engineer, Silicon

2 Months ago • 5-8 Years • 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

Telastra - Staff Software Engineer (Pega Infra)

Telastra

Australia (On-Site)
1 Month ago
Impact Theory - Senior Game Programmer

Impact Theory

Los Angeles, California, United States (On-Site)
7 Months ago
Hashone Careers - Cloud Engineer

Hashone Careers

Bengaluru, Karnataka, India (Remote)
7 Months ago
HP - Graduate - Care Center Indigo Remote Support Engineer - German Speaking

HP

Sant Cugat Del Vallès, Catalonia, Spain (Remote)
1 Month ago
Smilegate - AI Developer ([LOST ARK Mobile])

Smilegate

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

Canva

Vienna, Vienna, Austria (Remote)
3 Months ago
NVIDIA - Solution Architect - Auto

NVIDIA

Beijing, Beijing, China (On-Site)
5 Months ago
NVIDIA - Senior AI-HPC Cluster Engineer

NVIDIA

Westford, Massachusetts, United States (Hybrid)
3 Months ago
Xsolla - Machine Learning Engineer

Xsolla

Montreal, Quebec, Canada (Remote)
2 Months ago

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

Microsoft - Software Engineer - Systems

Microsoft

Redmond, Washington, United States (On-Site)
2 Months ago
Intelerad - System Support Specialist I

Intelerad

Canada (Remote)
1 Month ago
Behaviour Interactive - Principal Gameplay Programmer - Dead by Daylight | Programmeur·se jouabilité Principal·e - Dead by Daylight

Behaviour Interactive

Middlesbrough, England, United Kingdom (Hybrid)
9 Months ago
The Walt Disney Company - Senior Software Engineer

The Walt Disney Company

Seattle, Washington, United States (On-Site)
2 Months ago
Scopely - Unity Client Engineer - Unannounced Project

Scopely

London, England, United Kingdom (Hybrid)
4 Months ago
Tesla - Dual Master's Sustainable Battery Production Engineering (M.Eng.)

Tesla

Brandenburg, Germany (On-Site)
4 Months ago
Optiv - Sr. Cloud Engineer

Optiv

Overland Park, Kansas, United States (Remote)
1 Month ago
Minecast - Enterprise Account Manager

Minecast

London, England, United Kingdom (On-Site)
1 Month ago
Stone Search - Senior Cloud (Azure) Engineer

Stone Search

Fair Lawn, New Jersey, United States (Hybrid)
1 Year ago
Tesla - Senior Business Planning Coordinator

Tesla

Zug, Zug, Switzerland (On-Site)
4 Months ago

Get notifed when new similar jobs are uploaded

Jobs in Bengaluru, Karnataka, India

Capgemini - NodeJS Developer with DevOps

Capgemini

Gurugram, Haryana, India (On-Site)
1 Month ago
bounteous - Senior Salesforce Engineer

bounteous

Chennai, Tamil Nadu, India (Hybrid)
10 Months ago
Ethernovia - BSP Embedded S/W

Ethernovia

Pune, Maharashtra, India (On-Site)
1 Month ago
Qualcomm - Embedded Dev (GFX) - Engineer, Senior

Qualcomm

Bengaluru, Karnataka, India (On-Site)
2 Months ago
smarsh - Lead Security Operations Analyst

smarsh

India (Hybrid)
1 Month ago
Google - Business Systems Analyst, Supply Chain

Google

Bengaluru, Karnataka, India (On-Site)
2 Months ago
Haleon - Data Engineer

Haleon

Bengaluru, Karnataka, India (On-Site)
1 Month ago
Trend Micro - Sales Engineer

Trend Micro

New Delhi, Delhi, India (Hybrid)
3 Weeks ago
Capgemini - ServiceNow HRSD Design Lead

Capgemini

Bengaluru, Karnataka, India (On-Site)
1 Month ago
Mindtickle - Senior Product Manager (Platform)

Mindtickle

Pune, Maharashtra, India (On-Site)
3 Months ago

Get notifed when new similar jobs are uploaded

Research Development Jobs

Google - Senior Staff Software Engineer, BigQuery Generative AI

Google

Kirkland, Washington, United States (On-Site)
2 Months ago
Google - Senior Staff Software Engineer, Accelerators Systems Software Architect

Google

Sunnyvale, California, United States (On-Site)
2 Months ago
Google - Software Engineer III, Embedded Systems/Firmware, Platforms Infrastructure Engineering

Google

Sunnyvale, California, United States (On-Site)
2 Months ago
bytedance - Senior XR Strategy Expert

bytedance

San Jose, California, United States (On-Site)
2 Months ago
Postman - Software Engineer - Applied AI Engineer

Postman

Bengaluru, Karnataka, India (On-Site)
8 Months ago
AI Fund - Curriculum Developer

AI Fund

Germany (Remote)
8 Months ago
Keywords Studios - AI - Technical Research Associate (Prompts)

Keywords Studios

Silesian Voivodeship, Poland (On-Site)
3 Months ago
Krafton - Deep Learning Engineer - Model Optimization

Krafton

Seoul, South Korea (On-Site)
2 Months ago
GoMotive - Software Engineer, Machine Learning

GoMotive

Pakistan (Remote)
3 Months ago
bytedance - Student Researcher (Doubao (Seed) - Foundation Model - Generative AI)

bytedance

Seattle, Washington, United States (On-Site)
3 Months ago

Get notifed when new similar jobs are uploaded

About The Company

Atlanta, Georgia, United States (On-Site)

New York, United States (On-Site)

London, England, United Kingdom (On-Site)

Taipei City, Taiwan (On-Site)

Kirkland, Washington, United States (On-Site)

Sunnyvale, California, United States (On-Site)

Sunnyvale, California, United States (On-Site)

Sunnyvale, California, United States (On-Site)

Kraków, Lesser Poland Voivodeship, Poland (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