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

[cpu]
tensorflow-cpu<2.16,>=1.13.1

[gpu]
tensorflow<2.16,>=1.13.1

[tests]
pytest

[visualization]
matplotlib
seaborn
