torch>=2.2
networkx>=3.0
numpy>=1.24
opt-einsum>=3.3

[all]
nbn[bench,bounds,examples,gp,mcmc,neural]

[bench]
scipy>=1.10
pyyaml>=6.0
tqdm>=4.65
pandas>=2.0
pyarrow>=14.0
pgmpy>=1.0
scikit-learn>=1.6
pomegranate>=1.0
matplotlib>=3.7
seaborn>=0.12
psutil>=5.9
packaging>=23.0
certifi>=2023.7

[bounds]
scipy>=1.10

[dev]
pytest>=8.0
pytest-cov>=5.0
pytest-xdist>=3.5
ruff<1.0,>=0.6
mypy>=1.9
pre-commit>=3.7
ipykernel>=6.0
nbmake>=1.5

[docs]
sphinx>=7.2
pydata-sphinx-theme>=0.15
myst-parser>=2.0
myst-nb>=1.0
sphinx-copybutton>=0.5
sphinx-design>=0.5

[examples]
matplotlib>=3.7
jupyterlab>=4.0
ipykernel>=6.0
zuko<2.0,>=1.2

[gp]
gpytorch>=1.11

[mcmc]
pyro-ppl>=1.9

[neural]
zuko<2.0,>=1.2
