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: 702886821
google-deepmind · Dec 11, 2024 · c27bb3c · c27bb3c
1 parent af38d6e
commit c27bb3c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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