corider
pytorch-lightning==1.3
torchmetrics
python-dotenv
supers>=0.2
matplotlib
seaborn
coloredlogs
gputil>=1.4
py-cpuinfo>=7.0
ptflops>=0.6
joblib>=1.0
psutil

[build]
setuptools
wheel
twine

[dev]
wandb
pytest
flake8
mypy
black
isort>=5.7
flake8-black
pytest==6.2
pytest-cov
pytest-xdist
scikit-learn
umap-learn
ray[tune]==1.2
torchvision
lightning-bolts

[docs]
sphinx>=3.0
nbsphinx>=0.8
pandoc>=1.0
docutils>=0.16
sphinxcontrib-fulltoc>=1.0
sphinxcontrib-mockautodoc
sphinx-autodoc-typehints>=1.0
sphinx-paramlinks>=0.4.0
sphinx-togglebutton>=0.2
sphinx-copybutton>=0.3
sphinx-autoapi>=1.7
m2r2>=0.2

[extras]
lightning-bolts
ray[tune]~=1.2
scikit-learn
torchvision
umap-learn
