tqdm
numpy
scikit-learn
torch
scipy
numba

[tests]
pytest
pytest-cov
