diff --git a/README.md b/README.md
index e791971..d497b94 100644
--- a/README.md
+++ b/README.md
@@ -63,7 +63,7 @@ jax.grad(loss_fn)(scores, labels)  # computes gradients w.r.t. scores.
 
 ## Installation
 
-See https://github.com/google/jax#installation for instructions on installing JAX.
+See https://github.com/jax-ml/jax#installation for instructions on installing JAX.
 
 We suggest installing the latest stable version of Rax by running:
 
diff --git a/docs/index.rst b/docs/index.rst
index 7d428a4..11774f3 100644
--- a/docs/index.rst
+++ b/docs/index.rst
@@ -26,7 +26,7 @@ Rax is a Learning-to-Rank (LTR) library built on top of JAX.
 Installation
 ------------
 
-See https://github.com/google/jax#pip-installation for instructions on
+See https://github.com/jax-ml/jax#pip-installation for instructions on
 installing JAX.
 
 We suggest installing the latest stable version of Rax by running::