numpy
gitpython

[ds]
torch
matplotlib
scipy
