torch>=1.7.1
gpytorch>=1.4
scipy

[dev]
pytest
pytest-cov
black
flake8
sphinx
sphinx-autodoc-typehints

[test]
pytest
pytest-cov

[tutorials]
ax-platform
cma
jupyter
matplotlib
memory_profiler
pykeops
torchvision
