Director, Engineering, ML Compiler

8 Hours ago • 15 Years + • Artificial Intelligence • $294,000 PA - $414,000 PA

Job Summary

Job Description

Google seeks an experienced leader to build and direct its ML Compiler team. This role shapes Google's hardware strategy by orchestrating development of cutting-edge machine learning compilers. The Director will work at the intersection of research, product development, hardware engineering, and core machine learning teams. Responsibilities include defining the long-term strategic vision, establishing technical direction, recruiting and mentoring a team, forging partnerships with other teams, optimizing system performance, and advocating for innovation. The role requires deep expertise in compilers and machine learning to support Google's TPU and GPU strategies and ensure optimal compiler solutions.
Must have:
  • Master's degree in CS/related field or equivalent experience
  • 15 years compiler development experience (ML optimization)
  • 10 years leading technical teams
  • Experience with ML accelerators (TPUs or GPUs)
  • Define long-term strategic vision for ML compiler technology
Good to have:
  • Experience in computer architecture, high-performance computing
  • Building compilers for custom hardware architectures
  • In-depth knowledge of ML frameworks (Jax, PyTorch, TensorFlow)
  • Outstanding communication skills
  • Contributions to open-source compiler projects
Perks:
  • Bonus
  • Equity
  • Benefits

Job Details

Minimum qualifications:

  • Master's degree in Computer Science, Computer Engineering, a related field, or equivalent practical experience.
  • 15 years of industry experience in compiler development, including machine learning optimization.
  • 10 years of experience leading and growing technical teams.
  • Experience with ML accelerators like TPUs or GPUs.

Preferred qualifications:

  • Experience in computer architecture, high-performance computing.
  • Experience building compilers for custom hardware architectures.
  • In-depth knowledge of machine learning frameworks (e.g., Jax, PyTorch, TensorFlow).
  • Outstanding communication skills tailored to both technical and non-technical audiences.
  • Contributions to open-source compiler projects.

About the job

Google is a leading ML/AI company. Google’s Core ML team aims to build state-of-the-art AI platforms, services, and tools behind Google’s AI research and AI-powered products. We are owners of the Google AI framework (e.g., TensorFlow, JAX), AI performance and efficiency, AI training and inference platforms, AI compilers (e.g. XLA), AI software/hardware co-design (e.g. TPU), and AI developer experience (e.g., Model Hub, Eval Hub). We are the driver to launch Large Models (like Gemini) from research to production and enable all Google products to adopt LMs to improve end-user experiences. The CoreML team is responsible for planning and optimizing ML capacity for all Google products and driving performance and efficiency work to maximize ROI for our AI investments. We look across Google’s AI efforts to build central solutions, break down technical barriers, and strengthen existing systems. As the Core team, we have a mandate and a unique opportunity to impact important technical decisions across the company.

We are seeking an exceptional and experienced leader to build and direct our ML Compiler team. This pivotal role presents the unique opportunity to shape Google's hardware strategy by orchestrating the development of cutting-edge machine learning compilers. You'll work at the intersection of research, product development, hardware engineering, and core machine learning teams to seamlessly execute the ML Compiler roadmap.

Google Cloud accelerates every organization’s ability to digitally transform its business and industry. We deliver enterprise-grade solutions that leverage Google’s cutting-edge technology, and tools that help developers build more sustainably. Customers in more than 200 countries and territories turn to Google Cloud as their trusted partner to enable growth and solve their most critical business problems.

The US base salary range for this full-time position is $294,000-$414,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

  • Define the long-term strategic vision for our ML compiler technology and roadmap. Establish technical direction, goals, and development priorities.
  • Recruit, mentor, and develop a highly skilled team of compiler engineers focused on machine learning optimization. Create a collaborative and high-achieving work environment.
  • Forge strong partnerships with research, product, hardware, and core ML teams. Drive alignment and effective collaboration to ensure optimal compiler solutions for Google's custom TPU and GPU strategies.
  • Apply your deep expertise in compilers and machine learning to optimize system performance, reduce software overheads, and seamlessly support new hardware. Advocate innovation and the application of new technologies.
  • Cultivate a deep understanding of customer needs, driving a relentless focus on enabling their success through compiler capabilities.

Similar Jobs

Meta - Research Scientist, Computer Vision for Generative AI (PhD)

Meta

Seattle, Washington, United States (On-Site)
5 Months ago
ION - Senior AI Engineer, Italy

ION

Pisa, Tuscany, Italy (On-Site)
6 Months ago
Resemble AI - Deep Learning Speech Researcher

Resemble AI

Mountain View, California, United States (On-Site)
8 Months ago
Google - Research Engineer, Vision Language Models

Google

Zürich, Zurich, Switzerland (On-Site)
8 Hours ago
King - AI/ML Engineer

King

