Skip to content

Commit

Permalink
Add --pre to the libtpu nightly install instructions.
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 609702726
  • Loading branch information
yashk2810 authored and jax authors committed Feb 23, 2024
1 parent 25416a4 commit 4bb0a5a
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion docs/installation.md
Original file line number Diff line number Diff line change
Expand Up @@ -180,7 +180,7 @@ pip install -U --pre jaxlib -f https://storage.googleapis.com/jax-releases/jaxli
* Jaxlib TPU:
```bash
pip install -U --pre jaxlib -f https://storage.googleapis.com/jax-releases/jaxlib_nightly_releases.html
pip install -U libtpu-nightly -f https://storage.googleapis.com/jax-releases/libtpu_releases.html
pip install -U --pre libtpu-nightly -f https://storage.googleapis.com/jax-releases/libtpu_releases.html
```

* Jaxlib GPU (Cuda 12):
Expand Down

0 comments on commit 4bb0a5a

Please sign in to comment.