Skip to content

GitLab

Unverified Commit d24a5231 authored by Patrick von Platen's avatar Patrick von Platen Committed by GitHub
Browse files

Update README.md

parent e3fce2f8
Hide whitespace changes
Inline Side-by-side
Showing with 7 additions and 1 deletion
examples/research_projects/jax-projects/README.md
View file @ d24a5231
...@@ -258,7 +258,13 @@ Doing so will allow you to share your fork of the Transformers library with your ...@@ -258,7 +258,13 @@ Doing so will allow you to share your fork of the Transformers library with your
install JAX's TPU version before cloning and installing the transformers repository. install JAX's TPU version before cloning and installing the transformers repository.
Otherwise, an incorrect version of JAX will be installed, and the following commands will Otherwise, an incorrect version of JAX will be installed, and the following commands will
throw an error. throw an error.
To install JAX's TPU version simply run the following command: To install JAX's TPU version first run the following command:
```
$ pip install requests
```
and then:
``` ```
$ pip install "jax[tpu]>=0.2.16" -f https://storage.googleapis.com/jax-releases/libtpu_releases.html $ pip install "jax[tpu]>=0.2.16" -f https://storage.googleapis.com/jax-releases/libtpu_releases.html
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment