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 making OpenXLA the go-to platform for ML accelerator vendors and compiler research, ensuring seamless integration of hardware-specific optimizations and code generation.

As part of Google's Core team, you'll contribute to the technical foundation behind Google's flagship products, working on central solutions and breaking down technical barriers. The position offers competitive compensation ($136,000-$200,000) plus bonus, equity, and comprehensive benefits.

The ideal candidate should have at least 2 years of experience in software development, data structures, algorithms, and ML/AI technologies. Experience with compiler development, open source, and system health diagnosis is highly valued. This role offers the opportunity to impact critical projects at Google while working with cutting-edge ML technologies and contributing to open-source development.

Join Google's dynamic environment where you'll collaborate with talented engineers, work on challenging technical problems, and help shape the future of machine learning acceleration. The role combines technical expertise in ML, compiler development, and system design with the opportunity to contribute to Google's mission of organizing the world's information.