Stockholm, Stockholm County, Sweden (On-Site)
12 Hours ago
Soul AI - Subject Matter Expert (AI Trainer)

Soul AI

Hyderabad, Telangana, India (On-Site)
7 Months ago
Canva - Machine Learning Research Engineering Manager - Image Generation

Canva

Vienna, Vienna, Austria (Remote)
2 Weeks ago
Numrah - Software Engineer (AI)

Numrah

(On-Site)
2 Weeks ago
Microsoft - Principal Engineering Manager

Microsoft

San Jose, California, United States (On-Site)
13 Hours ago
Google - Senior Software Engineer, Machine Learning, Google Ads

Google

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

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

ByteDance - Research Scientist in Foundation Model, Music Core Machine Learning Graduates - 2024 Start (PhD)

ByteDance

San Jose, California, United States (On-Site)
5 Months ago
NVIDIA - Senior Software QA Test Development Engineer

NVIDIA

Santa Clara, California, United States (On-Site)
1 Week ago
Tencent - Senior Staff Researcher

Tencent

California, United States (On-Site)
1 Month ago
Roofstacks - AI/ML Engineer

Roofstacks

İstanbul, İstanbul, Türkiye (Hybrid)
3 Weeks ago
PlayStation Global - Senior Machine Learning Graphics Engineer

PlayStation Global

London, England, United Kingdom (Hybrid)
1 Week ago
ByteDance - Research Engineer Intern (Doubao (Seed) - Machine Learning System) - 2025 Summer (MS)

ByteDance

San Jose, California, United States (On-Site)
5 Months ago
Google - Software Engineer III, AI/ML, YouTube Ads

Google

Mountain View, California, United States (On-Site)
8 Hours ago
Axinous - Sr. Staff ML Engineer

Axinous

San Jose, California, United States (Hybrid)
3 Months ago
N-iX - Senior Data Scientist/AI Engineer

N-iX

Poland (Hybrid)
19 Hours ago
ByteDance - Software Engineer, Machine Learning Training

ByteDance

Singapore (On-Site)
2 Days ago

Get notifed when new similar jobs are uploaded

Jobs in Sunnyvale, California, United States

ByteDance - Senior Backend Software Engineer - Global E-Commerce Supply Chain Merchant Platform

ByteDance

San Jose, California, United States (On-Site)
5 Months ago
Fluence - Sales Engineer/Senior Sales Engineer - Battery Energy Storage

Fluence

Houston, Texas, United States (Hybrid)
6 Months ago
Patreon - iOS Engineer

Patreon

San Francisco, California, United States (Hybrid)
3 Weeks ago
Google - Staff Software Engineer, Serverless Container Management

Google

San Francisco, California, United States (On-Site)
9 Hours ago
USE Insider - Social Media Manager - US

USE Insider

United States (Remote)
6 Months ago
DraftKings - Senior Analyst, Digital Marketing

DraftKings

Boston, Massachusetts, United States (On-Site)
2 Weeks ago
AGS - American Gaming Systems - Field Service Technician I

AGS - American Gaming Systems

Michigan, United States (On-Site)
1 Month ago
IGT - Product Trainer II

IGT

Rhode Island, United States (On-Site)
4 Months ago
NVIDIA - Senior Product Architect

NVIDIA

Santa Clara, California, United States (On-Site)
6 Days ago
Trek - Part Time Sales Associate

Trek

Leesburg, Virginia, United States (On-Site)
1 Month ago

Get notifed when new similar jobs are uploaded

Artificial Intelligence Jobs

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

ByteDance

San Jose, California, United States (On-Site)
5 Months ago
Google - Software Engineer III, Core Machine Learning, Google Cloud

Google

Sunnyvale, California, United States (On-Site)
5 Months ago
NVIDIA - Director of AI Research

NVIDIA

Tel Aviv-Yafo, Tel Aviv District, Israel (On-Site)
1 Month ago
NVIDIA - Senior System Networking Engineer, InfiniBand

NVIDIA

Yokne'am Illit, North District, Israel (On-Site)
2 Months ago
Google - Principal, Data Science, Cloud AI Innovation

Google

Sunnyvale, California, United States (On-Site)
8 Hours ago
Microsoft - Principal Engineering Manager

Microsoft

San Jose, California, United States (On-Site)
13 Hours ago
Google - Customer Engineer II, AI/ML, Google Cloud

Google

Sunnyvale, California, United States (On-Site)
9 Hours ago
Google - Field Solutions Developer II, Generative AI, Google Cloud

Google

San Francisco, California, United States (On-Site)
10 Hours ago
Social Discovery Group - Senior NLP Engineer

Social Discovery Group

Serbia (Remote)
5 Months ago
Google - Head of Generative AI Specialist, Blackbelt

Google

Hyderabad, Telangana, India (On-Site)
11 Hours 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