Senior ML Compiler Engineer

Google organizes the world's information and makes it universally accessible and useful through AI, Software, and Hardware innovation.
Machine Learning
Senior Software Engineer
Contact Company
5+ years of experience
AI · Enterprise SaaS
This job posting may no longer be active. You may be interested in these related jobs instead:
Senior Software Developer, AI/Machine Learning, Applied AI

Senior Software Developer role at Google Cloud focusing on AI/ML applications for financial crime detection, requiring expertise in software development, machine learning, and financial technology.

AI/ML Engineer, National Security, Public Sector

Senior AI/ML Engineering role at Google focusing on national security applications, requiring extensive ML experience and security clearance, offering competitive compensation and benefits.

App Development Customer Engineer, Global Partnerships Sellside

Senior technical role combining AI/ML expertise with business strategy to drive partner success in app development and advertising solutions.

Senior Software Engineer, Generative AI and LLMs

Senior Software Engineer position at Google focusing on developing AI agents and LLMs for Pixel devices, combining machine learning expertise with mobile development.

Senior Imaging and On-Device Machine Learning Software Engineer, Silicon

Senior role developing imaging and ML software for Google Tensor SoCs and Pixel devices, focusing on Generative AI and Computational Photography implementation.

Description For Senior ML Compiler Engineer

Google is seeking a Senior ML Compiler Engineer to join their team in developing next-generation technologies. This role focuses on developing Machine Learning (ML) compilers for the Tensor TPU to accelerate Generative AI (GenAI) and other complex machine learning models running on custom hardware accelerators. The position combines software engineering expertise with compiler development, requiring both technical depth and project management skills.

As part of Google's mission to organize the world's information and make it universally accessible, you'll work at the intersection of Google AI, Software, and Hardware to create groundbreaking experiences. The role involves building compilers and tools for ML models, with a particular emphasis on always-on computing use cases. You'll be evaluating various parallelization strategies and collaborating with both ML researchers and hardware engineers.

The ideal candidate should have strong experience in compiler development, software engineering, and machine learning. You'll be working on critical projects involving the Pixel Edge TPU, including Gemini Nano, Camera, Imaging, and Speech ML models. This position offers the opportunity to impact billions of users while working with cutting-edge technology in areas such as distributed computing, large-scale system design, artificial intelligence, and hardware acceleration.

Google offers a collaborative environment where engineers can be versatile, display leadership qualities, and tackle new problems across the full stack. The role provides opportunities to switch teams and projects as both you and the fast-paced business grow and evolve. If you're passionate about compiler optimization, machine learning, and hardware acceleration, this position offers the perfect blend of challenging technical work and meaningful impact.

Last updated a month ago

Responsibilities For Senior ML Compiler Engineer

  • Build compilers and tools that map ML models (with a particular focus on always-on computing use cases) to the hardware Instruction Set Architecture (ISA)
  • Evaluate various trade-offs of different parallelization strategies such as performance, power, energy and memory consumption
  • Collaborate with machine learning researchers to improve the domain-specific compiler
  • Collaborate with hardware engineers to evolve future accelerators
  • Productize various ML workloads on the Pixel Edge TPU including Gemini Nano, Camera, Imaging and Speech ML models

Requirements For Senior ML Compiler Engineer

  • Bachelor's degree in Computer Science, a related technical field, or equivalent practical experience
  • 5 years of experience with Compilers
  • 3 years of experience testing, maintaining, or launching software products
  • 1 year of experience with software design and architecture
  • Experience in power and performance optimizations
  • Experience with domain-specific compilers for machine learning
  • Knowledge of hardware that provides a degree of parallelism

Interested in this job?