numpy
gitpython
rich

[ds]
torch
matplotlib
scipy
