Taro Logo

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 systems.

The position involves working at the intersection of machine learning and hardware optimization, building tools that efficiently map ML models to hardware architectures. You'll be particularly focused on always-on computing use cases and optimizing performance, power consumption, and memory usage. The role requires collaboration with both ML researchers and hardware engineers to evolve compiler technology and future accelerators.

As part of Google's broader mission to organize the world's information, you'll be working on critical projects that impact billions of users. The team combines Google's expertise in AI, Software, and Hardware to create innovative solutions. You'll be specifically 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 on cutting-edge ML hardware acceleration technology. The role offers the chance to impact Google's ML infrastructure while working with state-of-the-art technology and collaborating with leading experts in both machine learning and hardware design.

Last updated 2 days 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 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, 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
  • Experience in power and performance optimizations (preferred)
  • Experience with domain-specific compilers for machine learning (preferred)
  • Knowledge of hardware that provides a degree of parallelism (preferred)

Benefits For Senior ML Compiler Engineer, Silicon

Medical Insurance
401k
Parental Leave
  • Comprehensive health coverage
  • Retirement benefits
  • Parental leave support

Interested in this job?

Jobs Related To Google Senior ML Compiler Engineer, Silicon