XLA

XLA (Accelerated Linear Algebra) is an open-source machine learning compiler for GPUs, CPUs, and ML accelerators. It takes models from frameworks such as PyTorch, TensorFlow, and JAX and optimizes them for high-performance execution across different hardware platforms. The project is hosted under OpenXLA and provides documentation and developer guides for contributors and integrators.