pin pytest jax version

cgiovanetti · cgiovanetti · commit 4f6c62f70513 · 2026-01-20T11:10:47.000+01:00
diff --git a/pytest_requirements.txt b/pytest_requirements.txt
@@ -4,6 +4,6 @@ matplotlib
 diffrax
 equinox
 interpax
-jax
+jax==0.8.1
 pytest
 classy
