Add jax to deps

singularitti · singularitti · commit bdf07a79a251 · 2024-02-17T20:15:26.000-05:00
diff --git a/pyproject.toml b/pyproject.toml
@@ -15,6 +15,7 @@ dependencies = [
     "pyyaml>=6.0.1",
     "matplotlib>=3.8.2",
     "numba>=0.58.1",
+    "jax>=0.4.24",
 ]
 requires-python = ">=3.11"
 readme = "README.md"
