Software Engineer, ML Compilers, Silicon

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

Level AI - Senior Machine Learning Engineer

Level AI

Noida, Uttar Pradesh, India (On-Site)
1 Month ago
Index Exchange - Backend Engineer, Data Products

Index Exchange

Bengaluru, Karnataka, India (On-Site)
8 Months ago
Google - Senior Software Engineer, Core, Access Intelligence

Google

São Paulo, State Of São Paulo, Brazil (On-Site)
1 Month ago
NVIDIA - Senior Full-Stack Web Applications Software Engineer

NVIDIA

Shanghai, Shanghai, China (On-Site)
1 Month ago
Google - Senior Software Engineer, Android, Find My Device

Google

Bengaluru, Karnataka, India (On-Site)
1 Month ago
Google - Software Engineering Manager, Chrome Sync Server

Google

Warsaw, Masovian Voivodeship, Poland (On-Site)
1 Month ago
NVIDIA - GPU Firmware Engineer (RDSS Intern)

NVIDIA

Taipei City, Taiwan (On-Site)
4 Months ago
Rivos - Logic Equivalence Check (LEC) Engineer

Rivos

Hsinchu, Hsinchu City, Taiwan (Hybrid)
7 Months ago
Google - Physical Design CAD Engineer, Silicon

Google

Bengaluru, Karnataka, India (On-Site)
1 Month ago
Google - Senior Software Engineer, Machine Learning, Payments

Google

Bengaluru, Karnataka, India (On-Site)
1 Month ago

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

ByteDance - Software Development Engineer in Test Graduate

ByteDance

Seattle, Washington, United States (On-Site)
1 Month ago
Inkittt - Lead Product Manager - GTV

Inkittt

San Francisco, California, United States (On-Site)
10 Months ago
Blind Squirrel Games - Senior Graphics Engineer

Blind Squirrel Games

Austin, Texas, United States (Hybrid)
7 Months ago
Epic Games - Rendering Programmer

Epic Games

Montreal, Quebec, Canada (On-Site)
5 Months ago
NVIDIA - Senior Applied LLM Engineer, AI – Chip Design

NVIDIA

Santa Clara, California, United States (On-Site)
4 Months ago
Google - Software Engineer, Site Reliability Engineering, Campus

Google

Sydney, New South Wales, Australia (On-Site)
1 Month ago
Qualcomm - Sr Lead Engineer - Audio DSP

Qualcomm

Hyderabad, Telangana, India (On-Site)
1 Month ago
Epic Games - Automation Engineer

Epic Games

(On-Site)
2 Months ago
Google - Senior Software Engineer, Recommendations, Ranking and Prediction, Google Workspace

Google

Sunnyvale, California, United States (On-Site)
1 Month ago
NVIDIA - Senior AI-HPC Storage Engineer

NVIDIA

Austin, Texas, United States (On-Site)
3 Months ago

Get notifed when new similar jobs are uploaded

Jobs in Bengaluru, Karnataka, India

FICO - Cyber Security Engineer II

FICO

Bengaluru, Karnataka, India (On-Site)
3 Weeks ago
seeking alpha - Lead Product Designer

seeking alpha

Bengaluru, Karnataka, India (On-Site)
1 Month ago
Diligent - Principal Software Engineer

Diligent

Bengaluru, Karnataka, India (On-Site)
1 Month ago
PwC - Consultant - RDC TC MSOFT

PwC

Kolkata, West Bengal, India (On-Site)
8 Months ago
Nagarro - Staff Engineer, NodeJS

Nagarro

India (Remote)
7 Months ago
PwC - IN-Associate_Full Stack Developer(Node JS)_MS Engg_Advisory_Kolkata

PwC

Kolkata, West Bengal, India (On-Site)
7 Months ago
Capgemini - Full Stack Developer (Python)

Capgemini

Bengaluru, Karnataka, India (On-Site)
1 Month ago
Fortanix - Staff Software Engineer

Fortanix

Bengaluru, Karnataka, India (Hybrid)
4 Months ago
Aristocrat Gaming - Senior Animator I (Spine)

Aristocrat Gaming

Gurugram, Haryana, India (Hybrid)
2 Months ago
Cyara - Senior Software Engineer - CCaaS Integration

Cyara

Hyderabad, Telangana, India (Hybrid)
11 Months ago

Get notifed when new similar jobs are uploaded

Research & Development Jobs

Easygo - Software Development Engineer, Payments & Fraud

Easygo

Melbourne, Victoria, Australia (On-Site)
6 Months ago
NVIDIA - Senior Developer Technology Engineer, Public Sector

NVIDIA

California, Maryland, United States (Remote)
2 Months ago
NVIDIA - HPC Operations Manager – Hardware Engineering

NVIDIA

Santa Clara, California, United States (On-Site)
4 Months ago
Google - Near Eye Display Metrology Engineer

Google

Mountain View, California, United States (On-Site)
1 Month ago
Tesla - Senior Construction Engineer

Tesla

Prüm, Rhineland-Palatinate, Germany (On-Site)
4 Months ago
NXP - Senior Principal Software Architect - Platform and RF Software

NXP

Bucharest, Bucharest, Romania (On-Site)
8 Months ago
Bright Machines - Research Perception Scientist, Computer Vision and Machine Learning

Bright Machines

San Francisco, California, United States (On-Site)
7 Months ago
Google - Senior Staff Software Engineer, Google Cloud

Google

Hyderabad, Telangana, India (On-Site)
7 Months ago
Google - Silicon Networking Microarchitecture and RTL Lead

Google

Bengaluru, Karnataka, India (On-Site)
1 Month ago
Google - Senior Software Engineer, YouTube

Google

Bengaluru, Karnataka, India (On-Site)
1 Month ago

Get notifed when new similar jobs are uploaded

About The Company

Regensburg, Bavaria, Germany (Remote)

Lanham, Maryland, United States (On-Site)

Toronto, Ontario, Canada (On-Site)

London, England, United Kingdom (On-Site)

Romania (On-Site)

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

Bengaluru, Karnataka, India (On-Site)

Taipei City, Taiwan (On-Site)

Kirkland, 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