XLA - AshokBhat/ml GitHub Wiki

XLA (Accelerated Linear Algebra)

• Domain-specific compiler for linear algebra
• TensorFlow graph → Sequence of computation kernels
• Exploits model-specific information for optimization
• Target support - GPU, CPU and TPU

StableHLO

• https://github.com/openxla/stablehlo

Framework support

Framework	Description
PyTorch	https://github.com/pytorch/xla
TensorFlow
JAX	Python package jaxlib is the XLA library for JAX

See Also

• MLIR
• [XLA]] ](/AshokBhat/ml/wiki/[[TVM) | GLOW
• [TensorFlow]] ](/AshokBhat/ml/wiki/[[PyTorch) | JAX