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.