Skip to content

Commit bd1a711

Browse files
committed
add dependencies for jax sequence models to pyproject.toml. Requirements.txt need to be eliminated
1 parent 415a687 commit bd1a711

1 file changed

Lines changed: 3 additions & 1 deletion

File tree

pyproject.toml

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,9 @@ dependencies = [
1919
"torchinfo>=1.8.0",
2020
"joblib>=1.3.2",
2121
"scipy",
22-
"pytest"
22+
"pytest",
23+
"jax>=0.6.2",
24+
"equinox>=0.13.8",
2325
]
2426

2527
[project.urls]

0 commit comments

Comments
 (0)