From b917a31f5602ee2152294183818de34efb01b589 Mon Sep 17 00:00:00 2001 From: Skye Wanderman-Milne Date: Wed, 19 Apr 2023 17:52:16 +0000 Subject: [PATCH] Update TPU install on main docs page --- docs/index.rst | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/docs/index.rst b/docs/index.rst index f0c498c74..01fb06ad8 100644 --- a/docs/index.rst +++ b/docs/index.rst @@ -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.