torch>=1.11
gpytorch==1.9.0
linear_operator==0.1.1
scipy
multipledispatch
pyro-ppl>=1.8.2

[dev]
pytest
pytest-cov
flake8
ufmt
flake8-docstrings
black==22.3.0
usort==1.0.2
sphinx

[test]
pytest
pytest-cov

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