numpy
tensorboardX
gitpython
torch
