Software Engineer, Compiler Frontend, Silicon

1 Week ago • 2-4 Years • Artificial Intelligence • $141,000 PA - $202,000 PA

Job Summary

Job Description

This role involves working on the core optimization and scheduling algorithms of a next-generation MLIR-based compiler for Google's Edge TPU. Responsibilities include developing operator fusions, parallelization, and scheduling algorithms to optimize AI/ML workloads. The engineer will work on efficient mapping of generative AI models to TPU instructions and collaborate with ML developers, researchers, and hardware/software teams. The goal is to accelerate the transition from research to exceptional user experiences on the Edge TPU, impacting the performance of AI/ML applications on Google devices. The position requires experience with compiler optimization, parallelization, and data structures/algorithms, along with proficiency in C++ or Python.
Must have:
  • Bachelor's degree or equivalent experience
  • 2+ years software development (C++) or 1 year with advanced degree
  • 2+ years experience with data structures/algorithms
  • 2+ years experience with compilers (optimizations, parallelization)
  • Compiler optimization and scheduling algorithms development
Good to have:
  • Master's/PhD in Computer Science
  • MLIR or LLVM experience
  • Compiler development for accelerator-based architectures
  • Machine learning and generative AI architecture experience
  • C++ or Python proficiency
Perks:
  • Bonus
  • Equity
  • Benefits

Job Details

Minimum qualifications:

  • Bachelor’s degree or equivalent practical experience.
  • 2 years of experience with software development in one or more programming languages (e.g., C++), or 1 year of experience with an advanced degree.
  • 2 years of experience with data structures or algorithms.
  • 2 years of experience working with compilers (e.g., compiler optimizations, parallelization, etc.).

Preferred qualifications:

  • Master's degree or PhD in Computer Science or related technical fields.
  • Experience in Multi-Level Intermediate Representation (MLIR) or Low Level Virtual Machine (LLVM).
  • Experience with compiler development in the context of accelerator-based architectures.
  • Experience with machine learning and generative AI architectures.
  • Experience with C++ or Python.

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.

In this role, you will work on the core optimization and scheduling algorithms of the next-generation compiler featuring a Multi-Level Intermediate Representation (MLIR) based compiler framework serving both production and hardware/software code sign needs for Edge Tensor Processing Unit (TPU), the core algorithms are critical to deliver the best Artificial Intelligence/Machine Learning (AI/ML) performance of the silicon to our end users. You will design and implement optimization algorithms and compiler transformation frameworks to enable production-quality compilation and optimization of workloads. You will be able to make an impact on hardware/software co-optimization of AI/ML use cases for current and future generations of Google devices, including the latest Generative AI (GenAI) models.

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.

The US base salary range for this full-time position is $141,000-$202,000 + bonus + equity + benefits. Our salary ranges are determined by role, level, and location. Within the range, individual pay is determined by work location and additional factors, including job-related skills, experience, and relevant education or training. Your recruiter can share more about the specific salary range for your preferred location during the hiring process.

Please note that the compensation details listed in US role postings reflect the base salary only, and do not include bonus, equity, or benefits. Learn more about .

Responsibilities

  • Work as part of the Edge Tensor Processing Unit (TPU) compiler team, including analyzing and improving the compiler quality and performance on optimization decisions, correctness, and compilation time.
  • Develop operator fusions, parallelization, and scheduling algorithms to optimize compute and data movement costs to execute Machine Learning workloads on the Edge Tensor Processing Unit (TPU).
  • Work on efficient mapping of generative AI models and other key workloads into Edge Tensor Processing Unit (TPU) instructions through the compiler.
  • Collaborate with Machine Learning model developers, researchers, and Edge Tensor Processing Unit (TPU) hardware/software teams to accelerate the transition from research ideas to exceptional user experiences running on the Edge Tensor Processing Unit (TPU).

Similar Jobs

Google - Embedded Senior Software Engineer, Pixel Power Software

Google

Kraków, Lesser Poland Voivodeship, Poland (On-Site)
1 Week ago
Google - Staff Software Engineer, Site Reliability Engineering

Google

Dublin, County Dublin, Ireland (On-Site)
1 Week ago
Altagram Group - Data Science Internship/Work Student

Altagram Group

Germany (On-Site)
3 Months ago
Mindtickle - SDE III - COE-ML

Mindtickle

Bengaluru, Karnataka, India (Hybrid)
1 Day ago
Wisk - Radar Staff Software Engineer

Wisk

Mountain View, California, United States (Hybrid)
1 Day ago
Interface AI - Engineering Manager

Interface AI

Hyderabad, Telangana, India (Remote)
5 Months ago
Microsoft - Member of Technical Staff, AI Pretraining Platform

