Software Engineer, Tensor Processing Units Compiler

Google is seeking a Software Engineer for their Tensor Processing Units (TPUs) Compiler team. This role involves working on the MLIR/LLVM based TPU compiler for Google's custom-designed processors that accelerate machine learning and scientific computing workloads. The ideal candidate will contribute to compiling high-performance implementations of operations at a distributed scale, work closely with TPU users to improve performance/efficiency, and collaborate with hardware designers for future processor co-design.

Key responsibilities include:

1. Contributing to a compiler for a novel processor designed to accelerate machine learning workloads.
2. Compiling high-performance implementations of operations at a distributed scale.
3. Working closely with TPU users to improve performance/efficiency.
4. Collaborating with hardware designers to co-design future processors.
5. Investigating high-level representations for effective programming of large-scale, distributed, and heterogeneous systems.

The role offers opportunities to work on various aspects of the compiler stack, including Low Level Virtual Machine (LLVM) and Multi-Level Intermediate Representation (MLIR) middle-end. The successful candidate will support new workloads, optimize for new models and characteristics, and support new TPU hardware across multiple generations.

This position is part of Google Cloud, which provides enterprise-grade solutions leveraging Google's cutting-edge technology to help organizations digitally transform their businesses. The role presents an exciting opportunity to work on advanced technology that impacts billions of users worldwide.

Google is committed to diversity, equality, and creating a culture of belonging. They offer equal employment opportunities regardless of race, creed, color, religion, gender, sexual orientation, gender identity/expression, national origin, disability, age, genetic information, veteran status, marital status, pregnancy or related condition, or any other protected basis.