jax>=0.4.35
jaxtyping>=0.2.30
joblib>=1.4
psutil>=5.9
numpy>=1.26
optax>=0.2
h5py>=3.11
grain>=0.2
array-record>=0.8.1
equinox>=0.13.2
zstandard>=0.21.0
msgpack-numpy>=0.4.8
proxide>=0.1.0a3
typer>=0.12.0

[benchmark]
torch>=2.12.0
prody>=2.6.1

[dev]
ruff
ty
jaxlint>=0.1.0a1
sphinx
sphinx-autodoc-typehints
sphinx-autobuild
myst-parser
sphinx-copybutton
sphinx-design
sphinxext-rediraffe
sphinx-book-theme
myst-nb
pytest
pytest-cov
chex
foldcomp
mdtraj

[docs]
sphinx
sphinx-autodoc-typehints
sphinx-autobuild
myst-parser
sphinx-copybutton
sphinx-design
sphinxext-rediraffe
sphinx-book-theme
myst-nb

[foldcomp]
foldcomp

[tests]
pytest
pytest-cov
chex
ty
foldcomp
pytest-xdist
mdtraj
beartype>=0.18.0
biopython>=1.83
dm-tree>=0.1.8