Microsoft

London, England, United Kingdom (On-Site)
1 Week ago
Google - Senior Research Engineer, AI/ML

Google

London, England, United Kingdom (On-Site)
1 Week ago
ByteDance - Research Scientist Intern (Doubao (Seed) - Machine Learning System) - 2025 Summer (PhD)

ByteDance

Seattle, Washington, United States (On-Site)
6 Months ago
NVIDIA - Deep Learning Intern - Fall 2025

NVIDIA

Shanghai, Shanghai, China (On-Site)
2 Weeks ago

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

Build A Rocket Boy - Senior Graphics Programmer

Build A Rocket Boy

(Remote)
1 Month ago
CynLr - Software Engineer - GPU performance

CynLr

Bengaluru, Karnataka, India (On-Site)
6 Months ago
Google - Software Engineering Manager II, Site Reliability Engineering

Google

San Bruno, California, United States (On-Site)
2 Days ago
Hedra - Machine Learning Engineer (CUDA)

Hedra

New York, New York, United States (On-Site)
1 Month ago
ByteDance - Senior Network Engineer- IAAS- San Jose

ByteDance

San Jose, California, United States (On-Site)
5 Months ago
Bragg - AI/ML Engineer

Bragg

Ljubljana, Ljubljana, Slovenia (Hybrid)
2 Weeks ago
Google - Senior Software Engineer, Google Ads

Google

Mountain View, California, United States (On-Site)
2 Days ago
Veeam Software - Junior C++ Developer

Veeam Software

(Remote)
7 Hours ago
NVIDIA - Senior Math Libraries Engineer - Dense Linear Algebra

NVIDIA

California, United States (Hybrid)
3 Months ago
Google - Staff Software Engineer, Core, Dagger and TikTok

Google

Mexico City, Mexico City, Mexico (On-Site)
2 Weeks ago

Get notifed when new similar jobs are uploaded

Jobs in Mountain View, California, United States

Google - Strategic Program Delivery Lead, Content and AI

Google

Austin, Texas, United States (On-Site)
1 Week ago
Google - Customer Engineering Manager III, API and Integration

Google

Sunnyvale, California, United States (On-Site)
2 Weeks ago
Upwork - Senior Workplace Specialist

Upwork

Palo Alto, California, United States (On-Site)
18 Hours ago
Next Level Business Services - ERP Test Lead

Next Level Business Services

Denver, Colorado, United States (On-Site)
6 Months ago
Valve corporation - Animator

Valve corporation

Bellevue, Washington, United States (On-Site)
6 Months ago
Google - Software Engineer III

Google

Mountain View, California, United States (On-Site)
2 Days ago
Infosys - Lead Java Fullstack Developer

Infosys

Atlanta, Georgia, United States (On-Site)
23 Hours ago
ByteDance - Senior Software Engineer, Multi Cloud CDN

ByteDance

San Jose, California, United States (On-Site)
3 Days ago
Scale AI - Forward Deployed Engineer - GenAI Quality

Scale AI

San Francisco, California, United States (Hybrid)
1 Day ago
Baton - Software Engineer - Backend

Baton

San Francisco, California, United States (Hybrid)
1 Day ago

Get notifed when new similar jobs are uploaded

Artificial Intelligence Jobs

Google - Conversational AI Consultant

Google

Gurugram, Haryana, India (On-Site)
2 Weeks ago
NVIDIA - Senior Deep Learning Engineer

NVIDIA

Redmond, Washington, United States (On-Site)
1 Month ago
Google - Senior Software Engineer, Generative AI

Google

Sunnyvale, California, United States (On-Site)
2 Weeks ago
Meta - Research Scientist, Machine Learning (PhD)

Meta

Sunnyvale, California, United States (On-Site)
5 Months ago
Zoox - Senior Technical Program Manager, Service Readiness

Zoox

Foster City, California, United States (On-Site)
6 Months ago
Zoox - Staff/Senior Staff Software Engineer, ML Performance Optimization

Zoox

Foster City, California, United States (On-Site)
6 Months ago
Enterprise Bot - Data Scientist

Enterprise Bot

Bengaluru, Karnataka, India (On-Site)
5 Months ago
Google - Software Engineer III, AI/ML, Core

Google

Sunnyvale, California, United States (On-Site)
2 Weeks ago
Mashgin - Senior Software Engineer, Computer Vision and Deep Learning

Mashgin

Palo Alto, California, United States (Hybrid)
6 Months ago
Google - Staff Software Engineer, Google Cloud Compute Infrastructure

Google

Seattle, Washington, United States (On-Site)
2 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