colorama
matplotlib
numpy
tqdm
