numba
numpy
torch
scipy
pyyaml
click
tensorboardX
