XLA

From HPCWIKI
Revision as of 18:38, 25 March 2023 by Admin (talk | contribs)
Jump to navigation Jump to search

섬네일|openxla XLA[1] (Accelerated Linear Algebra) is an open-source machine learning (ML) 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.

Reference