Skip to content

Commit

Permalink
Point to the conda-forge jaxlib wheels in the JAX readme.
Browse files Browse the repository at this point in the history
  • Loading branch information
hawkinsp committed Aug 12, 2022
1 parent 88f2b5e commit 6bd9fde
Showing 1 changed file with 7 additions and 1 deletion.
8 changes: 7 additions & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -388,7 +388,7 @@ Some standouts:

JAX is written in pure Python, but it depends on XLA, which needs to be
installed as the `jaxlib` package. Use the following instructions to install a
binary package with `pip`, or to [build JAX from
binary package with `pip` or `conda`, or to [build JAX from
source](https://jax.readthedocs.io/en/latest/developer.html#building-from-source).

We support installing or building `jaxlib` on Linux (Ubuntu 16.04 or later) and
Expand Down Expand Up @@ -512,6 +512,12 @@ jax.tools.colab_tpu.setup_tpu()
Colab TPU runtimes use an older TPU architecture than Cloud TPU VMs, so installing `jax[tpu]` should be avoided on Colab.
If for any reason you would like to update the jax & jaxlib libraries on a Colab TPU runtime, follow the CPU instructions above (i.e. install `jax[cpu]`).

### Conda installation

There is a community-supported Conda build of `jaxlib`. See the
[jaxlib conda-forge](https://github.com/conda-forge/jaxlib-feedstock#installing-jaxlib)
project for installation instructions.

### Building JAX from source
See [Building JAX from
source](https://jax.readthedocs.io/en/latest/developer.html#building-from-source).
Expand Down

0 comments on commit 6bd9fde

Please sign in to comment.