Software Engineer, OpenXLA

Google is seeking a Software Engineer for the OpenXLA project, an open-source ML compiler that powers Tensorflow, JAX, and PyTorch/XLA. The role focuses on making OpenXLA the best place for Machine Learning (ML) acceleration innovation.

You will work on the stack, including StableHLO, HLO, XLA components and API, and compiler tooling for faster model debugging. The goal is to make OpenXLA the preferred platform for ML accelerator and compiler research, ensuring vendors can integrate their hardware-specific optimizations and contributions.

As part of Google Cloud, you'll contribute to enterprise-grade solutions leveraging cutting-edge technology. The role involves writing code, participating in design reviews, reviewing code from other developers, contributing to documentation, and debugging system issues.

Key responsibilities include:

• Developing product or system code
• Leading or participating in design reviews
• Reviewing code and providing feedback
• Contributing to documentation and educational content
• Triaging and resolving product or system issues
• Learning and contributing to StableHLO and XLA
• Engaging with stakeholders who integrate with OpenXLA

The ideal candidate should have experience with software development, data structures/algorithms, and testing/maintaining software products. Experience with C++ or Python is required, and knowledge of ML developer infrastructure is preferred.

Google offers a competitive salary range of $161,000-$239,000 plus bonus, equity, and benefits. They are committed to diversity, equality, and creating a culture of belonging for all employees.