jax>=0.11.0
bayesflow>=2.0.12
numba>=0.63.1
ipywidgets>=8.1.8

[all]
ipython
ipykernel
jupyter
jupyterlab
line-profiler
nbconvert
pre-commit
ruff
myst-nb~=1.2
numpydoc~=1.8
pydata-sphinx-theme~=0.16
sphinx~=8.1.3
sphinx-polyversion==1.1
sphinx_design~=0.6
sphinxcontrib-bibtex~=2.6
snowballstemmer~=2.2.0
pytest
pytest-cov
pytest-rerunfailures
pytest-xdist

[cuda]
jax[cuda13]>=0.11.0

[dev]
jupyter
jupyterlab
line-profiler
pre-commit
ruff

[docs]
myst-nb~=1.2
numpydoc~=1.8
pydata-sphinx-theme~=0.16
sphinx~=8.1.3
sphinx-polyversion==1.1
sphinx_design~=0.6
sphinxcontrib-bibtex~=2.6
snowballstemmer~=2.2.0

[test]
nbconvert
ipython
ipykernel
pytest
pytest-cov
pytest-rerunfailures
pytest-xdist
