From d88df1d92dac30fc367cb77d1c84a8e050d14f0f Mon Sep 17 00:00:00 2001
From: Jake Harmon <jakeharmon@google.com>
Date: Wed, 4 Dec 2024 15:46:04 -0800
Subject: [PATCH] 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
---
 README.md | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/README.md b/README.md
index 4f01df1..d7eff49 100644
--- a/README.md
+++ b/README.md
@@ -98,7 +98,7 @@ pip install -r vit_jax/requirements.txt
 pip install -r vit_jax/requirements-tpu.txt
 ```
 
-For newer versions of [JAX](https://github.com/google/jax), follow the instructions
+For newer versions of [JAX](https://github.com/jax-ml/jax), follow the instructions
 provided in the corresponding repository linked here. Note that installation
 instructions for CPU, GPU and TPU differs slightly.