From cd6bbf01dc524f1b3b457f112c93b255d4679634 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?V=C3=ADt=20=C5=A1krh=C3=A1k?= Date: Fri, 30 Jan 2026 10:13:14 -0500 Subject: [PATCH] specify JAX version --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index 52fe5c45..3816eb60 100644 --- a/README.md +++ b/README.md @@ -47,7 +47,7 @@ typical install time ≈ 10 minutes. first install jax (with GPU support) -``` pip install "jax[cuda]" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html ``` +``` pip install "jax[cuda]==0.7" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html ``` second install colabdesign