torch>=1.13.1

[test]
torchvision>=0.14.1
tqdm
wandb
