mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-17 20:36:05 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update TPU install on main docs page

Browse Source
This commit is contained in:
Skye Wanderman-Milne 2023-04-19 17:52:16 +00:00
parent a2fbd59e63
commit b917a31f56
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
docs/index.rst
View File

@ -70,7 +70,7 @@ Installation
.. code-block:: bash
pip install "jax[tpu]>=0.2.16" -f https://storage.googleapis.com/jax-releases/libtpu_releases.html
pip install "jax[tpu]" -f https://storage.googleapis.com/jax-releases/libtpu_releases.html
For more information about supported accelerators and platforms, and for other
installation options, see the `Install Guide`_ in the project README.