torch>=2.8.0
torch-geometric>=2.6.1

[dask]
dask[distributed]>=2024.1.0

[dev]
build>=1.2.0
pytest>=8.4.0
pytest-cov>=6.0.0
pytest-xdist>=3.6.0
ruff>=0.15.0
pre-commit>=4.0.0
nbmake>=0.5.0
matplotlib>=3.8.0
scipy>=1.11.0
ipykernel>=6.29.0
pandas>=2.0.0
gymnasium>=1.0.0
h5py>=3.10.0

[distributed]
lightning!=2.6.2,!=2.6.3,>=2.4.0
ray>=2.9.0
dask[distributed]>=2024.1.0

[docs]
sphinx>=8.0.0
furo>=2024.8.6
sphinx-copybutton>=0.5.2

[equivariance]
e3nn>=0.5.1

[lightning]
lightning!=2.6.2,!=2.6.3,>=2.4.0

[mpc]
osqp>=0.6.0
scipy>=1.11.0

[msm]
deeptime>=0.4

[ray]
ray>=2.9.0

[rl]
gymnasium>=1.0.0
stable-baselines3>=2.3.0

[symmetry]
networkx>=3.0
