update doc (#84)

Author: fishjojo

README.md

@@ -36,7 +36,7 @@ The dependent C library `pyscfadlib` can be compiled from source following the i
 `numpy`, `scipy`,
 `pyscf>=2.3.0`,
 `pyscfadlib>=0.1.11`, and
-`jax==0.7.0`.
+`jax>=0.7.0,<0.8`.
 
 Citing PySCFAD
 --------------

doc/source/getting_started/install.rst

@@ -8,7 +8,7 @@ Installation
 Python version support
 ----------------------
 
-Officially Python 3.10, 3.11, 3.12, and 3.13.
+Officially Python 3.11, 3.12, and 3.13.
 
 Installing from PyPI
 --------------------
@@ -51,8 +51,8 @@ the following command at the top directory of pyscfad.
    cd $HOME/pyscfad
    pip install -e .
 
-Installing pyscfadlib
-~~~~~~~~~~~~~~~~~~~~~
+Installing pyscfadlib (CPU only)
+~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
 
 pyscfadlib is the C extension to pyscfad that provides efficient gradient backpropagation implementations.
 Similarly, one can install pyscfadlib locally via pip.
@@ -89,14 +89,8 @@ Package                                                supported versions
 `numpy <https://numpy.org>`_                           >=1.17
 `scipy <https://scipy.org>`_                           >=1.7
 `h5py <https://www.h5py.org/>`_                        >=2.7
-`jax <https://jax.readthedocs.io/en/latest/>`_         ==0.4.35
+`jax <https://jax.readthedocs.io/en/latest/>`_         >=0.7.0,<0.8
 `pyscf <https://pyscf.org/>`_                          >=2.3.0
-`pyscfadlib <https://pypi.org/project/pyscfadlib/>`_   >=0.1.4
+`pyscfadlib <https://pypi.org/project/pyscfadlib/>`_   >=0.1.10
 =====================================================  ==================
 
-.. note::
-
-    Since jax version 0.4.36, the tracing machinery has been modified
-    to eliminate data-dependent tracing, which conflicts with pyscfad's flexibility.
-    A comprehensive update to ensure compatibility with later jax versions will be
-    introduced in the future pyscfad 0.2 release.