Software Engineer, ML Compilers, Silicon

Google organizes the world's information and makes it universally accessible and useful through innovative technology solutions.
Machine Learning
Mid-Level Software Engineer
In-Person
5,000+ Employees
2+ years of experience
AI

Description For Software Engineer, ML Compilers, Silicon

Google is seeking a Software Engineer to join their ML Compilers team working on Google Tensor, the custom System-on-Chip (SoC) that powers Pixel phones. This role focuses on developing ML compilers for the Tensor TPU to accelerate Generative AI and other complex machine learning models running on custom hardware accelerators. The team's work enables Gemini Nano to run efficiently on Pixel phones.

The position combines software engineering with machine learning and hardware optimization, requiring expertise in C++ programming, compiler development, and understanding of hardware architectures. You'll work at the intersection of ML innovation and hardware acceleration, collaborating with both ML researchers and hardware engineers to optimize performance and efficiency.

As part of Google's broader mission to organize world's information, this role contributes to creating radically helpful experiences by combining Google's AI, Software, and Hardware capabilities. The team researches, designs, and develops new technologies to make computing faster and more powerful, ultimately aiming to improve people's lives through technology.

The ideal candidate will have strong programming skills, particularly in C++, understanding of compiler technology, and interest in machine learning systems. Experience with performance optimization and hardware architecture would be valuable. This is an opportunity to work on cutting-edge technology that directly impacts millions of users through Google's Pixel devices.

Last updated 13 hours ago

Responsibilities For Software Engineer, ML Compilers, Silicon

  • Build compilers and tools that efficiently map ML models (with a particular focus on always-on computing use cases) to the hardware ISA
  • Evaluate various trade-offs of different parallelization strategies such as performance, power, energy and memory consumption
  • Collaborate with machine learning researchers to constantly improve the domain-specific compiler
  • Collaborate with hardware engineers to evolve future accelerators

Requirements For Software Engineer, ML Compilers, Silicon

  • Bachelor's degree in Electrical/Electronics Engineering, Computer Engineering, Computer Science, or equivalent practical experience
  • 2 years of development experience in C++
  • Experience in data structures and algorithms

Interested in this job?

Jobs Related To Google Software Engineer, ML Compilers, Silicon

Field Solutions Developer II, Generative AI, Google Cloud

Field Solutions Developer II position at Google Cloud focusing on Generative AI implementation, requiring 5 years of Python experience and AI expertise, offering $147K-$216K base salary.

Software Engineer III, AI/ML

Software Engineer III position at Google focusing on AI/ML implementation and infrastructure, requiring 2 years of software development experience and machine learning expertise.

Software Engineer III, AI/ML, Google Cloud AI

Software Engineer III position at Google Cloud AI focusing on AI/ML development, infrastructure, and implementation of machine learning solutions.

Software Engineer III, Machine Learning, Google Ads

Software Engineer III position focused on machine learning and AI development for Google Ads, requiring 2 years of experience in software development and ML algorithms.

Field Solutions Developer II, Generative AI, Google Cloud

Field Solutions Developer II position at Google Cloud focusing on Generative AI implementation, requiring 5+ years of experience in AI/ML and Python programming.