torch
torchvision
tqdm
scipy
matplotlib

[test]
pytest
