torch>=1.5
scipy
networkx
pytest
sacred
PyYAML
neptune-contrib
psutil
