├── .gitattributes ├── .github ├── ISSUE_TEMPLATE │ ├── bug_report.yml │ ├── config.yml │ ├── enhancement.md │ ├── meeting.md │ └── sampler_proposal.md ├── PULL_REQUEST_TEMPLATE.md └── workflows │ ├── benchmark.yml │ ├── build_documentation.yml │ ├── publish_documentation.yml │ ├── release.yml │ └── test.yml ├── .gitignore ├── .gitlint ├── .pre-commit-config.yaml ├── .readthedocs.yaml ├── CODE_OF_CONDUCT.md ├── CONTRIBUTING.md ├── GOVERNANCE.md ├── LICENSE ├── MANIFEST.in ├── Makefile ├── README.md ├── blackjax ├── __init__.py ├── adaptation │ ├── __init__.py │ ├── adjusted_mclmc_adaptation.py │ ├── base.py │ ├── chees_adaptation.py │ ├── mass_matrix.py │ ├── mclmc_adaptation.py │ ├── meads_adaptation.py │ ├── pathfinder_adaptation.py │ ├── step_size.py │ └── window_adaptation.py ├── base.py ├── diagnostics.py ├── mcmc │ ├── __init__.py │ ├── adjusted_mclmc.py │ ├── adjusted_mclmc_dynamic.py │ ├── barker.py │ ├── diffusions.py │ ├── dynamic_hmc.py │ ├── elliptical_slice.py │ ├── ghmc.py │ ├── hmc.py │ ├── integrators.py │ ├── mala.py │ ├── marginal_latent_gaussian.py │ ├── mclmc.py │ ├── metrics.py │ ├── nuts.py │ ├── periodic_orbital.py │ ├── proposal.py │ ├── random_walk.py │ ├── rmhmc.py │ ├── termination.py │ └── trajectory.py ├── optimizers │ ├── __init__.py │ ├── dual_averaging.py │ └── lbfgs.py ├── progress_bar.py ├── sgmcmc │ ├── __init__.py │ ├── csgld.py │ ├── diffusions.py │ ├── gradients.py │ ├── sghmc.py │ ├── sgld.py │ └── sgnht.py ├── smc │ ├── __init__.py │ ├── adaptive_persistent_sampling.py │ ├── adaptive_tempered.py │ ├── base.py │ ├── ess.py │ ├── from_mcmc.py │ ├── inner_kernel_tuning.py │ ├── partial_posteriors_path.py │ ├── persistent_sampling.py │ ├── pretuning.py │ ├── resampling.py │ ├── solver.py │ ├── tempered.py │ ├── tuning │ │ ├── __init__.py │ │ ├── from_kernel_info.py │ │ └── from_particles.py │ └── waste_free.py ├── types.py ├── util.py └── vi │ ├── __init__.py │ ├── meanfield_vi.py │ ├── pathfinder.py │ ├── schrodinger_follmer.py │ └── svgd.py ├── codecov.yml ├── docs ├── _static │ ├── blackjax.png │ └── custom.css ├── bib.rst ├── conf.py ├── examples │ ├── data │ │ ├── blackjax.png │ │ └── google.csv │ ├── howto_custom_gradients.md │ ├── howto_metropolis_within_gibbs.md │ ├── howto_other_frameworks.md │ ├── howto_reproduce_the_blackjax_image.md │ ├── howto_sample_multiple_chains.md │ ├── howto_use_aesara.md │ ├── howto_use_numpyro.md │ ├── howto_use_oryx.md │ ├── howto_use_pymc.md │ ├── howto_use_tfp.md │ ├── quickstart.md │ └── scatter.gif ├── index.md └── refs.bib ├── jupytex.toml ├── mypy.ini ├── pyproject.toml ├── pytest.ini ├── requirements-doc.txt ├── requirements.txt ├── setup.cfg └── tests ├── __init__.py ├── adaptation ├── __init__.py ├── test_adaptation.py ├── test_mass_matrix.py └── test_step_size.py ├── mcmc ├── __init__.py ├── test_barker.py ├── test_integrators.py ├── test_latent_gaussian.py ├── test_metrics.py ├── test_proposal.py ├── test_random_walk_without_chex.py ├── test_sampling.py ├── test_trajectory.py └── test_uturn.py ├── optimizers ├── __init__.py ├── test_optimizers.py └── test_pathfinder.py ├── smc ├── __init__.py ├── test_inner_kernel_tuning.py ├── test_kernel_compatibility.py ├── test_partial_posteriors_smc.py ├── test_persistent_sampling.py ├── test_pretuning.py ├── test_resampling.py ├── test_smc.py ├── test_smc_ess.py ├── test_solver.py ├── test_tempered_smc.py └── test_waste_free_smc.py ├── test_benchmarks.py ├── test_compilation.py ├── test_diagnostics.py ├── test_util.py └── vi ├── __init__.py ├── test_meanfield_vi.py ├── test_schrodinger_follmer.py └── test_svgd.py /.gitattributes: -------------------------------------------------------------------------------- 1 | blackjax/_version.py export-subst 2 | -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/bug_report.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/.github/ISSUE_TEMPLATE/bug_report.yml -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/config.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/.github/ISSUE_TEMPLATE/config.yml -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/enhancement.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/.github/ISSUE_TEMPLATE/enhancement.md -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/meeting.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/.github/ISSUE_TEMPLATE/meeting.md -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/sampler_proposal.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/.github/ISSUE_TEMPLATE/sampler_proposal.md -------------------------------------------------------------------------------- /.github/PULL_REQUEST_TEMPLATE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/.github/PULL_REQUEST_TEMPLATE.md -------------------------------------------------------------------------------- /.github/workflows/benchmark.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/.github/workflows/benchmark.yml -------------------------------------------------------------------------------- /.github/workflows/build_documentation.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/.github/workflows/build_documentation.yml -------------------------------------------------------------------------------- /.github/workflows/publish_documentation.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/.github/workflows/publish_documentation.yml -------------------------------------------------------------------------------- /.github/workflows/release.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/.github/workflows/release.yml -------------------------------------------------------------------------------- /.github/workflows/test.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/.github/workflows/test.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/.gitignore -------------------------------------------------------------------------------- /.gitlint: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/.gitlint -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /.readthedocs.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/.readthedocs.yaml -------------------------------------------------------------------------------- /CODE_OF_CONDUCT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/CODE_OF_CONDUCT.md -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /GOVERNANCE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/GOVERNANCE.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/LICENSE -------------------------------------------------------------------------------- /MANIFEST.in: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/MANIFEST.in -------------------------------------------------------------------------------- /Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/Makefile -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/README.md -------------------------------------------------------------------------------- /blackjax/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/blackjax/__init__.py -------------------------------------------------------------------------------- /blackjax/adaptation/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/blackjax/adaptation/__init__.py -------------------------------------------------------------------------------- /blackjax/adaptation/adjusted_mclmc_adaptation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/blackjax/adaptation/adjusted_mclmc_adaptation.py -------------------------------------------------------------------------------- /blackjax/adaptation/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/blackjax/adaptation/base.py -------------------------------------------------------------------------------- /blackjax/adaptation/chees_adaptation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/blackjax/adaptation/chees_adaptation.py -------------------------------------------------------------------------------- /blackjax/adaptation/mass_matrix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/blackjax/adaptation/mass_matrix.py -------------------------------------------------------------------------------- /blackjax/adaptation/mclmc_adaptation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/blackjax/adaptation/mclmc_adaptation.py -------------------------------------------------------------------------------- /blackjax/adaptation/meads_adaptation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/blackjax/adaptation/meads_adaptation.py -------------------------------------------------------------------------------- /blackjax/adaptation/pathfinder_adaptation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/blackjax/adaptation/pathfinder_adaptation.py -------------------------------------------------------------------------------- /blackjax/adaptation/step_size.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/blackjax/adaptation/step_size.py -------------------------------------------------------------------------------- /blackjax/adaptation/window_adaptation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/blackjax/adaptation/window_adaptation.py -------------------------------------------------------------------------------- /blackjax/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/blackjax/base.py -------------------------------------------------------------------------------- /blackjax/diagnostics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/blackjax/diagnostics.py -------------------------------------------------------------------------------- /blackjax/mcmc/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/blackjax/mcmc/__init__.py -------------------------------------------------------------------------------- /blackjax/mcmc/adjusted_mclmc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/blackjax/mcmc/adjusted_mclmc.py -------------------------------------------------------------------------------- /blackjax/mcmc/adjusted_mclmc_dynamic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/blackjax/mcmc/adjusted_mclmc_dynamic.py -------------------------------------------------------------------------------- /blackjax/mcmc/barker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/blackjax/mcmc/barker.py -------------------------------------------------------------------------------- /blackjax/mcmc/diffusions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/blackjax/mcmc/diffusions.py -------------------------------------------------------------------------------- /blackjax/mcmc/dynamic_hmc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/blackjax/mcmc/dynamic_hmc.py -------------------------------------------------------------------------------- /blackjax/mcmc/elliptical_slice.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/blackjax/mcmc/elliptical_slice.py -------------------------------------------------------------------------------- /blackjax/mcmc/ghmc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/blackjax/mcmc/ghmc.py -------------------------------------------------------------------------------- /blackjax/mcmc/hmc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/blackjax/mcmc/hmc.py -------------------------------------------------------------------------------- /blackjax/mcmc/integrators.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/blackjax/mcmc/integrators.py -------------------------------------------------------------------------------- /blackjax/mcmc/mala.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/blackjax/mcmc/mala.py -------------------------------------------------------------------------------- /blackjax/mcmc/marginal_latent_gaussian.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/blackjax/mcmc/marginal_latent_gaussian.py -------------------------------------------------------------------------------- /blackjax/mcmc/mclmc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/blackjax/mcmc/mclmc.py -------------------------------------------------------------------------------- /blackjax/mcmc/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/blackjax/mcmc/metrics.py -------------------------------------------------------------------------------- /blackjax/mcmc/nuts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/blackjax/mcmc/nuts.py -------------------------------------------------------------------------------- /blackjax/mcmc/periodic_orbital.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/blackjax/mcmc/periodic_orbital.py -------------------------------------------------------------------------------- /blackjax/mcmc/proposal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/blackjax/mcmc/proposal.py -------------------------------------------------------------------------------- /blackjax/mcmc/random_walk.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/blackjax/mcmc/random_walk.py -------------------------------------------------------------------------------- /blackjax/mcmc/rmhmc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/blackjax/mcmc/rmhmc.py -------------------------------------------------------------------------------- /blackjax/mcmc/termination.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/blackjax/mcmc/termination.py -------------------------------------------------------------------------------- /blackjax/mcmc/trajectory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/blackjax/mcmc/trajectory.py -------------------------------------------------------------------------------- /blackjax/optimizers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/blackjax/optimizers/__init__.py -------------------------------------------------------------------------------- /blackjax/optimizers/dual_averaging.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/blackjax/optimizers/dual_averaging.py -------------------------------------------------------------------------------- /blackjax/optimizers/lbfgs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/blackjax/optimizers/lbfgs.py -------------------------------------------------------------------------------- /blackjax/progress_bar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/blackjax/progress_bar.py -------------------------------------------------------------------------------- /blackjax/sgmcmc/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/blackjax/sgmcmc/__init__.py -------------------------------------------------------------------------------- /blackjax/sgmcmc/csgld.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/blackjax/sgmcmc/csgld.py -------------------------------------------------------------------------------- /blackjax/sgmcmc/diffusions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/blackjax/sgmcmc/diffusions.py -------------------------------------------------------------------------------- /blackjax/sgmcmc/gradients.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/blackjax/sgmcmc/gradients.py -------------------------------------------------------------------------------- /blackjax/sgmcmc/sghmc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/blackjax/sgmcmc/sghmc.py -------------------------------------------------------------------------------- /blackjax/sgmcmc/sgld.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/blackjax/sgmcmc/sgld.py -------------------------------------------------------------------------------- /blackjax/sgmcmc/sgnht.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/blackjax/sgmcmc/sgnht.py -------------------------------------------------------------------------------- /blackjax/smc/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/blackjax/smc/__init__.py -------------------------------------------------------------------------------- /blackjax/smc/adaptive_persistent_sampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/blackjax/smc/adaptive_persistent_sampling.py -------------------------------------------------------------------------------- /blackjax/smc/adaptive_tempered.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/blackjax/smc/adaptive_tempered.py -------------------------------------------------------------------------------- /blackjax/smc/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/blackjax/smc/base.py -------------------------------------------------------------------------------- /blackjax/smc/ess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/blackjax/smc/ess.py -------------------------------------------------------------------------------- /blackjax/smc/from_mcmc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/blackjax/smc/from_mcmc.py -------------------------------------------------------------------------------- /blackjax/smc/inner_kernel_tuning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/blackjax/smc/inner_kernel_tuning.py -------------------------------------------------------------------------------- /blackjax/smc/partial_posteriors_path.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/blackjax/smc/partial_posteriors_path.py -------------------------------------------------------------------------------- /blackjax/smc/persistent_sampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/blackjax/smc/persistent_sampling.py -------------------------------------------------------------------------------- /blackjax/smc/pretuning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/blackjax/smc/pretuning.py -------------------------------------------------------------------------------- /blackjax/smc/resampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/blackjax/smc/resampling.py -------------------------------------------------------------------------------- /blackjax/smc/solver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/blackjax/smc/solver.py -------------------------------------------------------------------------------- /blackjax/smc/tempered.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/blackjax/smc/tempered.py -------------------------------------------------------------------------------- /blackjax/smc/tuning/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /blackjax/smc/tuning/from_kernel_info.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/blackjax/smc/tuning/from_kernel_info.py -------------------------------------------------------------------------------- /blackjax/smc/tuning/from_particles.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/blackjax/smc/tuning/from_particles.py -------------------------------------------------------------------------------- /blackjax/smc/waste_free.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/blackjax/smc/waste_free.py -------------------------------------------------------------------------------- /blackjax/types.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/blackjax/types.py -------------------------------------------------------------------------------- /blackjax/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/blackjax/util.py -------------------------------------------------------------------------------- /blackjax/vi/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/blackjax/vi/__init__.py -------------------------------------------------------------------------------- /blackjax/vi/meanfield_vi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/blackjax/vi/meanfield_vi.py -------------------------------------------------------------------------------- /blackjax/vi/pathfinder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/blackjax/vi/pathfinder.py -------------------------------------------------------------------------------- /blackjax/vi/schrodinger_follmer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/blackjax/vi/schrodinger_follmer.py -------------------------------------------------------------------------------- /blackjax/vi/svgd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/blackjax/vi/svgd.py -------------------------------------------------------------------------------- /codecov.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/codecov.yml -------------------------------------------------------------------------------- /docs/_static/blackjax.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/docs/_static/blackjax.png -------------------------------------------------------------------------------- /docs/_static/custom.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/docs/_static/custom.css -------------------------------------------------------------------------------- /docs/bib.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/docs/bib.rst -------------------------------------------------------------------------------- /docs/conf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/docs/conf.py -------------------------------------------------------------------------------- /docs/examples/data/blackjax.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/docs/examples/data/blackjax.png -------------------------------------------------------------------------------- /docs/examples/data/google.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/docs/examples/data/google.csv -------------------------------------------------------------------------------- /docs/examples/howto_custom_gradients.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/docs/examples/howto_custom_gradients.md -------------------------------------------------------------------------------- /docs/examples/howto_metropolis_within_gibbs.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/docs/examples/howto_metropolis_within_gibbs.md -------------------------------------------------------------------------------- /docs/examples/howto_other_frameworks.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/docs/examples/howto_other_frameworks.md -------------------------------------------------------------------------------- /docs/examples/howto_reproduce_the_blackjax_image.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/docs/examples/howto_reproduce_the_blackjax_image.md -------------------------------------------------------------------------------- /docs/examples/howto_sample_multiple_chains.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/docs/examples/howto_sample_multiple_chains.md -------------------------------------------------------------------------------- /docs/examples/howto_use_aesara.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/docs/examples/howto_use_aesara.md -------------------------------------------------------------------------------- /docs/examples/howto_use_numpyro.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/docs/examples/howto_use_numpyro.md -------------------------------------------------------------------------------- /docs/examples/howto_use_oryx.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/docs/examples/howto_use_oryx.md -------------------------------------------------------------------------------- /docs/examples/howto_use_pymc.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/docs/examples/howto_use_pymc.md -------------------------------------------------------------------------------- /docs/examples/howto_use_tfp.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/docs/examples/howto_use_tfp.md -------------------------------------------------------------------------------- /docs/examples/quickstart.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/docs/examples/quickstart.md -------------------------------------------------------------------------------- /docs/examples/scatter.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/docs/examples/scatter.gif -------------------------------------------------------------------------------- /docs/index.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/docs/index.md -------------------------------------------------------------------------------- /docs/refs.bib: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/docs/refs.bib -------------------------------------------------------------------------------- /jupytex.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/jupytex.toml -------------------------------------------------------------------------------- /mypy.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/mypy.ini -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/pyproject.toml -------------------------------------------------------------------------------- /pytest.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/pytest.ini -------------------------------------------------------------------------------- /requirements-doc.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/requirements-doc.txt -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/requirements.txt -------------------------------------------------------------------------------- /setup.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/setup.cfg -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/adaptation/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/adaptation/test_adaptation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/tests/adaptation/test_adaptation.py -------------------------------------------------------------------------------- /tests/adaptation/test_mass_matrix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/tests/adaptation/test_mass_matrix.py -------------------------------------------------------------------------------- /tests/adaptation/test_step_size.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/tests/adaptation/test_step_size.py -------------------------------------------------------------------------------- /tests/mcmc/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/mcmc/test_barker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/tests/mcmc/test_barker.py -------------------------------------------------------------------------------- /tests/mcmc/test_integrators.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/tests/mcmc/test_integrators.py -------------------------------------------------------------------------------- /tests/mcmc/test_latent_gaussian.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/tests/mcmc/test_latent_gaussian.py -------------------------------------------------------------------------------- /tests/mcmc/test_metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/tests/mcmc/test_metrics.py -------------------------------------------------------------------------------- /tests/mcmc/test_proposal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/tests/mcmc/test_proposal.py -------------------------------------------------------------------------------- /tests/mcmc/test_random_walk_without_chex.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/tests/mcmc/test_random_walk_without_chex.py -------------------------------------------------------------------------------- /tests/mcmc/test_sampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/tests/mcmc/test_sampling.py -------------------------------------------------------------------------------- /tests/mcmc/test_trajectory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/tests/mcmc/test_trajectory.py -------------------------------------------------------------------------------- /tests/mcmc/test_uturn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/tests/mcmc/test_uturn.py -------------------------------------------------------------------------------- /tests/optimizers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/optimizers/test_optimizers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/tests/optimizers/test_optimizers.py -------------------------------------------------------------------------------- /tests/optimizers/test_pathfinder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/tests/optimizers/test_pathfinder.py -------------------------------------------------------------------------------- /tests/smc/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/tests/smc/__init__.py -------------------------------------------------------------------------------- /tests/smc/test_inner_kernel_tuning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/tests/smc/test_inner_kernel_tuning.py -------------------------------------------------------------------------------- /tests/smc/test_kernel_compatibility.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/tests/smc/test_kernel_compatibility.py -------------------------------------------------------------------------------- /tests/smc/test_partial_posteriors_smc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/tests/smc/test_partial_posteriors_smc.py -------------------------------------------------------------------------------- /tests/smc/test_persistent_sampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/tests/smc/test_persistent_sampling.py -------------------------------------------------------------------------------- /tests/smc/test_pretuning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/tests/smc/test_pretuning.py -------------------------------------------------------------------------------- /tests/smc/test_resampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/tests/smc/test_resampling.py -------------------------------------------------------------------------------- /tests/smc/test_smc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/tests/smc/test_smc.py -------------------------------------------------------------------------------- /tests/smc/test_smc_ess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/tests/smc/test_smc_ess.py -------------------------------------------------------------------------------- /tests/smc/test_solver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/tests/smc/test_solver.py -------------------------------------------------------------------------------- /tests/smc/test_tempered_smc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/tests/smc/test_tempered_smc.py -------------------------------------------------------------------------------- /tests/smc/test_waste_free_smc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/tests/smc/test_waste_free_smc.py -------------------------------------------------------------------------------- /tests/test_benchmarks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/tests/test_benchmarks.py -------------------------------------------------------------------------------- /tests/test_compilation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/tests/test_compilation.py -------------------------------------------------------------------------------- /tests/test_diagnostics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/tests/test_diagnostics.py -------------------------------------------------------------------------------- /tests/test_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/tests/test_util.py -------------------------------------------------------------------------------- /tests/vi/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/vi/test_meanfield_vi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/tests/vi/test_meanfield_vi.py -------------------------------------------------------------------------------- /tests/vi/test_schrodinger_follmer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/tests/vi/test_schrodinger_follmer.py -------------------------------------------------------------------------------- /tests/vi/test_svgd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blackjax-devs/blackjax/HEAD/tests/vi/test_svgd.py --------------------------------------------------------------------------------