Senior ML Compiler Engineer, Silicon

1 Month 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

Winzo - Strategy & Growth (Trade Marketing)

Winzo

New Delhi, Delhi, India (On-Site)
2 Weeks ago
Instawork - Senior Account Executive - Field Sales

Instawork

Los Angeles, California, United States (Hybrid)
1 Month ago
Capgemini - DC-ACI/Nexus

Capgemini

Bengaluru, Karnataka, India (On-Site)
2 Weeks ago
Playtika - Senior Network Engineer

Playtika

Ukraine (On-Site)
1 Month ago
Google - Software Engineer, PhD, Early Career, Campus, Systems and Infrastructure, 2025 Start

Google

Atlanta, Georgia, United States (On-Site)
6 Months ago
Google - Software Engineer III, AI/ML Recommendations, Rankings, Predictions, YouTube

Google

Mountain View, California, United States (On-Site)
1 Month ago
Spellbrush - AI Infrastructure Engineer

Spellbrush

San Francisco, California, United States (On-Site)
2 Months ago
Twitch - Sr. Applied Scientist

Twitch

San Francisco, California, United States (On-Site)
2 Months ago
Google - Senior Software Engineer, Distributed Machine Learning

Google

Mountain View, California, United States (On-Site)
1 Month ago
N-ix - Senior DS/AI Engineer

N-ix

Poland (Hybrid)
1 Month ago

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

Google - Software Engineer III, Mobile, Android

Google

Bengaluru, Karnataka, India (On-Site)
6 Months ago
Google - Staff Software Engineer, Host Networking

Google

Sunnyvale, California, United States (On-Site)
1 Month ago
Google - Software Engineer II, Android Automotive

Google

Kraków, Lesser Poland Voivodeship, Poland (On-Site)
1 Month ago
JMA - Backend GO Developer

JMA

Bologna, Emilia-Romagna, Italy (Hybrid)
3 Weeks ago
high radius - Recruiter

high radius

Houston, Texas, United States (On-Site)
3 Weeks ago
Google - Software Engineer, Android, Photos

Google

Sydney, New South Wales, Australia (On-Site)
1 Month ago
Ansys - Senior R&D Engineer

Ansys

Pune, Maharashtra, India (On-Site)
1 Week ago
Hashone Careers - Cloud Engineer

Hashone Careers

Bengaluru, Karnataka, India (Remote)
6 Months ago
Google - Software Engineer, People with Disabilities

Google

Belo Horizonte, State Of Minas Gerais, Brazil (On-Site)
6 Months ago
bytedance - Senior/Tech Lead AI/LLM Network Software Development Engineer - Seattle

bytedance

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

Get notifed when new similar jobs are uploaded

Jobs in Bengaluru, Karnataka, India

Nagarro - Associate Staff Engineer ,Mobile Developer (React Native)

Nagarro

Bengaluru, Karnataka, India (On-Site)
7 Months ago
Pay2 - Android Engineer

Pay2

Gurugram, India (On-Site)
4 Days ago
Mindtickle - Staff Engineer

Mindtickle

Pune, Maharashtra, India (On-Site)
2 Months ago
Zelis  - Data Engineer

Zelis

Hyderabad, Telangana, India (Hybrid)
5 Days ago
AlphaSense - Content Editor

AlphaSense

Pune, Maharashtra, India (On-Site)
6 Days ago
Neolytix - US Healthcare Trainees– Medical billing and Credentialing Process

Neolytix

Gurugram, Haryana, India (On-Site)
1 Week ago
Rackspace Technology - Marketing Operations Analyst II

Rackspace Technology

Gurugram, Haryana, India (Remote)
2 Months ago
AlphaSense - Join AlphaSense India Talent Community

AlphaSense

Mumbai, Maharashtra, India (On-Site)
6 Days ago
Progress carrers - Salesforce CPQ Technical Lead/Developer

Progress carrers

Hyderabad, Telangana, India (Hybrid)
2 Months ago
Accenture - Manager

Accenture

Gurugram, Haryana, India (On-Site)
1 Week ago

Get notifed when new similar jobs are uploaded

Artificial Intelligence Jobs

Inworld AI - AI Trainer (Contractor) - Writing & Gaming

Inworld AI

Mountain View, California, United States (Remote)
2 Months ago
Google - Staff Software Engineer, Machine Learning

Google

Los Angeles, California, United States (On-Site)
1 Month ago
CharacterAI - Research Engineer, Post-Training

CharacterAI

New York, New York, United States (On-Site)
2 Months ago
Social Discovery Group - Senior NLP Engineer

Social Discovery Group

Serbia (Remote)
7 Months ago
Google - Software Engineer III, Embedded Systems/Firmware, Platforms Infrastructure Engineering

Google

Madison, Wisconsin, United States (On-Site)
1 Month ago
Rackspace Technology - Machine Learning Architect (AWS)

Rackspace Technology

(Remote)
4 Months ago
Rackspace Technology - Machine Learning Architect (AWS)

Rackspace Technology

San Diego, California, United States (Remote)
2 Months ago
bytedance - Student Researcher (Doubao (Seed) - Foundation Model - MultiModal Generative Model)

bytedance

San Jose, California, United States (On-Site)
2 Months ago
Ubisoft - Senior ML Data Scientist

Ubisoft

Montreal, Quebec, Canada (On-Site)
4 Months ago
Reality Games - Machine Learning Engineer - Monopoly World

Reality Games

Kraków, Lesser Poland Voivodeship, Poland (On-Site)
3 Months ago

Get notifed when new similar jobs are uploaded

About The Company

London, England, United Kingdom (On-Site)

Bengaluru, Karnataka, India (On-Site)

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

Bengaluru, Karnataka, India (On-Site)

Taipei City, Taiwan (On-Site)

Zürich, Zurich, Switzerland (On-Site)

Kirkland, Washington, United States (On-Site)

New Taipei, New Taipei City, Taiwan (On-Site)

Seattle, Washington, United States (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