We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 415a687 commit bd1a711Copy full SHA for bd1a711
1 file changed
pyproject.toml
@@ -19,7 +19,9 @@ dependencies = [
19
"torchinfo>=1.8.0",
20
"joblib>=1.3.2",
21
"scipy",
22
- "pytest"
+ "pytest",
23
+ "jax>=0.6.2",
24
+ "equinox>=0.13.8",
25
]
26
27
[project.urls]
0 commit comments