Software Engineer, Machine Learning Compilers

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

Google - Senior Software Engineer, CPU Performance Modeling Engineer

Google

Tel Aviv-Yafo, Tel Aviv District, Israel (On-Site)
4 Days ago
Google - Software Developer III, Google Cloud Business Platforms

Google

Waterloo, Ontario, Canada (On-Site)
1 Week ago
Enphase Energy - Assistant Manager, Business Analytics

Enphase Energy

Bengaluru, Karnataka, India (On-Site)
4 Months ago
Google - Software Engineer III, Google Cloud Compute Infrastructure

Google

Seattle, Washington, United States (On-Site)
5 Months ago
Epic Games - Senior Rendering Engineer

Epic Games

London, England, United Kingdom (On-Site)
1 Week ago
Meta - Software Engineer, Systems ML - SW/HW Co-design

Meta

Austin, Texas, United States (On-Site)
5 Months ago
Inkittt - Director of AI

Inkittt

San Francisco, California, United States (On-Site)
8 Months ago
NVIDIA - Software Engineer Intern, Perception - Autonomous Vehicles - 2025

NVIDIA

Shanghai, Shanghai, China (On-Site)
3 Months ago
ByteDance - Student Researcher (Doubao (Seed) - Foundation Model - Vision Generative AI)

ByteDance

San Jose, California, United States (On-Site)
1 Week ago
NVIDIA - Senior Deep Learning Research Engineer, Advanced AI Systems

NVIDIA

Santa Clara, California, United States (On-Site)
1 Month ago

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

Google - Software Engineer II, Mobile, Platforms and Devices

Google

Warsaw, Masovian Voivodeship, Poland (On-Site)
6 Days ago
Trend Micro - (Sr.) Data Engineer/AI Trainer

Trend Micro

Taipei City, Taiwan (On-Site)
6 Months ago
Ubisoft - Intermediate/Senior Generalist Engine Programmer

Ubisoft

Malmö, Skåne County, Sweden (Hybrid)
1 Month ago
Google - Software Engineer III, Google Cloud

Google

Bengaluru, Karnataka, India (On-Site)
1 Week ago
Google - Software Engineer II, Site Reliability Engineering, Cloud Logs

Google

Warsaw, Masovian Voivodeship, Poland (On-Site)
1 Week ago
Hitachi - Data Scientist

Hitachi

Pune, Maharashtra, India (Remote)
6 Months ago
Trendyol - Data Science Professionals - Trendyol GO

Trendyol

İzmir, İzmir, Türkiye (Hybrid)
5 Months ago
Google - Senior Software Engineer, Multiplatform, Mobile

Google

Mexico City, Mexico City, Mexico (On-Site)
1 Week ago
Google - Software Engineer III, Infrastructure, Google Cloud Compute Infrastructure

Google

Sunnyvale, California, United States (On-Site)
1 Week ago
NVIDIA - Web Software Development Intern - 2025

NVIDIA

Shanghai, Shanghai, China (On-Site)
3 Months ago

Get notifed when new similar jobs are uploaded

Jobs in Bengaluru, Karnataka, India

Cyara - Sales Operations Analyst – Data

Cyara

Hyderabad, Telangana, India (Hybrid)
4 Months ago
House Sparrow Films - Motion Designer and 2D Animator

House Sparrow Films

Bengaluru, Karnataka, India (On-Site)
9 Months ago
Syniverse - QA Automation Engineer

Syniverse

Bengaluru, Karnataka, India (Hybrid)
5 Months ago
Cognite - Senior Data Scientist

Cognite

Bengaluru, Karnataka, India (Hybrid)
6 Months ago
PwC - "Senior Associate - Bangalore The Millenia - Technology Consulting"

PwC

Bengaluru, Karnataka, India (On-Site)
6 Months ago
Google - Engineering Manager, Networking

Google

Bengaluru, Karnataka, India (On-Site)
1 Week ago
Nagarro - Staff Consultant ,SAP Basis

Nagarro

India (Remote)
6 Months ago
STAGE - Graphic Designer Intern

STAGE

Noida, Uttar Pradesh, India (On-Site)
1 Month ago
Google - Digital Marketing Strategist

Google

Hyderabad, Telangana, India (On-Site)
6 Days ago
Level AI - Principal Software Engineer

Level AI

Noida, Uttar Pradesh, India (Hybrid)
6 Months ago

Get notifed when new similar jobs are uploaded

Artificial Intelligence Jobs

Bally's Interactive - Machine Learning Developer

Bally's Interactive

Toronto, Ontario, Canada (On-Site)
1 Week ago
Google - Developer Relations Engineer, AI Developer, Cloud AI

Google

Sunnyvale, California, United States (On-Site)
6 Days ago
Google - Senior Photonics Engineer, Platforms Infrastructure

Google

Sunnyvale, California, United States (On-Site)
1 Week ago
Microsoft - Member of Technical Staff - AI Multimodal

Microsoft

Zürich, Zurich, Switzerland (On-Site)
1 Week ago
NVIDIA - Senior Software Engineer, TensorRT-LLM

NVIDIA

California, United States (Hybrid)
3 Weeks ago
Google - Software Engineering Manager, AICore, Applied ML

Google

Taipei City, Taiwan (On-Site)
1 Week ago
Google - Senior Software Engineer, AI/ML, Google Cloud

Google

Hyderabad, Telangana, India (On-Site)
6 Days ago
Google - Field Solutions Architect, Generative AI, Google Cloud

Google

Madrid, Community Of Madrid, Spain (On-Site)
1 Week ago
Microsoft - Member of Technical Staff, Platform Engineer

Microsoft

Redmond, Washington, United States (Hybrid)
1 Week ago
Google - Conversational AI Strategic Cloud Engineer

Google

Bengaluru, Karnataka, India (On-Site)
1 Week 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.

Fremont, California, United States (On-Site)

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

Bengaluru, Karnataka, India (On-Site)

Dublin, County Dublin, Ireland (On-Site)

Atlanta, Georgia, United States (On-Site)

Ho Chi Minh City, Ho Chi Minh City, Vietnam (On-Site)

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