numpy
torch

[colorama]
colorama>=0.4.3
