diff --git a/README.md b/README.md
index 179a4de..88dbb33 100644
--- a/README.md
+++ b/README.md
@@ -2,7 +2,7 @@
 
 ![PyPI version](https://img.shields.io/pypi/v/jax-triton)
 
-The `jax-triton` repository contains integrations between [JAX](https://github.com/google/jax) and [Triton](https://github.com/openai/triton).
+The `jax-triton` repository contains integrations between [JAX](https://github.com/jax-ml/jax) and [Triton](https://github.com/openai/triton).
 
 Documentation can be found [here](https://jax-ml.github.io/jax-triton).
 
diff --git a/docs/index.md b/docs/index.md
index 8d510da..6c40bd0 100644
--- a/docs/index.md
+++ b/docs/index.md
@@ -1,6 +1,6 @@
 # JAX-Triton documentation
 
-JAX-Triton is a repository containing containing integrations between [JAX](https://github.com/google/jax)
+JAX-Triton is a repository containing containing integrations between [JAX](https://github.com/jax-ml/jax)
 and [Triton](https://github.com/openai/triton).
 
 JAX is a Python library for accelerated numerical computing and Triton is a Python library and compiler for writing custom GPU kernels.
@@ -20,7 +20,7 @@ For example you could run:
 ```bash
 $ pip install "jax[cuda11_cudnn82]" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
 ```
-Check out the [JAX installation guide](https://github.com/google/jax#pip-installation-gpu-cuda) for details.
+Check out the [JAX installation guide](https://github.com/jax-ml/jax#pip-installation-gpu-cuda) for details.
 
 ### Installation at HEAD