numpy
jax[cuda12]
jax_dataloader
jupyter
matplotlib
cmake
tensorflow-cpu
tensorflow-probability
optax
distrax
flax
optuna
numpyro
flowMC
pytorch-lightning
getdist

[dev]
jaxili[docs,lint,release,test]

[docs]
myst-parser
numpydoc
sphinx
sphinx-book-theme
nbsphinx
tf-keras

[lint]
black
isort

[release]
build
twine

[test]
pytest
pytest-cov
pytest-emoji
pytest-pydocstyle
sbibm
sbi==0.23.3
