From 3477d0c779b8ed0822f05bee90b14d8eaf5d4cb1 Mon Sep 17 00:00:00 2001
From: Costa Huang <costa.huang@outlook.com>
Date: Thu, 23 Jun 2022 10:26:31 -0400
Subject: [PATCH] Fix instruction for installing cuda jax

---
 README.md | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/README.md b/README.md
index 040c21d..ccb4dd3 100644
--- a/README.md
+++ b/README.md
@@ -70,7 +70,7 @@ To install, run
 ```bash
 poetry install
 # For GPU support run
-pip install "jax[cuda]==0.3.10" -f https://storage.googleapis.com/jax-releases/jax_releases.html
+pip install "jax[cuda]==0.3.10" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
 ```
 
 For further instructions on running this code on GPU, please follow instructions from [the official repository](https://github.com/google/jax).