From 415810664b7b090d7f8d07bf84aa677cde9e2cf3 Mon Sep 17 00:00:00 2001 From: Stas Bekman Date: Thu, 23 Dec 2021 13:12:59 -0800 Subject: [PATCH] [doc] install - add jax (#14912) As `jax` cuda requires special instructions to be installed correctly add a link to jax installation instructions. Note: Flax install page only covers cpu jax installation info. --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index 13a2b31a5c..de609238bc 100644 --- a/README.md +++ b/README.md @@ -198,7 +198,7 @@ You should install 🤗 Transformers in a [virtual environment](https://docs.pyt First, create a virtual environment with the version of Python you're going to use and activate it. Then, you will need to install at least one of Flax, PyTorch or TensorFlow. -Please refer to [TensorFlow installation page](https://www.tensorflow.org/install/), [PyTorch installation page](https://pytorch.org/get-started/locally/#start-locally) and/or [Flax installation page](https://github.com/google/flax#quick-install) regarding the specific install command for your platform. +Please refer to [TensorFlow installation page](https://www.tensorflow.org/install/), [PyTorch installation page](https://pytorch.org/get-started/locally/#start-locally) and/or [Flax](https://github.com/google/flax#quick-install) and [Jax](https://github.com/google/jax#installation) installation pages regarding the specific install command for your platform. When one of those backends has been installed, 🤗 Transformers can be installed using pip as follows: