tensorflow
numpy
tensorflow-probability
tqdm
