Software Engineer, Compiler, Scheduling and Optimization, Silicon

2 Weeks 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 EdgeTPU. Responsibilities include developing parallelization and scheduling algorithms, efficient mapping of generative AI models, designing the hardware/software interface, and collaborating with architects and product managers. The engineer will analyze and improve compiler quality, performance, and compilation time, focusing on optimizing compute and data movement costs for machine learning workloads. This position requires experience with compiler optimizations, parallelization, data structures, and algorithms, preferably with MLIR or LLVM. The work will impact hardware/software co-optimization of AI/ML use cases for current and future Google devices.
Must have:
  • Bachelor's degree or equivalent experience
  • 2+ years software development experience (C++)
  • 2+ years experience with data structures/algorithms
  • 2+ years experience with compilers
  • Develop parallelization and scheduling algorithms
Good to have:
  • Master's/PhD in Computer Science
  • MLIR/LLVM experience
  • Compiler development for accelerator-based architectures
  • Machine learning/generative AI architecture experience
  • C++ or Python experience

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 (compiler optimizations, parallelization, etc.).

Preferred qualifications:

  • Master's degree or PhD in Computer Science or a 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 in developing accessible technologies.
  • 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 an Multi-Level Intermediate Representation (MLIR)-based compiler framework serving both production and hardware/software codesign needs for EdgeTPU, 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 EdgeTPU compiler team, including analyzing and improving the compiler quality and performance on optimization decisions, correctness, and compilation time.
  • Develop parallelization and scheduling algorithms to optimize compute and data movement costs to execute Machine Learning workloads on the EdgeTPU.
  • Work on efficient mapping of generative AI models and other workloads into EdgeTPU instructions through the compiler.
  • Work with EdgeTPU architects to design the hardware/software interface and co-optimizations of the next generation EdgeTPU architectures.
  • Work with product managers, researchers in identifying Machine Learning trends, future use cases, etc.

Similar Jobs

Genies - Lead Machine Learning Engineer, 3D Gen AI & Graphics

Genies

Los Angeles, California, United States (On-Site)
2 Months ago
PlayStation Global - Staff Machine Learning Engineer, Anomaly Detection

PlayStation Global

London, England, United Kingdom (Hybrid)
5 Months ago
NVIDIA - Senior High-Performance System Architect

NVIDIA

Tel Aviv-Yafo, Tel Aviv District, Israel (On-Site)
1 Month ago
Google - Staff Software Engineer, Storage

Google

Sunnyvale, California, United States (On-Site)
2 Days ago
Socure - Data Science Intern

Socure

(Remote)
1 Day ago
ByteDance - Senior Software Engineer / Researcher, AI-Native Database Systems

ByteDance

San Jose, California, United States (On-Site)
3 Days ago
Lionbridge Games - Language AI Specialist (Test & Tech)

Lionbridge Games

Masovian Voivodeship, Poland (On-Site)
2 Months ago
NetEase Games - Game AI Research Leader

NetEase Games

Singapore (On-Site)
2 Months ago
Genies - Machine Learning Infrastructure Engineer, 3D Model Inference & Deployment

Genies

Los Angeles, California, United States (On-Site)
2 Months ago
Meta - Software Engineer, Machine Learning

Meta

Los Angeles, California, United States (On-Site)
5 Months ago

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

Google - Software Engineer III, Infrastructure, Google Cloud Business Platforms

Google

Seattle, Washington, United States (On-Site)
1 Week ago
Whoop - Senior Sensor Algorithm Engineer

Whoop

Boston, Massachusetts, United States (On-Site)
23 Hours ago
sony global (Games) - Software Design Engineer

sony global (Games)

Shanghai, China (On-Site)
23 Hours ago
Google - Product Manager, Content Discovery

Google

Bengaluru, Karnataka, India (On-Site)
2 Days ago
Absurd Ventures - Principal Rendering Engineer

Absurd Ventures

Santa Monica, California, United States (On-Site)
20 Hours ago
Google - Senior Software Engineer, Infrastructure, Google Ads

Google

Mountain View, California, United States (On-Site)
1 Week ago
Google - Staff Software Engineer, Android Content

Google

Bengaluru, Karnataka, India (On-Site)
2 Weeks ago
Google - Software Engineer III, Research

Google

Tel Aviv-Yafo, Tel Aviv District, Israel (On-Site)
2 Weeks ago
N-iX - Senior C++ Engineer (High Performance Computing)

N-iX

Colombia (Remote)
2 Weeks ago
Axon - DSP Engineer

Axon

Kassel, Hessen, Germany (On-Site)
7 Hours ago

Get notifed when new similar jobs are uploaded

Jobs in Mountain View, California, United States

The Walt Disney Company - Manager, Product Design (Disney +)

The Walt Disney Company

Seattle, Washington, United States (On-Site)
5 Months ago
ByteDance - Benefits Program Manager

ByteDance

Los Angeles, California, United States (On-Site)
1 Month ago
Epic Games - Data Analyst - CorpTech Analytics

Epic Games

Cary, North Carolina, United States (On-Site)
1 Month ago
Whatnot - Software Engineer, Developer Tooling

Whatnot

San Francisco, California, United States (On-Site)
1 Day ago
Netflix - Product Manager, Connected Data

Netflix

Los Gatos, California, United States (Remote)
2 Weeks ago
Crunchyroll - Vice President, COO Initiatives

Crunchyroll

Culver City, California, United States (Hybrid)
2 Months ago
Haleon - Occupational Health Nurse

Haleon

New York, United States (On-Site)
5 Hours ago
Scout - Specialist, Parts Planning

Scout

Novi, Michigan, United States (On-Site)
6 Hours ago
Google - Senior Staff Research Scientist, Google Cloud AI

Google

San Francisco, California, United States (On-Site)
2 Weeks ago
Fashionphile - Luxury Sales Specialist

Fashionphile

Newport Beach, California, United States (On-Site)
3 Weeks ago

Get notifed when new similar jobs are uploaded

Artificial Intelligence Jobs

ByteDance - Software Engineer (Applied Machine Learning - Enterprise)

ByteDance

San Jose, California, United States (On-Site)
2 Weeks ago
Microsoft - Senior Principal Researcher – Generative AI

Microsoft

Redmond, Washington, United States (On-Site)
2 Weeks ago
Cornerstone OnDemand - Principal Data Scientist

Cornerstone OnDemand

Pune, Maharashtra, India (Hybrid)
7 Months ago
ByteDance - Research Scientist Graduate (Foundation Model Speech & Audio Generation)

ByteDance

Seattle, Washington, United States (On-Site)
3 Days ago
Google - Senior Software Engineer, GPU Performance, Google Scale

Google

Sunnyvale, California, United States (On-Site)
2 Days ago
Meta - Research Scientist Intern, Machine Perception for Input and Interaction (PhD)

Meta

Burlingame, California, United States (On-Site)
5 Months ago
Zoox - Senior/Staff Machine Learning Engineer - Prediction & Behavior ML

Zoox

Foster City, California, United States (Hybrid)
6 Months ago
Meta - Research Scientist Intern, Smart Glasses in Wearables AI (PhD)

Meta

Washington, District Of Columbia, United States (On-Site)
5 Months ago
ByteDance - Senior Software Engineer / Researcher, AI-Native Database Systems

ByteDance

San Jose, California, United States (On-Site)
3 Days ago
Google - Student Researcher, BS/MS, Winter/Summer 2025

Google

Mountain View, California, United States (On-Site)
5 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.

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