numpy>=1.18.5
scipy<=1.7.3,>=1.4.1
scikit-learn>=0.23.2
bayesian-optimization>=1.2.0

[dev]
black>=21.1.0
build>=0.7.0
mypy>=0.910
twine>=3.7.1
absl-py>=0.13.0
sphinx==4.1.1
sphinx-rtd-theme==0.5.2
sphinx-autodoc-typehints==1.12.0
matplotlib==3.2.1
nbsphinx==0.8.7
pandoc==2.0.1
pandocfilters==1.5.0
ipython==7.30.1
ipykernel==6.6.0

[docs]
sphinx==4.1.1
sphinx-rtd-theme==0.5.2
sphinx-autodoc-typehints==1.12.0
matplotlib==3.2.1
nbsphinx==0.8.7
pandoc==2.0.1
pandocfilters==1.5.0
ipython==7.30.1
ipykernel==6.6.0

[hnswlib]
pybind11>=2.5.0
hnswlib>=0.6.0

[jax_cpu]
jax[cpu]>=0.2.26
tensorflow-probability[jax]>=0.16.0

[jax_cuda]
jax[cuda11_cudnn805]
tensorflow-probability[jax]>=0.16.0

[jax_cuda11_cudnn805]
jax[cuda11_cudnn805]
tensorflow-probability[jax]>=0.16.0

[jax_cuda11_cudnn82]
jax[cuda11_cudnn82]
tensorflow-probability[jax]>=0.16.0

[tests]
absl-py>=0.13.0
