torch>=1.8.2
torchvision>=0.9.2
numpy
tqdm
wandb

[dev]
nox
pytest
mkdocs
