如何只安装XLA？

Question

我想使用XLA作为我的项目的后端。是否有建议的方法来安装它自己(没有其余的TensorFlow)。Jax可能会这样做，但在他们的存储库中，不太清楚是如何做到的。

UPDATE I为此向TensorFlow提出了一张票

Answer 1

长生不老药项目已经发布了XLA双星。

XLA接口在tensorflow/compiler/xla中。其他的，IIUC，正在支持功能。

Answer 2

除了tensorflow之外，没有支持安装XLA的方法。

也就是说，JAX确实在雅克斯利卜包中从tensorflow中分别提取、构建和捆绑XLA。您可以在以下各种平台上看到jaxlib的相关构建脚本：https://github.com/google/jax/tree/main/build

特别是，看看wheel.py，它包含从tensorflow源代码提取相关XLA片段的脚本，作为jaxlib构建的一部分。