Also known as: XLA
XLA, or Accelerated Linear Algebra, is an open-source compiler technology for optimizing machine learning computations across CPUs, GPUs, and specialized accelerators. It translates model operations from frameworks such as TensorFlow, JAX, or PyTorch into forms intended to execute more efficiently on the target hardware.
An open-source machine learning compiler for GPUs, CPUs, and ML accelerators. The XLA compiler takes models from popular ML frameworks such as PyTorch, TensorFlow, and JAX, and optimizes them for high-performance execution across different hardware platforms including GPUs, CPUs, and ML accelerators.