Senior ML Compiler Engineer, Silicon

9 Hours 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

Epic Games - Lead Engine Programmer, Verse Framework

Epic Games

Cary, North Carolina, United States (On-Site)
2 Days ago
Google - Customer Engineer, Infrastructure Modernization, Google Cloud

Google

Seattle, Washington, United States (On-Site)
11 Hours ago
NVIDIA - Malware Research Architect

NVIDIA

Germany (Remote)
4 Weeks ago
Google - Cloud Technical Solutions Engineer, Infrastructure Compute

Google

Warsaw, Masovian Voivodeship, Poland (On-Site)
10 Hours ago
NVIDIA - Solutions Architect, Networking

NVIDIA

Santa Clara, California, United States (Hybrid)
2 Weeks ago
Globalization Partners - Principal AI Engineer

Globalization Partners

(Remote)
3 Months ago
Rackspace Technology - AI/ML Architect

Rackspace Technology

Vietnam (Remote)
1 Month ago
Microsoft - Platform Engineering Manager

Microsoft

Redmond, Washington, United States (Hybrid)
19 Hours ago
Google - Technical Program Manager III, Hardware and Software Systems, Cloud AI Systems

Google

Sunnyvale, California, United States (On-Site)
9 Hours ago
Windranger Labs - Technical AI Researcher

Windranger Labs

Singapore (On-Site)
3 Weeks ago

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

Google - Staff Software Engineer, Embedded Systems, Platforms Infrastructure Engineering

Google

Sunnyvale, California, United States (On-Site)
8 Hours ago
Google - Cloud Technical Solutions Engineer, Networking

Google

Tokyo, Japan (On-Site)
10 Hours ago
Nintendo - Intern - Digital Design

Nintendo

Redmond, Washington, United States (On-Site)
5 Months ago
ByteDance - Site Reliability Engineer, Compute Platform

ByteDance

San Jose, California, United States (On-Site)
5 Months ago
NVIDIA - Senior Chip Architect

NVIDIA

Tel Aviv-Yafo, Tel Aviv District, Israel (On-Site)
1 Month ago
Google - Senior Software Engineer, Infrastructure, Google Cloud Runtimes

Google

San Francisco, California, United States (On-Site)
10 Hours ago
InMobiInMobi - Sales Manager

InMobiInMobi

Riyadh, Riyadh Province, Saudi Arabia (On-Site)
9 Months ago
Nintendo - Senior Sourcing Recruiter

Nintendo

Redmond, Washington, United States (Hybrid)
8 Months ago
Google - Emerging Market Lead, Devices and Services

Google

Gurugram, Haryana, India (On-Site)
9 Hours ago
Google - Formal Verification Engineer, Silicon

Google

Bengaluru, Karnataka, India (On-Site)
11 Hours ago

Get notifed when new similar jobs are uploaded

Jobs in Bengaluru, Karnataka, India

Nagarro - Senior Engineer, Data Analytics

Nagarro

India (Remote)
6 Months ago
CloudHire - Scala API Architect

CloudHire

Karnataka, India (Remote)
4 Weeks ago
Paytm - Sales - Team Lead - Meerut

Paytm

Meerut, Uttar Pradesh, India (On-Site)
5 Months ago
PwC - Manager_Conversational AI Developer_Advisory Corporate_Advisory_Bangalore

PwC

Bengaluru, Karnataka, India (On-Site)
6 Months ago
Gallagher - Data Scientist

Gallagher

Bengaluru, Karnataka, India (On-Site)
5 Months ago
Quickplay - Senior Software Engineer (React Native-Android UI Development)

Quickplay

Chennai, Tamil Nadu, India (Hybrid)
9 Months ago
Google - Software Developer, Pixel Watch, System Software

Google

Bengaluru, Karnataka, India (On-Site)
10 Hours ago
Assystems - BIM Modeler - Water

Assystems

Bengaluru, Karnataka, India (On-Site)
6 Months ago
CloudHire - Senior Scala Architect

CloudHire

Bengaluru, Karnataka, India (Remote)
3 Weeks ago
GoMotive - Associate Enterprise Systems Product Manager

GoMotive

Bengaluru, Karnataka, India (Remote)
1 Month ago

Get notifed when new similar jobs are uploaded

Artificial Intelligence Jobs

ByteDance - AI Security Researcher - Security - San Jose

ByteDance

San Jose, California, United States (On-Site)
5 Months ago
Google - Digital Transformation Consultant

Google

Hyderabad, Telangana, India (On-Site)
9 Hours ago
GoMotive - Software Engineer, Machine Learning-Perception

GoMotive

India (Remote)
4 Weeks ago
Google - Staff Software Engineer

Google

Warsaw, Masovian Voivodeship, Poland (On-Site)
8 Hours ago
Microsoft - Principal Applied Scientist

Microsoft

Redmond, Washington, United States (On-Site)
19 Hours ago
AI Fund - Curriculum Developer

AI Fund

(Remote)
6 Months ago
Google - Software Engineering Manager, Google Kubernetes AI Infrastructure

Google

Kirkland, Washington, United States (On-Site)
11 Hours ago
Equivalent Jobs - DL RESEARCHER

Equivalent Jobs

(Remote)
5 Months ago
Krafton  - Applied Research Scientist/Engineer - LLM Game Agent

Krafton

Seoul, South Korea (On-Site)
3 Weeks ago
Interface AI - Software Development Engineer IV - Backend

Interface AI

India (Remote)
2 Months 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.

Bucharest, Bucharest, Romania (On-Site)

Bengaluru, Karnataka, India (On-Site)

Bengaluru, Karnataka, India (On-Site)

Dublin, County Dublin, Ireland (On-Site)

Bengaluru, Karnataka, India (On-Site)

Sunnyvale, California, United States (On-Site)

Sunnyvale, California, 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