Software Engineer III, OpenXLA

Google is seeking a Software Engineer III for their OpenXLA team. OpenXLA is an open-source ML compiler that powers Tensorflow, JAX and PyTorch/XLA, accelerating ML models on CPUs and GPUs. The role involves working on the hardware-independent parts of the stack, including StableHLO, HLO, XLA components and APIs, and compiler tooling for faster model debugging. The ideal candidate will contribute to making OpenXLA the best platform for ML acceleration innovation and ensure it's accessible for hardware vendors and compiler researchers. This position is part of Google's Core team, which builds the technical foundation behind Google's flagship products. The role offers opportunities to impact important technical decisions across the company and contribute to central solutions that break down technical barriers and strengthen existing systems. The position requires a strong background in software development, data structures, algorithms, and machine learning, with a focus on compiler development and open source experience.