Software Engineer, ML Compilers, Silicon

2 Weeks ago • 2 Years + • Research & Development

Job Summary

Job Description

Google is seeking a Software Engineer to develop ML compilers for the Tensor TPU, accelerating Generative AI and complex machine learning models on custom hardware. Responsibilities include building compilers and tools to efficiently map ML models to the hardware ISA, evaluating parallelization strategies (performance, power, energy, memory), collaborating with ML researchers to improve the compiler, and collaborating with hardware engineers to evolve future accelerators. The role involves managing project priorities, deadlines, and deliverables. This position is crucial to Google's development of its custom System-on-Chip (SoC), Tensor, which powers Pixel phones and enables Gemini Nano, an efficient AI model for on-device tasks.
Must have:
  • Bachelor's degree in relevant field
  • 2 years C++ development experience
  • Data structures and algorithms expertise
  • Build ML compilers for Tensor TPU
  • Evaluate parallelization strategies
Good to have:
  • Master's or PhD in relevant field
  • Compiler experience
  • Power/performance optimization experience
  • Hardware understanding (parallelism)

Job Details

Minimum qualifications:

  • Bachelor's degree in Electrical/Electronics Engineering, Computer Engineering, Computer Science, or equivalent practical experience
  • 2 years of development experience in C++
  • Experience in data structures and algorithms

Preferred qualifications:

  • Master's degree or PhD in Electrical/Electronics Engineering, Computer Engineering, Computer Science, or a related field
  • Experience with compilers
  • Experience in power and performance optimizations
  • Understanding of hardware, especially hardware that provides a high 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.

We are the team that builds Google Tensor - Google’s custom System-on-Chip (SoC) that powers the latest Pixel phones. Tensor makes transformative user experiences possible with the help of cutting-edge Machine Learning (ML) running on Tensor TPU. Our team’s work enables Gemini Nano, our efficient AI model for on-device tasks to run on Pixel phones. Our goal is to productize the latest ML innovations and research by delivering computing hardware and software.vAs a Software Engineer you will work on developing ML compilers for the Tensor TPU to accelerate Generative AI 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 efficiently map ML models (with a particular focus on always-on computing use cases) to the hardware ISA.
  • Evaluate various trade-offs of different parallelization strategies such as performance, power, energy and memory consumption.
  • Collaborate with machine learning researchers to constantly improve the domain-specific compiler.
  • Collaborate with hardware engineers to evolve future accelerators.

Similar Jobs

ByteDance - Research Scientist Graduate (Foundation Model, Video Generation) - 2025 Start (PhD)

ByteDance

Seattle, Washington, United States (On-Site)
6 Months ago
Google - Tech Lead, Software Engineering, Black Community Inclusion

Google

Belo Horizonte, State Of Minas Gerais, Brazil (On-Site)
5 Months ago
ByteDance - Product Operations - Search Ads Team

ByteDance

Tokyo, Japan (On-Site)
1 Month ago
Netflix - Software Engineer 5 - Streaming Algorithms

Netflix

United States (Remote)
6 Months ago
Social Discovery Group - Head/Lead Social Media Manager

Social Discovery Group

Portugal (Remote)
6 Months ago
The Walt Disney Company - Ride Development Engineer Senior

The Walt Disney Company

Glendale, California, United States (On-Site)
1 Week ago
Google - Networking Design Verification Engineer

Google

Haifa, Haifa District, Israel (On-Site)
2 Days ago
NVIDIA - Senior System Power Management Engineer

NVIDIA

Santa Clara, California, United States (On-Site)
3 Weeks ago
NVIDIA - Senior GPU Memory Architect

NVIDIA

Canada (On-Site)
1 Month ago
ByteDance - Research Scientist, Data Management and Security - Infrastructure System Lab

ByteDance

San Jose, California, United States (On-Site)
2 Weeks ago

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

Varonis  - Cloud Security Researcher

Varonis

Herzliya, Tel Aviv District, Israel (On-Site)
6 Months ago
Every matrix - Senior Java Developer

