Update references to JAX's GitHub repo

JAX has moved from https://github.com/google/jax to https://github.com/jax-ml/jax PiperOrigin-RevId: 702886844
google · Dec 5, 2024 · 8ae5b71 · 8ae5b71
1 parent d008bcf
commit 8ae5b71
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/README.md b/README.md
@@ -21,7 +21,7 @@ pip install "git+https://github.com/google/init2winit.git#egg=init2winit"
 pip install --upgrade jax jaxlib==0.1.66+cuda111 -f https://storage.googleapis.com/jax-releases/jax_releases.html
 ```
 
-where `cuda111` corresponds to the installed version of CUDA. For more Jax install information see the [Jax README](https://github.com/google/jax#installation).
+where `cuda111` corresponds to the installed version of CUDA. For more Jax install information see the [Jax README](https://github.com/jax-ml/jax#installation).
 
 ## Usage