fix URL redirect

Author: partev

docs/index.rst

@@ -256,7 +256,7 @@ Related Libraries
 .. _distrax_hmm: https://github.com/deepmind/distrax/blob/master/distrax/_src/utils/hmm.py
 .. _filterpy: https://github.com/rlabbe/Kalman-and-Bayesian-Filters-in-Python
 .. _hmmlearn: https://github.com/hmmlearn/hmmlearn
-.. _JAX: https://github.com/google/jax
+.. _JAX: https://github.com/jax-ml/jax
 .. _JSL: https://github.com/probml/JSL
 .. _linderman-ssm-jax: https://github.com/lindermanlab/ssm-jax
 .. _linderman-ssm-numpy: https://github.com/lindermanlab/ssm