torch>=1.8.2
torchvision>=0.9.2
numpy
tqdm
wandb
ezkfg
pyyaml
urllib3==1.25.3

[dev]
nox
pytest
mkdocs

[test]
pytest
mkdocs
