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).