Software Engineer, ML Compilers, Silicon

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

NVIDIA - Signal and Power Integrity Engineer

NVIDIA

Taipei City, Taiwan (On-Site)
1 Month ago
Meta - Research Engineer (Robotics)

Meta

Menlo Park, California, United States (On-Site)
2 Days ago
Google - Senior Software Engineer, Chrome Sync Server

Google

Warsaw, Masovian Voivodeship, Poland (On-Site)
7 Hours ago
Google - Software Engineer III, Infrastructure, Google Cloud Networking

Google

Dublin, County Dublin, Ireland (On-Site)
8 Hours ago
Google - Senior Software Engineer, Dataform

Google

Warsaw, Masovian Voivodeship, Poland (On-Site)
9 Hours ago
NVIDIA - Solution Architect - CSP Cloud

NVIDIA

Shanghai, Shanghai, China (On-Site)
1 Month ago
ByteDance - Senior Machine Learning Ops Engineer, ML System - Foundation Model

ByteDance

San Jose, California, United States (On-Site)
2 Months ago
Google - Staff Software Engineer, Computational Photography and Generative AI

Google

Bengaluru, Karnataka, India (On-Site)
10 Hours ago
Samsung Semiconductor - Staff Engineer, DRAM Design

Samsung Semiconductor

San Jose, California, United States (On-Site)
3 Weeks ago
CloudHire - Sr. Java Application Architect

CloudHire

Karnataka, India (Remote)
3 Weeks ago

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

Singhania Buildcon Group | No 1 Real Estate Company - social media manager

Singhania Buildcon Group | No 1 Real Estate Company

Raipur, Chhattisgarh, India (On-Site)
5 Months ago
ByteDance - Software Engineer Intern, Information System - 2025 Start

ByteDance

Dubai, Dubai, United Arab Emirates (On-Site)
3 Weeks ago
Scanline VFX - Research Intern (Summer 2025)

Scanline VFX

Los Angeles, California, United States (Hybrid)
5 Months ago
Lucid Reality Labs - ML/AI Engineer

Lucid Reality Labs

Poland (Remote)
3 Weeks ago
Google - Software Engineer III, Infrastructure, Google Cloud

Google

Hyderabad, Telangana, India (On-Site)
5 Months ago
Trend Micro - Sr. Data Scientist (AI Lab)

Trend Micro

Taipei City, Taiwan (On-Site)
6 Months ago
Google - Software Engineer III, AI/ML Recommendations, Rankings, Predictions, Google Cloud

Google

Hyderabad, Telangana, India (On-Site)
8 Hours ago
Google - Software Developer, Chrome Enterprise

Google

Warsaw, Masovian Voivodeship, Poland (On-Site)
10 Hours ago
Google - Software Engineer II, Fitbit Health Platform

Google

Bucharest, Bucharest, Romania (On-Site)
11 Hours ago
NVIDIA - System Software Engineer - GPU

NVIDIA

Canada (On-Site)
6 Days ago

Get notifed when new similar jobs are uploaded

Jobs in Bengaluru, Karnataka, India

Blissclub - Garment Technologist

Blissclub

Bengaluru, Karnataka, India (On-Site)
6 Months ago
Alp Consulting  - Audio Multimedia

Alp Consulting

Karnataka, India (On-Site)
7 Months ago
Google - SoC Design/Integration Engineer, Silicon

Google

Bengaluru, Karnataka, India (On-Site)
8 Hours ago
Antarctica Global - Business Development Executive

Antarctica Global

Maharashtra, India (Hybrid)
2 Weeks ago
ByteDance - Content Moderator - Bengali Speaker

ByteDance

Mumbai, Maharashtra, India (On-Site)
2 Days ago
PwC - IN-Director_SAP Functional_SAP_Advisory_Bangalore

PwC

Gurugram, Haryana, India (On-Site)
6 Months ago
Google - Product Manager, Connector Platform

Google

Bengaluru, Karnataka, India (On-Site)
10 Hours ago
Google - Partner and Campaigns Lead, Business Development Representative

Google

Bengaluru, Karnataka, India (On-Site)
8 Hours ago
Sporty Group - IN Associate - Payment Operations

Sporty Group

Mumbai, Maharashtra, India (On-Site)
3 Weeks ago
Warner Bros Games - Senior Software Engineer

Warner Bros Games

Hyderabad, Telangana, India (Hybrid)
1 Month ago

Get notifed when new similar jobs are uploaded

Research & Development Jobs

Epic Games - Principal Programmer, Language Engineering

Epic Games

United States (On-Site)
3 Months ago
NVIDIA - Senior Software Engineer

NVIDIA

Tel Aviv-Yafo, Tel Aviv District, Israel (On-Site)
1 Month ago
NVIDIA - Senior Physical Design Verification Layout Engineer

NVIDIA

Yokne'am Illit, North District, Israel (On-Site)
2 Months ago
NVIDIA - Senior GPU Kernel Performance Lead

NVIDIA

Santa Clara, California, United States (On-Site)
2 Months ago
NVIDIA - Senior CUDA Compute Systems Software Engineer

NVIDIA

Santa Clara, California, United States (On-Site)
2 Months ago
NVIDIA - Power Integrity Engineer

NVIDIA

Tel Aviv-Yafo, Tel Aviv District, Israel (On-Site)
1 Month ago
NVIDIA - Senior Math Libraries Engineers - Python APIs

NVIDIA

Louisiana, United States (Remote)
1 Month ago
NVIDIA - Senior Synthesis Flow CAD Engineer

NVIDIA

Santa Clara, California, United States (On-Site)
2 Months ago
Krafton  - HR Planning Specialist (5+ years)

Krafton

Seoul, South Korea (On-Site)
1 Day ago
Hello Games - Machine Learning Engineer

Hello Games

United Kingdom (On-Site)
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