numpy
colorama
mypy
tensorboard
tensorflow
torch
