Lower jax version requirement

Author: rafaelha

pyproject.toml

@@ -12,7 +12,7 @@ dependencies = [
     "equinox>=0.13.2",
     "galois>=0.4.7",
     "ipywidgets>=8.1.8",
-    "jax>=0.8.0",
+    "jax>=0.5.0",
     "lxml>=6.0.2",
     "numpy>=2.2.6",
     "pyperclip>=1.11.0",