tqdm
torch
scipy
numpy

[dev]
pytest
black
flake8
