AI product Open source
StableHLO is a backward-compatible operation set for high-level machine-learning operations and a portability layer between ML frameworks and compilers. Frameworks such as TensorFlow, JAX, and PyTorch can produce StableHLO programs, while compilers such as XLA and IREE can consume them; the programs can also be used for on-device execution through Google AI Edge.
Based on the MHLO dialect, StableHLO adds serialization and versioning through MLIR bytecode, with backward- and forward-compatibility guarantees. The project includes the StableHLO specification and an MLIR-based C++ and Python implementation for defining programs and integrating them with compatible compilers.
1 in the library.