matplotlib
numpy
scipy
tqdm
torch
pytest
hypothesis
