.gitattributes
.gitignore
.gitlint
.pre-commit-config.yaml
.python-version
.readthedocs.yaml
LICENSE
README.md
pyproject.toml
uv.lock
.claude/CLAUDE.md
.github/workflows/ci.yaml
.github/workflows/release.yaml
docs/.gitignore
docs/Makefile
docs/conf.py
docs/custom_loops.rst
docs/design.rst
docs/examples.rst
docs/index.rst
docs/migration.rst
docs/references.bib
docs/references.rst
docs/requirements.txt
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/api/index.rst
docs/api/sbijax.experimental.rst
docs/api/sbijax.mcmc.rst
docs/api/sbijax.nn.rst
docs/api/sbijax.rst
docs/api/sbijax.simulators.rst
docs/api/sbijax.util.rst
docs/notebooks/eeg_data_example.ipynb
docs/notebooks/examples.ipynb
docs/notebooks/getting_started.ipynb
docs/notebooks/more_detailed_intro.ipynb
examples/nass_nle.py
examples/nass_smcabc.py
examples/nle.py
examples/npse.py
examples/sequential_npe.py
examples/simulators.py
examples/smcabc.py
examples/snle.py
experiments/sabc-performance/.gitignore
experiments/sabc-performance/README.md
experiments/sabc-performance/main.py
experiments/sabc-performance/requirements.txt
experiments/sabc-performance/setup.sh
experiments/sabc-performance/tasks.py
experiments/sabc-performance/adapters/__init__.py
experiments/sabc-performance/adapters/mlx_adapter.py
experiments/sabc-performance/adapters/numpy_adapter.py
experiments/sabc-performance/adapters/sbijax_adapter.py
experiments/sabc-performance/results/posterior_gaussian_mixture.png
experiments/sabc-performance/results/posterior_mixture_distractors.png
experiments/sabc-performance/results/posterior_two_moons.png
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/conftest.py
sbijax/_src/diagnostics/__init__.py
sbijax/_src/diagnostics/convergence.py
sbijax/_src/diagnostics/convergence_test.py
sbijax/_src/diagnostics/sbc.py
sbijax/_src/diagnostics/sbc_test.py
sbijax/_src/experimental/__init__.py
sbijax/_src/experimental/_truncated.py
sbijax/_src/experimental/aio.py
sbijax/_src/experimental/aio_test.py
sbijax/_src/experimental/cmpe.py
sbijax/_src/experimental/cmpe_objective_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/inference/__init__.py
sbijax/_src/inference/_sample_info.py
sbijax/_src/inference/_sample_info_test.py
sbijax/_src/inference/conformance_test.py
sbijax/_src/inference/sequential.py
sbijax/_src/inference/sequential_objective_test.py
sbijax/_src/inference/abc/__init__.py
sbijax/_src/inference/abc/_sabc_engine.py
sbijax/_src/inference/abc/_sampler.py
sbijax/_src/inference/abc/_smcabc_engine.py
sbijax/_src/inference/abc/conformance_test.py
sbijax/_src/inference/abc/sabc.py
sbijax/_src/inference/abc/sabc_return_test.py
sbijax/_src/inference/abc/smcabc.py
sbijax/_src/inference/likelihood/__init__.py
sbijax/_src/inference/likelihood/nle.py
sbijax/_src/inference/likelihood/nle_objective_test.py
sbijax/_src/inference/likelihood/snle.py
sbijax/_src/inference/posterior/__init__.py
sbijax/_src/inference/posterior/_sampling.py
sbijax/_src/inference/posterior/fmpe.py
sbijax/_src/inference/posterior/fmpe_objective_test.py
sbijax/_src/inference/posterior/npe.py
sbijax/_src/inference/posterior/npe_objective_test.py
sbijax/_src/inference/posterior/npse.py
sbijax/_src/inference/posterior/npse_test.py
sbijax/_src/inference/ratio/__init__.py
sbijax/_src/inference/ratio/nre.py
sbijax/_src/inference/summary/__init__.py
sbijax/_src/inference/summary/_compose.py
sbijax/_src/inference/summary/_summary_net.py
sbijax/_src/inference/summary/compose_test.py
sbijax/_src/inference/summary/conformance_test.py
sbijax/_src/inference/summary/nass.py
sbijax/_src/inference/summary/nasss.py
sbijax/_src/inference/summary/summary_objective_test.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/sampler.py
sbijax/_src/mcmc/sampler_test.py
sbijax/_src/mcmc/slice.py
sbijax/_src/mcmc/slice_test.py
sbijax/_src/mcmc/util.py
sbijax/_src/mcmc/util_test.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/simulate/__init__.py
sbijax/_src/simulate/simulate.py
sbijax/_src/simulate/simulate_test.py
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/train/__init__.py
sbijax/_src/train/_types.py
sbijax/_src/train/_types_test.py
sbijax/_src/train/sample.py
sbijax/_src/train/sample_test.py
sbijax/_src/train/train.py
sbijax/_src/train/train_test.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.py
sbijax/_src/util/train_state.py
sbijax/_src/util/train_test.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