matplotlib
numpy
torch
pytest
scikit-learn
scipy
tqdm