torch>=1.8.1
gpytorch>=1.5.1
scipy

[dev]
pytest
pytest-cov
black
flake8
sphinx
usort

[test]
pytest
pytest-cov

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