xla

XLA(Accelerated Linear Algebra)是面向 GPU、CPU 和 ML 加速器的开源机器学习编译器。昇腾 NPU 支持由华为 xla-npu 后端提供,相关文档如下: