mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-15 19:36:06 +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

4
docs/index.rst
View File

@ -13,7 +13,7 @@ JAX is Autograd_ and XLA_, brought together for high-performance numerical compu
:class-card: sd-border-0
:shadow: None
JAX provides a familiar NumPy-style API for ease of adoption by researchers and engineers.
JAX provides a familiar NumPy-style API for ease of adoption by researchers and engineers.
.. grid-item-card:: Transformations
:columns: 12 6 6 4
@ -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.