numpy
torch>=2.2.0
einops
einconv

[doc]
mkdocs==1.4.3
mkdocs-material==9.1.17
mkdocstrings[python]==0.22.0
mkdocs-gallery==0.7.8
matplotlib
torchvision

[fine_tuning]
wandb
transformers
datasets
evaluate
py7zr
tqdm
torchmetrics
backpack-for-pytorch

[lint]
black
isort
flake8
mccabe
pycodestyle
pyflakes
pep8-naming
flake8-bugbear
flake8-comprehensions
flake8-tidy-imports
darglint
pydocstyle

[test]
coveralls
pytest
pytest-cov
pytest-optional-tests
ipdb
torchvision
matplotlib
imageio
memory_profiler
codetiming
