XLA(加速线性代数)是一个面向 GPU、CPU 和 ML 加速器的开源机器学习编译器。
XLA 编译器接收来自 PyTorch、TensorFlow 和 JAX 等主流机器学习框架的模型,并针对包括 GPU、CPU 和 ML 加速器在内的不同硬件平台进行优化,以实现高性能执行。
项目的官方网站是 openxla.org。
如果你想使用 XLA 来编译你的机器学习项目,请参考对应机器学习框架的文档:
如果你不打算为 XLA 编译器贡献代码,则无需克隆和构建此仓库。这里的所有内容主要面向希望开发编译器的 XLA 贡献者,以及希望为机器学习前端和硬件后端添加支持或进行调试的 XLA 集成者。
如果你想为 XLA 做出贡献,请先阅读如何贡献,然后查看开发者指南。
在 TensorFlow 治理下,SIG OpenXLA 的所有社区空间均需遵守 TensorFlow 行为准则。