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 involves working on the hardware-independent parts of the stack, including StableHLO, HLO, XLA components and API, and compiler tooling for faster model debugging. The goal is to make OpenXLA the best place for ML acceleration innovation and for ML accelerator vendors and compiler research.

Key responsibilities include:

• Writing product or system development code
• Participating in design reviews and deciding among available technologies
• Reviewing code developed by other developers
• Contributing to documentation and educational content
• Triaging and resolving product or system issues
• Learning and contributing to StableHLO, XLA, and engaging with stakeholders

The ideal candidate should have:

• Bachelor's degree or equivalent practical experience
• 5 years of experience with software development and data structures/algorithms
• 3 years of experience testing, maintaining, or launching software products
• 1 year of experience with software design and architecture
• Experience in C++ or Python
• Interest in ML developer infrastructure and tools

Google Cloud accelerates digital transformation for organizations worldwide, leveraging cutting-edge technology and developer tools. The company offers a competitive salary range, bonus, equity, and benefits package. Google is committed to diversity, equality, and creating a culture of belonging for all employees.