From 6bd9fde7b5bd7c0c9004d67b0c3e7c197f0af0b7 Mon Sep 17 00:00:00 2001 From: Peter Hawkins Date: Fri, 12 Aug 2022 19:52:34 +0000 Subject: [PATCH] Point to the conda-forge jaxlib wheels in the JAX readme. --- README.md | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/README.md b/README.md index 4e0ca7f6efac..ea059a52ed1d 100644 --- a/README.md +++ b/README.md @@ -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 @@ -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).