Staff Software Engineer, Machine Learning Compilers, Silicon

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

Description For Staff Software Engineer, Machine Learning Compilers, Silicon

Google is seeking a Staff Software Engineer to join their Machine Learning Compilers team, focusing on developing next-generation technologies that will impact billions of users. This role combines Google's expertise in AI, Software, and Hardware to create groundbreaking experiences. The position requires deep expertise in compiler development, machine learning frameworks, and hardware-software co-design.

The successful candidate will work on critical projects involving ML model compilation and optimization for EdgeTPU, collaborating with various teams including ML model developers, researchers, and hardware/software specialists. This role offers the opportunity to work at the intersection of machine learning and hardware acceleration, directly impacting Google's AI infrastructure.

The position demands strong technical skills with requirements including 8 years of software development experience, particularly in C++ and algorithms, along with significant experience in compilers and ML model optimization. The role offers competitive compensation ranging from $189,000 to $284,000, plus additional benefits including bonus and equity.

Working at Google's Mountain View location, you'll be part of a team that's pushing the boundaries of ML compiler technology, specifically focusing on the EdgeTPU platform. This role is perfect for someone who wants to contribute to cutting-edge ML infrastructure while working with some of the best minds in the industry.

The position offers the opportunity to work on complex technical challenges, from extending ML authoring frameworks to optimizing compiler performance. You'll be involved in designing hardware-software interfaces and co-optimizations between different processing units, making this an ideal role for someone passionate about both ML and systems engineering.

Last updated 6 minutes ago

Responsibilities For Staff Software Engineer, Machine Learning Compilers, Silicon

  • Work with and extend Machine Learning (ML) authoring frameworks to compile ML models for the EdgeTPU
  • Work with ML runtime systems to deploy optimized ML models on the EdgeTPU
  • Work with EdgeTPU architects to design the HW/SW interface, and co-optimizations between CPU, GPU, and TPU
  • Work as part of the EdgeTPU compiler team, including analyzing and improving compiler quality and performance
  • Collaborate with ML model developers, researchers, and EdgeTPU hardware/software teams

Requirements For Staff Software Engineer, Machine Learning Compilers, Silicon

Python
  • Bachelor's degree or equivalent practical experience
  • 8 years of experience in software development (e.g. C++), and with data structures/algorithms
  • 5 years of experience testing, and launching software products
  • 3 years of experience with software design and architecture
  • 5 years of experience with compilers
  • Experience with hardware-software co-design
  • Experience in optimizing Machine Learning (ML) models for inference
  • Experience in Multi-Level Intermediate Representation (MLIR) and low level virtual machines (LLVM)
  • Experience in compiler development within accelerator-based architectures

Benefits For Staff Software Engineer, Machine Learning Compilers, Silicon

Medical Insurance
Dental Insurance
Vision Insurance
  • bonus
  • equity
  • benefits

Interested in this job?

Jobs Related To Google Staff Software Engineer, Machine Learning Compilers, Silicon

Staff Software Engineer, Cloud ML Compute Services

Staff Software Engineer position at Google Cloud working on Cloud ML Compute Services, focusing on TPU and GPU services and ML frameworks.

Software Engineer Manager, Unified Cloud Search

Lead software engineering team for Google Cloud's Unified Search, focusing on ML models and search technologies.

Staff Software Engineer, ML Performance, GPUs

Staff Software Engineer position focused on ML Performance and GPUs at Google, working on optimizing Large Language Models and GPU infrastructure.

Staff Software Engineer, AI/ML Computer Vision, Google Ads

Staff Software Engineer position at Google focusing on AI/ML Computer Vision for Google Ads, offering competitive compensation and benefits.

Staff Software Engineer, AI/ML GenAI, Gemini

Lead AI/ML engineer position at Google, focusing on GenAI and Gemini development, requiring extensive experience in machine learning and software engineering.