Update installation instruction (brainpy#651)

* update

* update installation docs

Author: chaoming0625

brainpylib-changelog.md

@@ -2,18 +2,20 @@
 
 ## Version 0.3.0
 
+- Support `brainpy>=2.5.0`
 - Fix bugs on windows platform
 - remove all customized C++ and CUDA operators
 
-
 ## Version 0.2.8
 
+- Support `brainpy>=2.5.0`
 - Fix bugs that the DLL cannot be loaded correctly when windows does not have a c++ environment,
 
 ## ~~Version 0.2.7(YANKED)~~
 
 ## Version 0.2.6
 
+- Support `brainpy>=2.5.0`
 - Fix bugs of taichi call function for single result
 
 ## Version 0.2.5

docs/index.rst

@@ -19,16 +19,24 @@ Installation
 
           pip install -U brainpy[cpu]  # windows, linux, macos
 
-    .. tab-item:: GPU (CUDA)
+    .. tab-item:: GPU (CUDA 11.0)
 
        .. code-block:: bash
 
-          # for CUDA 11.0, Linux only
           pip install -U brainpy[cuda11] -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
 
-          # for CUDA 12.0, Linux only
+    .. tab-item:: GPU (CUDA 12.0)
+
+       .. code-block:: bash
+
           pip install -U brainpy[cuda12] -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
 
+    .. tab-item:: TPU
+
+       .. code-block:: bash
+
+          pip install -U brainpy[tpu] -f https://storage.googleapis.com/jax-releases/libtpu_releases.html
+
 
 For more information, please see `installation <quickstart/installation.html>`_ section.
 

docs/quickstart/installation.rst

@@ -17,8 +17,7 @@ To install brainpy with minimum requirements (has installed ``jax`` and ``jaxlib
 
 .. code-block:: bash
 
-    pip install brainpy # for CPU
-
+    pip install brainpy
 
 
 Minimum requirements (with dependencies)
@@ -35,6 +34,9 @@ To install brainpy with minimum requirements (only depends on ``jax``), you can
     pip install brainpy[cuda11_mini] -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html  # for CUDA 11.0
     pip install brainpy[cuda12_mini] -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html  # for CUDA 12.0
 
+    # or
+
+    pip install brainpy[tpu] -f https://storage.googleapis.com/jax-releases/libtpu_releases.html  # for google TPU
 
 
 CPU with all dependencies
@@ -61,6 +63,18 @@ To install a GPU-only version of BrainPy, you can run
 
 
 
+TPU with all dependencies
+-------------------------
+
+BrainPy supports Google Cloud TPU. To install BrainPy along with appropriate versions of jax,
+you can run the following in your cloud TPU VM:
+
+.. code-block:: bash
+
+    pip install brainpy[tpu] -f https://storage.googleapis.com/jax-releases/libtpu_releases.html  # for google TPU
+
+
+
 ``brainpylib``
 --------------