我想使用 XLA 作为我项目的后端。有没有推荐的方法来独立安装它（没有 TensorFlow 的其余部分）。Jax 可能会这样做，但在他们的存储库中查看它并不明显。
更新我为此向TensorFlow提出了一张票
问问题
95 次
2 回答
1
除了 tensorflow 之外，没有支持自行安装 XLA 的方法。
也就是说，JAX 确实在jaxlib包中与 tensorflow 分开提取、构建和捆绑 XLA。您可以在此处查看各种平台上 jaxlib 的相关构建脚本：https ://github.com/google/jax/tree/main/build
特别是，看看build_wheel.py，其中包含从 tensorflow 源中提取 XLA 的相关片段作为 jaxlib 构建的一部分的脚本。
于 2021-07-07T23:10:45.333 回答