torch>=2.0.0
tqdm
numpy
matplotlib
scikit-learn

[benchmarks]
pandas<3,>=2.0

[dev]
black<25,>=24
mypy<2,>=1.11
pre-commit<4,>=3.8
ruff<1,>=0.6.9

[docs]
mkdocs<2,>=1.6
mkdocs-git-revision-date-localized-plugin<2,>=1.2
mkdocs-material<10,>=9.5
mkdocs-minify-plugin<1,>=0.8
mkdocs-section-index<1,>=0.3
mkdocstrings[python]<1,>=0.25

[notebooks]
ipykernel<7,>=6
jupyter<2,>=1.0

[tests]
pytest<9,>=8
