torch>=1.10
gpytorch>=1.7
scipy
multipledispatch
pyro-ppl>=1.8.0

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

[test]
pytest
pytest-cov

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