Software Engineer, Machine Learning Compilers

8 Hours ago • 1 Years + • Artificial Intelligence

Job Summary

Job Description

As a Software Engineer, you will work on developing Machine Learning (ML) compilers for the Tensor TPU to accelerate Generative AI and other machine learning models running on custom hardware accelerators. You will manage project priorities, deadlines, and deliverables. 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 to improve the compiler, and collaborating with hardware engineers to evolve future accelerators. The role requires expertise in C++, data structures, algorithms, and ideally experience with power/performance optimizations, machine learning, compilers, and hardware parallelism.
Must have:
  • Bachelor's degree or equivalent experience
  • 1 year C++ experience
  • Data structures and algorithms expertise
  • Build ML compilers for Tensor TPU
  • Evaluate parallelization strategies
Good to have:
  • Master's degree in Computer Science
  • Power & performance optimization experience
  • Machine learning experience
  • Compiler experience
  • Hardware understanding (parallelism)

Job Details

Minimum qualifications:

  • Bachelor's degree or equivalent practical experience.
  • 1 year of experience in C++.
  • Experience in data structures and algorithms.

Preferred qualifications:

  • Master's degree in computer science or equivalent practical experience.
  • Experience in power and performance optimizations.
  • Experience with machine learning.
  • Experience with compilers.
  • Understanding of hardware, especially 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 and other machine learning models running on custom hardware accelerators. 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 Machine Learning 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

Social Discovery Group - Senior NLP Engineer

Social Discovery Group

(Remote)
3 Months ago
ByteDance - Search Operation Quality Assurance - Ads

ByteDance

Pasig, Metro Manila, Philippines (On-Site)
5 Months ago
The Walt Disney Company - Lead Software Engineer

The Walt Disney Company

Glendale, California, United States (On-Site)
2 Days ago
ByteDance - ML Systems Software Engineer Graduate (AML - Machine Learning Systems)

ByteDance

San Jose, California, United States (On-Site)
2 Days ago
Inkittt - Senior Data Engineer (m/f/d)

Inkittt

Krakow Am See, Mecklenburg-Vorpommern, Germany (Hybrid)
4 Months ago
Google - Senior Risk and Compliance Lead, AI and Content

Google

Washington, District Of Columbia, United States (On-Site)
10 Hours ago
Meta - Research Engineer, SysML - FAIR

Meta

Boston, Massachusetts, United States (Remote)
11 Hours ago
Virtuos - Machine Learning Engineer

Virtuos

Singapore (On-Site)
3 Weeks ago
Zoox - Senior Software Engineer - High Performance Computing

Zoox

Foster City, California, United States (Hybrid)
6 Months ago
SparkCognition - Data Scientist

SparkCognition

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

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

Dream11 - Lead ML Scientist

Dream11

Mumbai, Maharashtra, India (On-Site)
6 Months ago
The Walt Disney Company - Software Engineer, Tools Internals (Core)

The Walt Disney Company

Emeryville, California, United States (On-Site)
3 Weeks ago
Toptracer - Software Engineer in Test (SDET)

Toptracer

Stockholm, Stockholm County, Sweden (Hybrid)
3 Months ago
Epic Games - Rendering Programmer

Epic Games

Montreal, Quebec, Canada (On-Site)
3 Months ago
NVIDIA - Deep Learning Performance Architect

NVIDIA

Shanghai, Shanghai, China (On-Site)
2 Months ago
CloudHire - Senior Data Analyst

CloudHire

Maharashtra, India (Remote)
4 Weeks ago
Inkittt - VP of Operations

Inkittt

San Francisco, California, United States (Hybrid)
4 Months ago
Wargaming - Senior Engine Developer (World of Tanks)

Wargaming

Vilnius, Vilnius County, Lithuania (Hybrid)
1 Month ago
Airlab Inc  - Senior Lead Programmer (Game Industry)

Airlab Inc

Quebec, Canada (On-Site)
3 Weeks ago
Google - Software Engineer, Cloud Networking, Cloud DNS for GKE

Google

Warsaw, Masovian Voivodeship, Poland (On-Site)
9 Hours ago

Get notifed when new similar jobs are uploaded

Jobs in Bengaluru, Karnataka, India

Dream Sports - Product Manager (Home)

Dream Sports

Mumbai, Maharashtra, India (On-Site)
3 Weeks ago
Google - Software Engineer III, Android Framework

Google

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

Sporty Group

Mumbai, Maharashtra, India (On-Site)
3 Weeks ago
CloudHire - Wordpress Full Stack Developer

CloudHire

Haryana, India (Remote)
4 Weeks ago
Adobe - Technical Account Manager

Adobe

India (Remote)
2 Weeks ago
CloudHire - Full Stack - Angular IONIC Developer

CloudHire

India (Remote)
4 Weeks ago
Google - Video Solutions Consultant

Google

Gurugram, Haryana, India (On-Site)
10 Hours ago
NVIDIA - Senior System Software Engineer

NVIDIA

Bengaluru, Karnataka, India (On-Site)
1 Month ago
Bajaj Finserv - Manager - CD - Growth North & East

Bajaj Finserv

Bhagalpur, Bihar, India (On-Site)
7 Months ago
Bounteous - Senior Braze Developer

Bounteous

Chennai, Tamil Nadu, India (Hybrid)
6 Months ago

Get notifed when new similar jobs are uploaded

Artificial Intelligence Jobs

Google - Head of Generative AI Specialist, Blackbelt

Google

Hyderabad, Telangana, India (On-Site)
11 Hours ago
Tencent - AI Technical Project Manager

Tencent

London, England, United Kingdom (On-Site)
1 Month ago
Ubisoft - Senior ML Data Scientist

Ubisoft

Montreal, Quebec, Canada (On-Site)
3 Months ago
ByteDance - Research Engineer Graduate (Vision AI Platform)

ByteDance

San Jose, California, United States (On-Site)
2 Months ago
Google - Software Engineering Manager, RDMA Networking

Google

Sunnyvale, California, United States (On-Site)
11 Hours ago
SmileGate - AI Developer ([LOST ARK Mobile])

SmileGate

Seongnam-si, Gyeonggi-do, South Korea (On-Site)
3 Months ago
NVIDIA - Senior Deep Learning Performance Architect

NVIDIA

Redmond, Washington, United States (On-Site)
1 Month ago
Lionbridge Games - Games Language AI Specialist (Linguist)

Lionbridge Games

Masovian Voivodeship, Poland (On-Site)
2 Days ago
Airlab Inc  - Junior Programmer Artificial Intelligence

Airlab Inc

Quebec, Canada (On-Site)
3 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.

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