jax<=0.4.30
jaxlib<=0.4.30
scipy<1.13
jax-md
jax-sgmc
optax
dm-haiku
sympy
tree_math
cloudpickle
chex
blackjax==0.3.0
jaxopt
dm-tree
networkx

[all]
mdtraj
matplotlib
tomli
tomli-w

[docs]
sphinx>=3
sphinx-book-theme
sphinx-autodoc-typehints==1.11.1
myst-nb
matplotlib
sphinx-remove-toctrees

[stable]
jax==0.3.25
jaxlib==0.3.25
chex==0.1.5
orbax-checkpoint==0.1.1
orbax==0.1.1

[test]
pylint
pytest
pytest-mock
