numpy
scikit-learn>0.20.0
annoy>=1.15.2
tqdm
dill

[cpu]
tensorflow-cpu

[gpu]
tensorflow

[tests]
pytest

[visualization]
matplotlib
seaborn
