torch>=2.8.0
torch-geometric>=2.6.1
safetensors>=0.4

[baselines-graphcast]

[baselines-ode]
torchdiffeq>=0.2.3

[cli]
pyyaml>=6

[control-lmi]
cvxpy>=1.4.0

[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[train]>=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

[explain]
captum>=0.7

[export]
onnx>=1.16.0

[federated]

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

[md]
mdtraj>=1.9.9

[mpc]
osqp>=0.6.0
scipy>=1.11.0

[msm]
deeptime>=0.4

[probabilistic]

[ray]
ray[train]>=2.9.0

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

[sparse]

[symmetry]
networkx>=3.0

[tda]

[tdl]
