diff --git a/README.md b/README.md
index 975cdcf..c15ce55 100644
--- a/README.md
+++ b/README.md
@@ -14,7 +14,7 @@ Model (LLM) by [Google DeepMind](https://deepmind.google/), based on Gemini
 research and technology.
 
 This repository contains an inference implementation and examples, based on the
-[Flax](https://github.com/google/flax) and [JAX](https://github.com/google/jax).
+[Flax](https://github.com/google/flax) and [JAX](https://github.com/jax-ml/jax).
 
 ### Learn more about Gemma