Stop supporting jax below 0.4.34

Author: ae-foster

pyproject.toml

@@ -26,8 +26,7 @@ classifiers = [
     "Programming Language :: Python :: 3.12",
 ]
 dependencies = [
-    "jax>=0.4.10; python_version < '3.13'",
-    "jax>=0.4.34; python_version >= '3.13'",
+    "jax>=0.4.34",
     "jaxtyping>=0.2.36",
     "numpy>=1.25; python_version < '3.13'",
     "numpy>=2.1; python_version >= '3.13'",
@@ -53,7 +52,7 @@ build-backend = "hatchling.build"
 [dependency-groups]
 dev = [
     "flax>=0.10.2",
-    "jax[cuda12]>=0.4.10",
+    "jax[cuda12]>=0.4.34",
     "parameterized>=0.9.0",
     "pre-commit-uv>=4.1.4",
     "pytest>=8.3.4",