matplotlib
numpy
pytest
scipy
tqdm
torch

pytest
hypothesis
