jobs

Staff ML Compiler Engineer, TPU Performance Optimizations

Google Cloud's software engineers develop the next-generation technologies that change how billions of users connect, explore, and interact with information and one another. We're looking for engineers who bring fresh ideas from all areas,

In inglese

Company

Google

Location

Bengaluru, Karnataka, India

Status

Closed

Posted

2026-07-14T14:39:19+00:00

Google Cloud is hiring a staff software engineer for its TPU compiler and ML software/hardware stack team in Bengaluru, working on-site. The job involves designing and implementing compiler optimizations such as pipelining, prefetching and fusions for ML workloads, performance debugging of custom kernels, and integrating frameworks including JAX and PyTorch. It also carries technical leadership and mentoring duties as a team lead. Requirements include eight years of software development experience, three years in design and architecture, and machine learning, compilers, computer architecture, C++ and Python.