Pin jax version temporarily (#1292)

Pin jax to `<0.4.34` until we fix bug listed in #1291
PlasmaControl · Oct 4, 2024 · 6c599b4 · 6c599b4
2 parents e35436d + e506d80
commit 6c599b4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/requirements.txt b/requirements.txt
@@ -2,7 +2,7 @@ colorama
 diffrax >= 0.4.1
 h5py >= 3.0.0, < 4.0
 interpax >= 0.3.3
-jax[cpu] >= 0.3.2, < 0.5.0
+jax[cpu] >= 0.3.2, < 0.4.34
 matplotlib >= 3.5.0, < 4.0.0
 mpmath >= 1.0.0, < 2.0
 netcdf4 >= 1.5.4, < 2.0