torch>=2.8
pyyaml
psutil
dill
numpy

[dev]
pytest
ruff
scipy

[docs]
sphinx
sphinx_rtd_theme