Every matrix

Kyiv, Kyiv City, Ukraine (Hybrid)
1 Month ago
N-iX - Senior Scala Engineer

N-iX

Ukraine (Remote)
2 Weeks ago
ByteDance - Imaging Systems Lead - Smart Wearable

ByteDance

San Jose, California, United States (On-Site)
2 Weeks ago
ByteDance - Senior Algorithm Engineer, Trust and Safety

ByteDance

Singapore (On-Site)
6 Months ago
N-iX - 2D/3D Visualization Engineer

N-iX

Ukraine (Remote)
2 Months ago
Google - Software Engineer III, YouTube Trust and Safety

Google

Zürich, Zurich, Switzerland (On-Site)
2 Days ago
Luxoft - Regular Data Engineer

Luxoft

(Remote)
5 Months ago
Wargaming - Senior Engine Developer (World of Tanks)

Wargaming

Belgrade, Serbia (Hybrid)
2 Months ago
Loyalty Juggernaut - Mobile Application Developer (Android/iOS)

Loyalty Juggernaut

Hyderabad, Telangana, India (On-Site)
2 Days ago

Get notifed when new similar jobs are uploaded

Jobs in Bengaluru, Karnataka, India

Xentrix Studios - Modeling – Artist

Xentrix Studios

India (On-Site)
5 Months ago
ION - SOC Manager

ION

Noida, Uttar Pradesh, India (On-Site)
6 Months ago
Virtana Corp - Senior Software Engineer

Virtana Corp

Pune, Maharashtra, India (Remote)
6 Months ago
C3 AI - Solution Engineer

C3 AI

Bengaluru, Karnataka, India (On-Site)
6 Months ago
Enphase Energy - Staff Thermal Engineer

Enphase Energy

Bengaluru, Karnataka, India (On-Site)
2 Months ago
Single Store - Technical Account Manager

Single Store

Pune, Maharashtra, India (Remote)
1 Month ago
Krafton india - Manager - Compensation & Benefits

Krafton india

Bengaluru, Karnataka, India (On-Site)
1 Day ago
Google - Analyst, Process Excellence

Google

Hyderabad, Telangana, India (On-Site)
2 Weeks ago
Keywords Studios - Product Manager II

Keywords Studios

Pune, Maharashtra, India (Hybrid)
1 Month ago
Nagarro - Associate Staff Engineer, Javascript Fullstack

Nagarro

Gurugram, Haryana, India (On-Site)
6 Months ago

Get notifed when new similar jobs are uploaded

Research & Development Jobs

NVIDIA - CSP Hardware Application Engineer

NVIDIA

Shenzhen, Guangdong Province, China (On-Site)
2 Months ago
Google - Senior SoC Power Engineer

Google

Tel Aviv-Yafo, Tel Aviv District, Israel (On-Site)
2 Days ago
Meta - Research Scientist, Machine Learning (PhD)

Meta

Sunnyvale, California, United States (On-Site)
5 Months ago
Google - Cloud Software Engineer

Google

Tel Aviv-Yafo, Tel Aviv District, Israel (On-Site)
2 Days ago
NVIDIA - Senior Hardware Security Architect

NVIDIA

Be'er Sheva, South District, Israel (On-Site)
1 Month ago
NVIDIA - Senior Chip Design Engineer

NVIDIA

Yokne'am Illit, North District, Israel (On-Site)
3 Months ago
Riot Games - Technical Program Manager II - DevCon

Riot Games

Los Angeles, California, United States (On-Site)
1 Month ago
Rivos - Accelerator Design Verification - Full Time

Rivos

Bengaluru, Karnataka, India (Hybrid)
6 Months ago
NVIDIA - Tegra Software Engineer (RDSS Intern)

NVIDIA

Taipei City, Taiwan (On-Site)
3 Months ago
NVIDIA - Senior Mixed Signal Design Engineer

NVIDIA

Santa Clara, California, United States (On-Site)
4 Weeks 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