numpy
torch
torchvision
torchmetrics
pytorch-metric-learning>=1.3.1.dev0

[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.31

[timm]
timm
