numpy
pandas
scipy
scikit-learn
pyyaml
tqdm
matplotlib

[cpu]
torch

[dev]
build
twine
pytest

[gpu]
torch
