Senior ML Compiler Engineer, Silicon

Google organizes the world's information and makes it universally accessible and useful, combining AI, Software, and Hardware to create helpful experiences.
Machine Learning
Senior Software Engineer
In-Person
5,000+ Employees
5+ years of experience
AI

Description For Senior ML Compiler Engineer, Silicon

Google is seeking a Senior ML Compiler Engineer to join their Silicon team, focusing on developing Machine Learning (ML) compilers for the Tensor TPU to accelerate Generative AI and other complex machine learning models running on custom hardware accelerators. This role combines software engineering expertise with specialized knowledge in compiler development and machine learning optimization.

The position requires deep technical expertise in compiler development, with a focus on mapping ML models to hardware architectures. You'll be working on critical projects that optimize performance, power efficiency, and memory consumption for ML workloads. The role involves close collaboration with both ML researchers and hardware engineers, contributing to the evolution of future accelerators and the improvement of domain-specific compilers.

As part of Google's broader mission to organize the world's information, you'll be working with cutting-edge technology that impacts billions of users. The team combines Google's expertise in AI, Software, and Hardware to create innovative solutions. You'll be particularly involved in productizing ML workloads on the Pixel Edge TPU, including work with Gemini Nano, Camera, Imaging, and Speech ML models.

This is an excellent opportunity for someone with strong compiler experience who wants to work at the intersection of machine learning and hardware optimization. The role offers the chance to impact Google's next-generation AI technologies while working with some of the most advanced ML hardware accelerators in the industry.

Last updated 9 hours ago

Responsibilities For Senior ML Compiler Engineer, Silicon

  • Build compilers and tools that map ML models to the hardware Instruction Set Architecture (ISA)
  • Evaluate various trade-offs of different parallelization strategies
  • 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, Silicon

Python
Java
  • 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

Interested in this job?

Jobs Related To Google Senior ML Compiler Engineer, Silicon

Senior Software Engineer, Generative AI and LLMs

Senior Software Engineering role at Google focusing on Generative AI and LLMs for Pixel devices, combining ML expertise with mobile development.

Senior Software Engineer, Recommendations, Ranking and Prediction, Google Workspace

Senior Software Engineering role at Google focusing on developing recommendation and ranking systems for Google Workspace, combining traditional engineering with machine learning expertise.

Senior Software Engineer, Machine Learning, Google Ads

Senior Software Engineer position at Google focusing on machine learning applications in advertising technology, offering competitive compensation and the opportunity to work on products that impact billions of users.

AI/ML Engineer, National Security, Public Sector

Senior AI/ML Engineering role at Google focusing on national security applications, requiring TS/SCI clearance and extensive machine learning experience.

Senior Software Engineer, Machine Learning, Core

Senior Software Engineer position at Google focusing on Machine Learning and AI, developing next-generation technologies that impact billions of users.