Software Engineer III, OpenXLA

Google is seeking a Software Engineer III for their OpenXLA team, focusing on making OpenXLA the premier platform for Machine Learning acceleration innovation. OpenXLA is an open-source ML compiler powering TensorFlow, JAX, and PyTorch/XLA, accelerating ML models on CPUs and GPUs.

The role involves working on the hardware-independent components of the stack, including StableHLO, HLO, XLA components, and API, as well as compiler tooling for enhanced model debugging. You'll be responsible for ensuring OpenXLA remains the top choice for ML accelerator vendors and compiler research, facilitating hardware-specific optimizations and code generation.

As part of the Core team, you'll contribute to the technical foundation behind Google's flagship products, working on underlying design elements, developer platforms, and infrastructure. This position offers the unique opportunity to impact technical decisions across the company and build central solutions that break down technical barriers.

The ideal candidate should have strong experience in software development, data structures, algorithms, and machine learning. Knowledge of compiler development, open source, and system health diagnosis is highly valued. You'll collaborate with peers and stakeholders, participate in design reviews, and ensure best practices in code development.

This role offers competitive compensation, including a base salary range of $136,000-$200,000, plus bonus, equity, and comprehensive benefits. You'll be part of Google's commitment to innovation and their mission to organize the world's information and make it universally accessible and useful.