Senior Software Engineer, Tensor Processing Units Compiler

Google develops next-generation technologies that change how billions of users connect, explore, and interact with information.
$161,000 - $239,000
Backend
Senior Software Engineer
In-Person
5,000+ Employees
5+ years of experience
AI · Enterprise SaaS

Description For Senior Software Engineer, Tensor Processing Units Compiler

Google is seeking a Senior Software Engineer to join their Tensor Processing Units (TPU) Compiler team. This role is crucial in developing and optimizing the compiler for Google's custom-designed TPU processors, which accelerate machine learning and scientific computing workloads. The position offers unique opportunities to work across the compiler stack, from LLVM to MLIR middle-end implementations.

The ideal candidate will have strong experience in compiler development, CUDA C++ applications, and machine learning technologies. You'll be working on cutting-edge technology that powers both internal Google services and external Cloud customers. The role involves close collaboration with TPU users and hardware designers to optimize performance and contribute to future processor designs.

The position offers competitive compensation ($161,000-$239,000) plus bonus, equity, and comprehensive benefits. You'll be part of Google Cloud's mission to accelerate digital transformation across industries, working with enterprise-grade solutions and cutting-edge technology. The team's work directly impacts how machine learning workloads are processed at scale, making this an exciting opportunity for engineers passionate about compiler optimization and machine learning infrastructure.

This role combines technical depth in compiler development with the broad impact of Google's machine learning infrastructure. You'll be at the forefront of hardware/software co-design, working on technology that powers next-generation AI applications. The position offers growth opportunities and the chance to work with leading experts in compiler development, machine learning, and hardware design.

Last updated 15 hours ago

Responsibilities For Senior Software Engineer, Tensor Processing Units Compiler

  • Contribute to a compiler for a novel processor designed to accelerate machine learning workloads
  • Compile high-performance implementations of operations at a distributed scale
  • Work closely with users of TPUs to improve performance/efficiency and hardware designers to co-design future processors
  • Investigate high-level representations to effectively program large-scale, distributed, and heterogeneous systems

Requirements For Senior Software Engineer, Tensor Processing Units Compiler

Python
  • Bachelor's degree or equivalent practical experience
  • 5 years of experience testing, maintaining, or launching software products
  • 1 year of experience with software design and architecture
  • 2 years of experience with CUDA C++ application development
  • 1 year of experience with Native Code, Just-In-Time (JIT), Cross, Source-to-Source or any other type of compilers
  • 2 years of experience with data structures or algorithms
  • Experience with machine learning algorithms and tools (e.g., TensorFlow)

Benefits For Senior Software Engineer, Tensor Processing Units Compiler

Medical Insurance
Dental Insurance
Vision Insurance
401k
Parental Leave
  • bonus
  • equity
  • benefits

Interested in this job?

Jobs Related To Google Senior Software Engineer, Tensor Processing Units Compiler

Senior Customer and Partner Solutions Engineer, gTech

Senior Customer and Partner Solutions Engineer position at Google, focusing on technical problem-solving and partner integration support with competitive compensation and benefits.

Senior Software Engineer, Full Stack

Senior Software Engineer position at Google Cloud, focusing on building innovative LLM-based solutions for security and compliance, requiring strong full-stack and ML expertise.

Senior Software Engineer, Full Stack, Google Cloud

Senior Software Engineer position at Google Cloud, focusing on full-stack development with 5+ years of experience required, based in Bengaluru.

Senior Software Engineer, Google Research

Senior Software Engineer position at Google Research, focusing on developing next-generation technologies and conducting fundamental research that impacts billions of users.

Senior Software Engineer, Labs

Senior Software Engineer position at Google Labs focusing on developing innovative technologies and early-stage products that advance Google's core mission.