pandas
numpy
h5py
PyYAML>=5.1
scikit-learn
tqdm

[cpu]
pytorch==1.0.*

[gpu]
pytorch==1.0.*
