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.
XLA has been moved to the OpenXLA GitHub organization. It can be installed on its own from there.