numpy>=1.24
torch>=2.5
pyyaml
tqdm

[dev]
pytest
pre-commit
ruff==0.15.20

[docs]
sphinx
sphinx-autoapi
sphinx-autobuild
sphinx-book-theme
myst-nb
matplotlib
pyro-ppl

[full]
torch-testbed[train]
torch-testbed[pyro]
matplotlib
wandb

[pyro]
pyro-ppl

[train]
hydra-core>=1.3

[wandb]
wandb
