diff --git a/mjx/README.md b/mjx/README.md
index 4bb727bdee..f4796a5fed 100644
--- a/mjx/README.md
+++ b/mjx/README.md
@@ -9,7 +9,7 @@
 
 This package is a re-implementation of the
 [MuJoCo physics engine](https://github.com/google-deepmind/mujoco) in
-[JAX](https://github.com/google/jax). This library is developed and maintained
+[JAX](https://github.com/jax-ml/jax). This library is developed and maintained
 by Google DeepMind, and is kept up-to-date with the latest developments in
 MuJoCo itself.
 
diff --git a/mjx/mujoco/mjx/_src/math.py b/mjx/mujoco/mjx/_src/math.py
index c7fffbcf1a..bd3b601956 100644
--- a/mjx/mujoco/mjx/_src/math.py
+++ b/mjx/mujoco/mjx/_src/math.py
@@ -51,7 +51,7 @@ def norm(
   """Calculates a linalg.norm(x) that's safe for gradients at x=0.
 
   Avoids a poorly defined gradient for jnp.linal.norm(0) see
-  https://github.com/google/jax/issues/3058 for details
+  https://github.com/jax-ml/jax/issues/3058 for details
   Args:
     x: A jnp.array
     axis: The axis along which to compute the norm