torch>=0.4.0
scipy
numpy
torchvision
datajoint
nnfabrik
pytest
