torch>=1.8.2
torchvision>=0.9.2
numpy
tqdm
wandb

[dev]
nox
pytest
mkdocs

[test]
pytest
mkdocs
