torch>=2.0.0
scikit-learn>=1.0
matplotlib
tqdm
pandas

[dev]
nvitop
ruff
black
mypy
pytest

[doc]
sphinx
myst-parser
sphinx_rtd_theme
nbsphinx
