torch>=2.0.0
numpy>=1.23
scipy>=1.10
psutil>=5.9

[dev]
ipython
pytest
ruff
mypy
build
twine

[explorer]
scikit-learn>=1.3
umap-learn>=0.5
pandas>=2.0

[lightning]
lightning>=2.2.0

[metrics]
pynvml>=11.5
