Software Engineer, Compiler, Scheduling and Optimization, Silicon

2 Months ago • 2-4 Years • $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

ByteDance - Backend Software Engineer

ByteDance

Dubai, Dubai, United Arab Emirates (On-Site)
2 Months ago
Argus Labs - Software Engineer (Infrastructure/Backend)

Argus Labs

(Remote)
3 Months ago
Google - Software Engineer II, Education and Activation

Google

Mexico City, Mexico City, Mexico (On-Site)
2 Months ago
Loyalty Juggernaut - Frontend Developer (Angular)

Loyalty Juggernaut

Hyderabad, Telangana, India (On-Site)
2 Months ago
Glean - SEO Manager

Glean

Bengaluru, Karnataka, India (On-Site)
1 Month ago
Dolby Laboratories - Senior Foundational AI Researcher

Dolby Laboratories

Bengaluru, Karnataka, India (Hybrid)
9 Months ago
Keywords Studios - Technical Research Associate - AI

Keywords Studios

(Remote)
3 Months ago
Google - Conversational AI Consultant

Google

Bengaluru, Karnataka, India (On-Site)
2 Months ago
Google - Senior Software Engineer, Distributed Machine Learning

Google

Mountain View, California, United States (On-Site)
2 Months ago
Ubisoft - Senior C++ Programmer - Machine Learning

Ubisoft

Montreal, Quebec, Canada (On-Site)
3 Months ago

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

ByteDance - Student Researcher (Doubao (Seed) - Foundation Model - Vision and Language) - 2025 Start (PhD)

ByteDance

Seattle, Washington, United States (On-Site)
8 Months ago
Demandbase - Senior Data Scientist

Demandbase

San Francisco, California, United States (On-Site)
1 Month ago
Google - Software Engineer III, Infrastructure, Google Cloud

Google

Kraków, Lesser Poland Voivodeship, Poland (On-Site)
2 Months ago
Playdead - Graphics Programmer

Playdead

Copenhagen, Denmark (On-Site)
10 Months ago
ByteDance - Research Scientist Graduate (Foundation Model, Video Generation) - 2025 Start (PhD)

ByteDance

Seattle, Washington, United States (On-Site)
8 Months ago
Fictiv - Software Engineer-Backend

Fictiv

Pune, Maharashtra, India (On-Site)
1 Month ago
Ansys - Senior R&D Engineer (C++, Qt)

Ansys

Athens, Greece (Hybrid)
1 Month ago
Google - Software Engineer III, Infrastructure, Google Cloud Compute Infrastructure

Google

Kirkland, Washington, United States (On-Site)
6 Months ago
Google - Senior Software Engineer, Infrastructure, Google Cloud AI

Google

Kirkland, Washington, United States (On-Site)
2 Months ago
ByteDance - Imaging Systems Lead - Smart Wearable

ByteDance

San Jose, California, United States (On-Site)
3 Months ago

Get notifed when new similar jobs are uploaded

Jobs in Mountain View, California, United States

Marsh McLennan - Sr. Manager - Health & Benefits Consulting

Marsh McLennan

Norwalk, Connecticut, United States (Hybrid)
1 Month ago
Postman - Account Development Representative

Postman

San Francisco, California, United States (Hybrid)
2 Months ago
Warner Bros Games - Lead Software Engineer

Warner Bros Games

Troy, New York, United States (Remote)
3 Months ago
Corsair - Product Engineer

Corsair

Milpitas, California, United States (On-Site)
3 Months ago
ByteDance - Backend Software Engineer

ByteDance

San Jose, California, United States (On-Site)
3 Months ago
Axon - Senior Business Intelligence Analyst

Axon

Denver, Colorado, United States (Hybrid)
1 Month ago
GoDaddy - Freelance Photographer

GoDaddy

Fayetteville, Arkansas, United States (On-Site)
1 Month ago
Philips - Marketing Lead

Philips

Malvern, Pennsylvania, United States (On-Site)
1 Month ago
Light Speed Studios - Senior Sound Designer

Light Speed Studios

California, United States (On-Site)
4 Months ago
Notion - AI Product Marketing Manager

Notion

San Francisco, California, United States (On-Site)
1 Month ago

Get notifed when new similar jobs are uploaded

Similar Category Jobs

Ubisoft - Senior C++ Programmer - Machine Learning

Ubisoft

Montreal, Quebec, Canada (On-Site)
3 Months ago
Pika - Summer Research Internship

Pika

Palo Alto, California, United States (On-Site)
4 Months ago
Meta - Research Scientist Intern, Language and Multimodal Research for MetaAI (PhD)

Meta

Bellevue, Washington, United States (On-Site)
7 Months ago
ByteDance - Senior Software Development Engineer, Large Language Models & Generative AI

ByteDance

San Jose, California, United States (On-Site)
8 Months ago
NVIDIA - Principal Technical Program Manager, AI and Enterprise Apps

NVIDIA

Santa Clara, California, United States (On-Site)
2 Months ago
Scale AI - Machine Learning Research Scientist / Research Engineer, MLDG

Scale AI

San Francisco, California, United States (On-Site)
8 Months ago
NVIDIA - Principal Engineer - DL and AI Software

NVIDIA

Santa Clara, California, United States (On-Site)
5 Months ago
NVIDIA - Senior Solutions Architect, Retail

NVIDIA

Arkansas, United States (Remote)
2 Months ago
Microsoft - Senior Applied Researcher

Microsoft

Redmond, Washington, United States (On-Site)
2 Months ago
Google - Software Engineer III, AI/ML GenAI, Search

Google

Mountain View, California, United States (On-Site)
2 Months ago

Get notifed when new similar jobs are uploaded

About The Company

Atlanta, Georgia, United States (On-Site)

London, England, United Kingdom (On-Site)

Taipei City, Taiwan (On-Site)

Kirkland, Washington, United States (On-Site)

Sunnyvale, California, United States (On-Site)

Sunnyvale, California, United States (On-Site)

Sunnyvale, California, United States (On-Site)

Kraków, Lesser Poland Voivodeship, Poland (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