openxla/xla

This is stars and forks stats for /openxla/xla repository. As of 26 Apr, 2024 this repository has 1670 stars and 197 forks.

XLA XLA (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. Get started If you want to use XLA to compile your ML project, refer to the corresponding documentation for your ML framework: PyTorch TensorFlow JAX If you're...