matplotlib
numpy
scipy
tqdm
torch

pytest
hypothesis
