.gitattributes
.gitignore
.gitlint
.pre-commit-config.yaml
.python-version
.readthedocs.yaml
LICENSE
Makefile
README.md
pyproject.toml
uv.lock
.github/workflows/ci.yaml
.github/workflows/examples.yaml
.github/workflows/release.yaml
docs/.gitignore
docs/Makefile
docs/conf.py
docs/examples.rst
docs/index.rst
docs/references.bib
docs/references.rst
docs/requirements.txt
docs/sbijax.experimental.rst
docs/sbijax.mcmc.rst
docs/sbijax.nn.rst
docs/sbijax.rst
docs/sbijax.simulators.rst
docs/sbijax.util.rst
docs/_static/sticker.png
docs/_static/sticker_empty.png
docs/_static/sticker_empty_small.png
docs/_static/sticker_middle.png
docs/_static/stickers.key
docs/_static/theme.css
docs/_static/stickers/stickers.004.png
docs/_static/stickers/stickers.005.png
docs/_static/stickers/stickers.006.png
docs/notebooks/density_estimators.ipynb
docs/notebooks/eeg_data_example.ipynb
docs/notebooks/examples.ipynb
docs/notebooks/figure_styling.ipynb
docs/notebooks/getting_started.ipynb
docs/notebooks/high_dimensional_inference.ipynb
docs/notebooks/more_detailed_intro.ipynb
docs/notebooks/neural_networks.ipynb
examples/gaussian_linear-aio.py
examples/gaussian_linear-smcabc.py
examples/mixture_model-cmpe.py
examples/mixture_model-fmpe.py
examples/mixture_model-nle.py
examples/mixture_model-npe.py
examples/mixture_model-npse.py
examples/mixture_model-nre.py
examples/simulators.py
examples/slcp-fmpe.py
examples/slcp-nass_nle.py
examples/slcp-nass_smcabc.py
examples/slcp-snle.py
manuscript+supplement/.gitignore
manuscript+supplement/README.md
manuscript+supplement/requirements.txt
manuscript+supplement/experimental_code/convert.sh
sbijax/__init__.py
sbijax.egg-info/PKG-INFO
sbijax.egg-info/SOURCES.txt
sbijax.egg-info/dependency_links.txt
sbijax.egg-info/requires.txt
sbijax.egg-info/top_level.txt
sbijax/_src/__init__.py
sbijax/_src/_ne_base.py
sbijax/_src/_sbi_base.py
sbijax/_src/cmpe.py
sbijax/_src/cmpe_test.py
sbijax/_src/conftest.py
sbijax/_src/fmpe.py
sbijax/_src/fmpe_test.py
sbijax/_src/nass.py
sbijax/_src/nass_test.py
sbijax/_src/nasss.py
sbijax/_src/nasss_test.py
sbijax/_src/nle.py
sbijax/_src/nle_test.py
sbijax/_src/npe.py
sbijax/_src/npe_test.py
sbijax/_src/nre.py
sbijax/_src/nre_test.py
sbijax/_src/snle.py
sbijax/_src/abc/__init__.py
sbijax/_src/abc/smc_abc.py
sbijax/_src/abc/smc_abc_test.py
sbijax/_src/experimental/__init__.py
sbijax/_src/experimental/aio.py
sbijax/_src/experimental/aio_test.py
sbijax/_src/experimental/npse.py
sbijax/_src/experimental/npse_test.py
sbijax/_src/experimental/nn/__init__.py
sbijax/_src/experimental/nn/make_score_network.py
sbijax/_src/experimental/nn/make_simformer.py
sbijax/_src/mcmc/__init__.py
sbijax/_src/mcmc/conftest.py
sbijax/_src/mcmc/irmh.py
sbijax/_src/mcmc/irmh_test.py
sbijax/_src/mcmc/mala.py
sbijax/_src/mcmc/mala_test.py
sbijax/_src/mcmc/nuts.py
sbijax/_src/mcmc/nuts_test.py
sbijax/_src/mcmc/rmh.py
sbijax/_src/mcmc/rmh_test.py
sbijax/_src/mcmc/slice.py
sbijax/_src/mcmc/slice_test.py
sbijax/_src/mcmc/util.py
sbijax/_src/nn/__init__.py
sbijax/_src/nn/make_consistency_model.py
sbijax/_src/nn/make_continuous_flow.py
sbijax/_src/nn/make_flow.py
sbijax/_src/nn/make_mdn.py
sbijax/_src/nn/make_mlp.py
sbijax/_src/nn/make_nass_network.py
sbijax/_src/nn/make_resnet.py
sbijax/_src/nn/nass_net.py
sbijax/_src/nn/nasss_net.py
sbijax/_src/plot/__init__.py
sbijax/_src/plot/plot.py
sbijax/_src/plot/styles/sbijax-bluish.mplstyle
sbijax/_src/plot/styles/sbijax-grayish.mplstyle
sbijax/_src/plot/styles/sbijax.mplstyle
sbijax/_src/simulators/__init__.py
sbijax/_src/simulators/conftest.py
sbijax/_src/simulators/hyperboloid.py
sbijax/_src/simulators/jansen_rit.py
sbijax/_src/simulators/mixture_model_distractors.py
sbijax/_src/simulators/simulators_test.py
sbijax/_src/simulators/sir.py
sbijax/_src/simulators/slcp.py
sbijax/_src/simulators/solar_dynamo.py
sbijax/_src/simulators/tree.py
sbijax/_src/simulators/two_moons.py
sbijax/_src/util/__init__.py
sbijax/_src/util/data.py
sbijax/_src/util/data_test.py
sbijax/_src/util/dataloader.py
sbijax/_src/util/early_stopping.py
sbijax/_src/util/train_state.py
sbijax/_src/util/types.py
sbijax/experimental/__init__.py
sbijax/experimental/nn/__init__.py
sbijax/mcmc/__init__.py
sbijax/nn/__init__.py
sbijax/simulators/__init__.py
sbijax/util/__init__.py