numpy
torch
torchvision
torchmetrics==0.9.3
pytorch-metric-learning>=1.6.3

[detection]
albumentations>=1.2.1

[dev]
black
isort
nbqa
flake8

[docs]
mkdocs-material
mkdocstrings[python]
griffe
mkdocs-gen-files
mkdocs-section-index
mkdocs-literate-nav

[ignite]
pytorch-ignite==0.4.9

[lightning]
pytorch-lightning

[record-keeper]
record-keeper>=0.9.32
tensorboard

[timm]
timm
