typing_extensions
pyre_extensions
gpytorch==1.14
linear_operator==0.6
torch>=2.0.1
pyro-ppl>=1.8.4
scipy
multipledispatch
threadpoolctl

[dev]
pytest
pytest-cov
flake8
ufmt
flake8-docstrings
black==24.4.2
ruff-api==0.1.0
stdlibs==2024.1.28
ufmt==2.8.0
usort==1.0.8.post1
sphinx
sphinx-rtd-theme

[test]
pytest
pytest-cov

[tutorials]
ax-platform
cma
jupyter
kaleido
matplotlib
memory_profiler
papermill
pykeops
torchvision
mdformat
pandas
lxml
mdformat-myst
tabulate
