torch>=1.9
gpytorch>=1.6
scipy
multipledispatch

[dev]
pytest
pytest-cov
flake8
sphinx
black==21.4b2
libcst==0.3.19
usort==0.6.4
ufmt

[test]
pytest
pytest-cov

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