CODEWITHSUNDEEP
tensorflowtensorflow2.0jaxtensorflow-xla
joel
joel

Reputation: 7877

How to install just XLA?

I want to use XLA as a backend for my project. Is there a recommended way to install it on its own (without the rest of TensorFlow). Jax probably does this, but looking in their repository it's not obvious how.

Upvotes: 2

Views: 1149

Answers (2)

joel
joel

Reputation: 7877

XLA has been moved to the OpenXLA GitHub organization. It can be installed on its own from there.

Upvotes: 0

jakevdp
jakevdp

Reputation: 86443

There is no supported way to install XLA on its own apart from tensorflow.

That said, JAX does extract, build, and bundle XLA separately from tensorflow within the jaxlib package. You can see the relevant build scripts for jaxlib on various platforms here: https://github.com/google/jax/tree/main/build

In particular, take a look at build_wheel.py, which contains the scripts that extract relevant pieces of XLA from the tensorflow source as part of the jaxlib build.

Upvotes: 1

Related Questions