numpy
jax[cuda12]
jupyter
matplotlib
cmake
tensorflow
tensorflow-probability==0.24.0
tf_keras
optax
distrax
flax
torch
pytorch-lightning
optuna
numpyro
flowMC
getdist

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

[docs]
myst-parser
numpydoc
sphinx
sphinx-book-theme

[lint]
black
isort

[release]
build
twine

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