diff --git a/.doctrees/environment.pickle b/.doctrees/environment.pickle index 2b0ae50f1..ed87d5b05 100644 Binary files a/.doctrees/environment.pickle and b/.doctrees/environment.pickle differ diff --git a/.doctrees/examples/howto_metropolis_within_gibbs.doctree b/.doctrees/examples/howto_metropolis_within_gibbs.doctree index 10ca92200..a3ce54ab6 100644 Binary files a/.doctrees/examples/howto_metropolis_within_gibbs.doctree and b/.doctrees/examples/howto_metropolis_within_gibbs.doctree differ diff --git a/.doctrees/examples/howto_other_frameworks.doctree b/.doctrees/examples/howto_other_frameworks.doctree index 187fc4096..f8cec04a6 100644 Binary files a/.doctrees/examples/howto_other_frameworks.doctree and b/.doctrees/examples/howto_other_frameworks.doctree differ diff --git a/.doctrees/examples/howto_sample_multiple_chains.doctree b/.doctrees/examples/howto_sample_multiple_chains.doctree index c1790fc5b..df6d96a96 100644 Binary files a/.doctrees/examples/howto_sample_multiple_chains.doctree and b/.doctrees/examples/howto_sample_multiple_chains.doctree differ diff --git a/.doctrees/examples/howto_use_aesara.doctree b/.doctrees/examples/howto_use_aesara.doctree index d72739b15..d89b12bd7 100644 Binary files a/.doctrees/examples/howto_use_aesara.doctree and b/.doctrees/examples/howto_use_aesara.doctree differ diff --git a/.doctrees/examples/howto_use_oryx.doctree b/.doctrees/examples/howto_use_oryx.doctree index b745a9a72..79e0edd49 100644 Binary files a/.doctrees/examples/howto_use_oryx.doctree and b/.doctrees/examples/howto_use_oryx.doctree differ diff --git a/.doctrees/examples/quickstart.doctree b/.doctrees/examples/quickstart.doctree index 7b1a31f5b..6fd55130d 100644 Binary files a/.doctrees/examples/quickstart.doctree and b/.doctrees/examples/quickstart.doctree differ diff --git a/_images/1370bb7c831a7b717da609d4cf1a486af32b2f4f517b176063b5d012e51c7387.png b/_images/1370bb7c831a7b717da609d4cf1a486af32b2f4f517b176063b5d012e51c7387.png deleted file mode 100644 index ba9d77208..000000000 Binary files a/_images/1370bb7c831a7b717da609d4cf1a486af32b2f4f517b176063b5d012e51c7387.png and /dev/null differ diff --git a/_images/357ed294e00bdf4b386f6e6e7052fe2dcea423567ecc546d7ab7ac0f367ecc2a.png b/_images/357ed294e00bdf4b386f6e6e7052fe2dcea423567ecc546d7ab7ac0f367ecc2a.png deleted file mode 100644 index aec55781c..000000000 Binary files a/_images/357ed294e00bdf4b386f6e6e7052fe2dcea423567ecc546d7ab7ac0f367ecc2a.png and /dev/null differ diff --git a/_images/7bf7fa3e84929059e0ba162f998b3e83bf6127f181c8d8408670ae99f30cc8de.png b/_images/7bf7fa3e84929059e0ba162f998b3e83bf6127f181c8d8408670ae99f30cc8de.png new file mode 100644 index 000000000..7ba81a60e Binary files /dev/null and b/_images/7bf7fa3e84929059e0ba162f998b3e83bf6127f181c8d8408670ae99f30cc8de.png differ diff --git a/_images/b56f390c1db629384beb2f9b5f84722452c284da88a463964eb6e7f34f41fe26.png b/_images/b56f390c1db629384beb2f9b5f84722452c284da88a463964eb6e7f34f41fe26.png new file mode 100644 index 000000000..2892e6519 Binary files /dev/null and b/_images/b56f390c1db629384beb2f9b5f84722452c284da88a463964eb6e7f34f41fe26.png differ diff --git a/_images/c9a0b4ad22e9723ee8751042c18d1353932aa3f59306e863d7efabbed571b209.png b/_images/c9a0b4ad22e9723ee8751042c18d1353932aa3f59306e863d7efabbed571b209.png new file mode 100644 index 000000000..509129e62 Binary files /dev/null and b/_images/c9a0b4ad22e9723ee8751042c18d1353932aa3f59306e863d7efabbed571b209.png differ diff --git a/_images/ef37183100e2f41c47ab75d7e7d0fd2ec8a34b1b411a4aaa5af9f9bd2d85461f.png b/_images/ef37183100e2f41c47ab75d7e7d0fd2ec8a34b1b411a4aaa5af9f9bd2d85461f.png deleted file mode 100644 index 3845919e0..000000000 Binary files a/_images/ef37183100e2f41c47ab75d7e7d0fd2ec8a34b1b411a4aaa5af9f9bd2d85461f.png and /dev/null differ diff --git a/_modules/blackjax/_version.html b/_modules/blackjax/_version.html index 0e3a66419..7a290ac53 100644 --- a/_modules/blackjax/_version.html +++ b/_modules/blackjax/_version.html @@ -353,8 +353,8 @@

Source code for blackjax._version

 __version_tuple__: VERSION_TUPLE
 
[docs]version_tuple: VERSION_TUPLE

 
-__version__ = version = '0.1.dev1+g8c2232c'
-__version_tuple__ = version_tuple = (0, 1, 'dev1', 'g8c2232c')
+__version__ = version = '0.1.dev1+g9713452'
+__version_tuple__ = version_tuple = (0, 1, 'dev1', 'g9713452')
diff --git a/examples/howto_metropolis_within_gibbs.html b/examples/howto_metropolis_within_gibbs.html index a557f26de..5a6b0a45e 100644 --- a/examples/howto_metropolis_within_gibbs.html +++ b/examples/howto_metropolis_within_gibbs.html @@ -579,8 +579,8 @@

Sampling -
CPU times: user 1.13 s, sys: 5.25 ms, total: 1.14 s
-Wall time: 1.13 s
+
CPU times: user 1.14 s, sys: 15 ms, total: 1.16 s
+Wall time: 1.15 s
@@ -718,8 +718,8 @@

Sampling

-
CPU times: user 799 ms, sys: 15.3 ms, total: 814 ms
-Wall time: 804 ms
+
CPU times: user 805 ms, sys: 14.4 ms, total: 819 ms
+Wall time: 810 ms
diff --git a/examples/howto_other_frameworks.html b/examples/howto_other_frameworks.html index dee08d304..c4554ee27 100644 --- a/examples/howto_other_frameworks.html +++ b/examples/howto_other_frameworks.html @@ -563,8 +563,8 @@

Use jax.experim

-
CPU times: user 2.76 s, sys: 4.04 ms, total: 2.76 s
-Wall time: 2.76 s
+
CPU times: user 2.73 s, sys: 0 ns, total: 2.73 s
+Wall time: 2.73 s
@@ -579,8 +579,8 @@

Use jax.experim

-
CPU times: user 10.2 s, sys: 128 ms, total: 10.4 s
-Wall time: 10.3 s
+
CPU times: user 10.5 s, sys: 104 ms, total: 10.6 s
+Wall time: 10.6 s
diff --git a/examples/howto_sample_multiple_chains.html b/examples/howto_sample_multiple_chains.html index 33400cee3..179b48e92 100644 --- a/examples/howto_sample_multiple_chains.html +++ b/examples/howto_sample_multiple_chains.html @@ -499,8 +499,8 @@

Using jax.vmap<

-
CPU times: user 15.9 s, sys: 27.3 ms, total: 15.9 s
-Wall time: 15.9 s
+
CPU times: user 16.1 s, sys: 24.9 ms, total: 16.1 s
+Wall time: 16.1 s
@@ -604,8 +604,8 @@

Back to our example -
CPU times: user 18.4 s, sys: 15.8 ms, total: 18.4 s
-Wall time: 5.97 s
+
CPU times: user 18.4 s, sys: 23.8 ms, total: 18.4 s
+Wall time: 5.96 s
diff --git a/examples/howto_use_aesara.html b/examples/howto_use_aesara.html index fab307965..5953d9b8b 100644 --- a/examples/howto_use_aesara.html +++ b/examples/howto_use_aesara.html @@ -630,7 +630,7 @@

Use with Aesara models

-
<ipykernel.iostream.OutStream at 0x7f58b96dc2e0>
+
<ipykernel.iostream.OutStream at 0x7f83d5f592e0>
diff --git a/examples/howto_use_oryx.html b/examples/howto_use_oryx.html index 08a93d0da..9d7be65d0 100644 --- a/examples/howto_use_oryx.html +++ b/examples/howto_use_oryx.html @@ -521,7 +521,7 @@

Use with Oryx models

-
CPU times: user 20.2 s, sys: 142 ms, total: 20.3 s
+
CPU times: user 20.1 s, sys: 198 ms, total: 20.3 s
 Wall time: 20.2 s
 

 

@@ -559,8 +559,8 @@ 
Use with Oryx models
-
CPU times: user 18.7 s, sys: 52.4 ms, total: 18.8 s
-Wall time: 18.7 s
+
CPU times: user 18.8 s, sys: 76.9 ms, total: 18.9 s
+Wall time: 18.8 s
diff --git a/examples/quickstart.html b/examples/quickstart.html index 95f406231..819a9590f 100644 --- a/examples/quickstart.html +++ b/examples/quickstart.html @@ -442,7 +442,7 @@

Set the Initial State

-
HMCState(position={'loc': 1.0, 'log_scale': 1.0}, logdensity=Array(-34004.82, dtype=float32), logdensity_grad={'loc': Array(1183.2325, dtype=float32, weak_type=True), 'log_scale': Array(63171.773, dtype=float32, weak_type=True)})
+
HMCState(position={'loc': 1.0, 'log_scale': 1.0}, logdensity=Array(-31933.953, dtype=float32), logdensity_grad={'loc': Array(1265.6089, dtype=float32, weak_type=True), 'log_scale': Array(59030.027, dtype=float32, weak_type=True)})
@@ -490,8 +490,8 @@

Inference -
CPU times: user 1.53 s, sys: 13.9 ms, total: 1.54 s
-Wall time: 1.54 s
+
CPU times: user 1.54 s, sys: 449 µs, total: 1.54 s
+Wall time: 1.53 s
@@ -516,7 +516,7 @@

Inference -../_images/1370bb7c831a7b717da609d4cf1a486af32b2f4f517b176063b5d012e51c7387.png +../_images/7bf7fa3e84929059e0ba162f998b3e83bf6127f181c8d8408670ae99f30cc8de.png

@@ -543,7 +543,7 @@

NUTS#

-
HMCState(position={'loc': 1.0, 'log_scale': 1.0}, logdensity=Array(-34004.82, dtype=float32), logdensity_grad={'loc': Array(1183.2325, dtype=float32, weak_type=True), 'log_scale': Array(63171.773, dtype=float32, weak_type=True)})
+
HMCState(position={'loc': 1.0, 'log_scale': 1.0}, logdensity=Array(-31933.953, dtype=float32), logdensity_grad={'loc': Array(1265.6089, dtype=float32, weak_type=True), 'log_scale': Array(59030.027, dtype=float32, weak_type=True)})
@@ -560,8 +560,8 @@

NUTS#

-
CPU times: user 6.52 s, sys: 6.08 ms, total: 6.52 s
-Wall time: 6.51 s
+
CPU times: user 6.49 s, sys: 20.7 ms, total: 6.51 s
+Wall time: 6.5 s
@@ -586,7 +586,7 @@

NUTS#
-../_images/ef37183100e2f41c47ab75d7e7d0fd2ec8a34b1b411a4aaa5af9f9bd2d85461f.png +../_images/c9a0b4ad22e9723ee8751042c18d1353932aa3f59306e863d7efabbed571b209.png

@@ -604,8 +604,8 @@

Use Stan’s Window Adaptation -
CPU times: user 1.69 s, sys: 15.5 ms, total: 1.7 s
-Wall time: 1.68 s
+
CPU times: user 1.67 s, sys: 43.3 ms, total: 1.72 s
+Wall time: 1.7 s
@@ -624,8 +624,8 @@

Use Stan’s Window Adaptation -
CPU times: user 1.29 s, sys: 15.6 ms, total: 1.31 s
-Wall time: 1.29 s
+
CPU times: user 1.28 s, sys: 9.15 ms, total: 1.29 s
+Wall time: 1.28 s
@@ -650,7 +650,7 @@

Use Stan’s Window Adaptation -../_images/357ed294e00bdf4b386f6e6e7052fe2dcea423567ecc546d7ab7ac0f367ecc2a.png +../_images/b56f390c1db629384beb2f9b5f84722452c284da88a463964eb6e7f34f41fe26.png

diff --git a/objects.inv b/objects.inv index 6343d025d..c7927f1a0 100644 Binary files a/objects.inv and b/objects.inv differ diff --git a/searchindex.js b/searchindex.js index a1f1b0799..54b2553e9 100644 --- a/searchindex.js +++ b/searchindex.js @@ -1 +1 @@ -Search.setIndex({"docnames": ["autoapi/blackjax/_version/index", "autoapi/blackjax/adaptation/base/index", "autoapi/blackjax/adaptation/chees_adaptation/index", "autoapi/blackjax/adaptation/index", "autoapi/blackjax/adaptation/mass_matrix/index", "autoapi/blackjax/adaptation/meads_adaptation/index", "autoapi/blackjax/adaptation/pathfinder_adaptation/index", "autoapi/blackjax/adaptation/step_size/index", "autoapi/blackjax/adaptation/window_adaptation/index", "autoapi/blackjax/base/index", "autoapi/blackjax/diagnostics/index", "autoapi/blackjax/index", "autoapi/blackjax/mcmc/diffusions/index", "autoapi/blackjax/mcmc/elliptical_slice/index", "autoapi/blackjax/mcmc/ghmc/index", "autoapi/blackjax/mcmc/hmc/index", "autoapi/blackjax/mcmc/index", "autoapi/blackjax/mcmc/integrators/index", "autoapi/blackjax/mcmc/mala/index", "autoapi/blackjax/mcmc/marginal_latent_gaussian/index", "autoapi/blackjax/mcmc/metrics/index", "autoapi/blackjax/mcmc/nuts/index", "autoapi/blackjax/mcmc/periodic_orbital/index", "autoapi/blackjax/mcmc/proposal/index", "autoapi/blackjax/mcmc/random_walk/index", "autoapi/blackjax/mcmc/termination/index", "autoapi/blackjax/mcmc/trajectory/index", "autoapi/blackjax/optimizers/dual_averaging/index", "autoapi/blackjax/optimizers/index", "autoapi/blackjax/optimizers/lbfgs/index", "autoapi/blackjax/progress_bar/index", "autoapi/blackjax/sgmcmc/csgld/index", "autoapi/blackjax/sgmcmc/diffusions/index", "autoapi/blackjax/sgmcmc/gradients/index", "autoapi/blackjax/sgmcmc/index", "autoapi/blackjax/sgmcmc/sghmc/index", "autoapi/blackjax/sgmcmc/sgld/index", "autoapi/blackjax/sgmcmc/sgnht/index", "autoapi/blackjax/smc/adaptive_tempered/index", "autoapi/blackjax/smc/base/index", "autoapi/blackjax/smc/ess/index", "autoapi/blackjax/smc/index", "autoapi/blackjax/smc/resampling/index", "autoapi/blackjax/smc/solver/index", "autoapi/blackjax/smc/tempered/index", "autoapi/blackjax/types/index", "autoapi/blackjax/util/index", "autoapi/blackjax/vi/index", "autoapi/blackjax/vi/meanfield_vi/index", "autoapi/blackjax/vi/pathfinder/index", "autoapi/blackjax/vi/svgd/index", "bib", "examples/howto_custom_gradients", "examples/howto_metropolis_within_gibbs", "examples/howto_other_frameworks", "examples/howto_sample_multiple_chains", "examples/howto_use_aesara", "examples/howto_use_numpyro", "examples/howto_use_oryx", "examples/howto_use_pymc", "examples/howto_use_tfp", "examples/quickstart", "index"], "filenames": ["autoapi/blackjax/_version/index.rst", "autoapi/blackjax/adaptation/base/index.rst", "autoapi/blackjax/adaptation/chees_adaptation/index.rst", "autoapi/blackjax/adaptation/index.rst", "autoapi/blackjax/adaptation/mass_matrix/index.rst", "autoapi/blackjax/adaptation/meads_adaptation/index.rst", "autoapi/blackjax/adaptation/pathfinder_adaptation/index.rst", "autoapi/blackjax/adaptation/step_size/index.rst", "autoapi/blackjax/adaptation/window_adaptation/index.rst", "autoapi/blackjax/base/index.rst", "autoapi/blackjax/diagnostics/index.rst", "autoapi/blackjax/index.rst", "autoapi/blackjax/mcmc/diffusions/index.rst", "autoapi/blackjax/mcmc/elliptical_slice/index.rst", "autoapi/blackjax/mcmc/ghmc/index.rst", "autoapi/blackjax/mcmc/hmc/index.rst", "autoapi/blackjax/mcmc/index.rst", "autoapi/blackjax/mcmc/integrators/index.rst", "autoapi/blackjax/mcmc/mala/index.rst", "autoapi/blackjax/mcmc/marginal_latent_gaussian/index.rst", "autoapi/blackjax/mcmc/metrics/index.rst", "autoapi/blackjax/mcmc/nuts/index.rst", "autoapi/blackjax/mcmc/periodic_orbital/index.rst", "autoapi/blackjax/mcmc/proposal/index.rst", "autoapi/blackjax/mcmc/random_walk/index.rst", "autoapi/blackjax/mcmc/termination/index.rst", "autoapi/blackjax/mcmc/trajectory/index.rst", "autoapi/blackjax/optimizers/dual_averaging/index.rst", "autoapi/blackjax/optimizers/index.rst", "autoapi/blackjax/optimizers/lbfgs/index.rst", "autoapi/blackjax/progress_bar/index.rst", "autoapi/blackjax/sgmcmc/csgld/index.rst", "autoapi/blackjax/sgmcmc/diffusions/index.rst", "autoapi/blackjax/sgmcmc/gradients/index.rst", "autoapi/blackjax/sgmcmc/index.rst", "autoapi/blackjax/sgmcmc/sghmc/index.rst", "autoapi/blackjax/sgmcmc/sgld/index.rst", "autoapi/blackjax/sgmcmc/sgnht/index.rst", "autoapi/blackjax/smc/adaptive_tempered/index.rst", "autoapi/blackjax/smc/base/index.rst", "autoapi/blackjax/smc/ess/index.rst", "autoapi/blackjax/smc/index.rst", "autoapi/blackjax/smc/resampling/index.rst", "autoapi/blackjax/smc/solver/index.rst", "autoapi/blackjax/smc/tempered/index.rst", "autoapi/blackjax/types/index.rst", "autoapi/blackjax/util/index.rst", "autoapi/blackjax/vi/index.rst", "autoapi/blackjax/vi/meanfield_vi/index.rst", "autoapi/blackjax/vi/pathfinder/index.rst", "autoapi/blackjax/vi/svgd/index.rst", "bib.rst", "examples/howto_custom_gradients.md", "examples/howto_metropolis_within_gibbs.md", "examples/howto_other_frameworks.md", "examples/howto_sample_multiple_chains.md", "examples/howto_use_aesara.md", "examples/howto_use_numpyro.md", "examples/howto_use_oryx.md", "examples/howto_use_pymc.md", "examples/howto_use_tfp.md", "examples/quickstart.md", "index.md"], "titles": ["blackjax._version", "blackjax.adaptation.base", "blackjax.adaptation.chees_adaptation", "blackjax.adaptation", "blackjax.adaptation.mass_matrix", "blackjax.adaptation.meads_adaptation", "blackjax.adaptation.pathfinder_adaptation", "blackjax.adaptation.step_size", "blackjax.adaptation.window_adaptation", "blackjax.base", "blackjax.diagnostics", "blackjax", "blackjax.mcmc.diffusions", "blackjax.mcmc.elliptical_slice", "blackjax.mcmc.ghmc", "blackjax.mcmc.hmc", "blackjax.mcmc", "blackjax.mcmc.integrators", "blackjax.mcmc.mala", "blackjax.mcmc.marginal_latent_gaussian", "blackjax.mcmc.metrics", "blackjax.mcmc.nuts", "blackjax.mcmc.periodic_orbital", "blackjax.mcmc.proposal", "blackjax.mcmc.random_walk", "blackjax.mcmc.termination", "blackjax.mcmc.trajectory", "blackjax.optimizers.dual_averaging", "blackjax.optimizers", "blackjax.optimizers.lbfgs", "blackjax.progress_bar", "blackjax.sgmcmc.csgld", "blackjax.sgmcmc.diffusions", "blackjax.sgmcmc.gradients", "blackjax.sgmcmc", "blackjax.sgmcmc.sghmc", "blackjax.sgmcmc.sgld", "blackjax.sgmcmc.sgnht", "blackjax.smc.adaptive_tempered", "blackjax.smc.base", "blackjax.smc.ess", "blackjax.smc", "blackjax.smc.resampling", "blackjax.smc.solver", "blackjax.smc.tempered", "blackjax.types", "blackjax.util", "blackjax.vi", "blackjax.vi.meanfield_vi", "blackjax.vi.pathfinder", "blackjax.vi.svgd", "Bibliography", "Use custom gradients", "How to build a Metropolis-Within-Gibbs sampler?", "Use a logdensity function that is not compatible with JAX\u2019s primitives", "Sample with multiple chains in parallel", "Use with Aesara models", "Use with Numpyro models", "Use with Oryx models", "Use with PyMC models", "Use with TFP models", "A Quick Introduction to Blackjax", "Welcome to Blackjax!"], "terms": {"type_check": 0, "fals": [0, 8, 19, 26, 57, 60], "sourc": [0, 1, 2, 4, 5, 6, 7, 8, 9, 10, 12, 13, 14, 15, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 29, 30, 31, 32, 33, 35, 36, 37, 38, 39, 40, 42, 43, 44, 45, 46, 48, 49, 50, 51], "version_tupl": 0, "version": [0, 14, 19, 20, 21, 24, 57, 59, 60, 62], "str": 0, "adaptationresult": [1, 8], "state": [1, 2, 4, 5, 6, 7, 8, 9, 13, 14, 15, 18, 19, 21, 22, 23, 24, 26, 27, 31, 35, 36, 37, 38, 39, 40, 44, 48, 49, 50, 52, 54, 55, 56, 57, 58, 59, 60, 62], "type": [1, 2, 4, 5, 6, 7, 8, 10, 13, 14, 15, 18, 19, 20, 21, 22, 23, 24, 25, 26, 29, 31, 33, 35, 36, 37, 38, 39, 40, 42, 43, 44, 46, 48, 49, 50, 53, 56], "arraytre": [1, 5, 13, 14, 15, 18, 21, 22, 24, 26, 31, 37, 39, 44, 45, 46, 48, 49], "paramet": [1, 2, 4, 5, 6, 7, 8, 9, 10, 13, 14, 15, 17, 18, 19, 20, 21, 22, 23, 26, 27, 29, 31, 32, 33, 35, 36, 37, 38, 39, 40, 43, 44, 46, 48, 49, 50, 51, 55, 56, 57, 58, 59, 60], "dict": [1, 22, 53, 58], "adaptationinfo": 1, "namedtupl": [1, 26, 39, 53, 60], "info": [1, 2, 9, 13, 14, 15, 18, 19, 21, 22, 24, 35, 36, 39, 52, 54, 56, 57, 58, 59, 60], "adaptation_st": 1, "public": [2, 13, 14, 15, 18, 19, 21, 22, 31, 35, 36, 37], "api": [2, 13, 14, 15, 18, 19, 21, 22, 31, 35, 36, 37, 54, 59], "chee": [2, 55], "hmc": [2, 5, 6, 7, 8, 11, 14, 16, 26, 35, 46, 51, 52, 53, 55, 60], "optimal_target_acceptance_r": 2, "0": [2, 5, 6, 7, 8, 10, 14, 22, 23, 27, 31, 32, 35, 37, 39, 43, 46, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62], "651": 2, "cheesadaptationst": 2, "scheme": [2, 5, 6, 7, 8, 27, 51], "step_siz": [2, 3, 5, 6, 8, 11, 14, 15, 17, 18, 21, 22, 26, 35, 36, 37, 53, 54, 55, 61, 62], "valu": [2, 4, 5, 6, 7, 8, 10, 13, 14, 15, 17, 18, 19, 20, 21, 22, 23, 24, 26, 27, 29, 31, 33, 40, 44, 51, 52, 53, 56, 58, 60, 61], "algorithm": [2, 4, 5, 6, 7, 8, 9, 13, 14, 15, 18, 21, 22, 24, 26, 27, 29, 31, 32, 33, 35, 36, 37, 39, 40, 44, 46, 49, 50, 51, 55, 56, 58, 61, 62], "log_step_size_moving_averag": 2, "run": [2, 4, 6, 8, 9, 14, 15, 26, 29, 31, 54, 55, 56, 57, 58, 59, 60, 61, 62], "move": [2, 5, 6, 8, 24, 55], "averag": [2, 7, 8, 21, 27, 57, 58, 60], "log": [2, 5, 6, 7, 8, 13, 14, 15, 18, 19, 21, 22, 24, 27, 31, 33, 35, 36, 37, 38, 39, 40, 44, 48, 49, 50, 52, 53, 54, 56, 57, 58, 59, 60, 61, 62], "trajectory_length": 2, "num_integration_step": [2, 15, 21, 35, 53, 57, 60, 61], "log_trajectory_length_moving_averag": 2, "optim_st": 2, "optax": [2, 48, 50], "optim": [2, 6, 31, 48, 49, 50, 51, 52, 53, 54], "us": [2, 4, 5, 6, 7, 8, 9, 10, 13, 14, 15, 17, 18, 19, 20, 21, 22, 23, 24, 26, 27, 29, 30, 38, 39, 43, 44, 46, 48, 49, 51, 53, 62], "maxim": [2, 5], "criterion": [2, 10, 25, 26, 52], "random_generator_arg": [2, 15], "util": [2, 15, 38, 57], "arrai": [2, 4, 5, 6, 8, 10, 13, 15, 19, 20, 22, 23, 24, 25, 29, 31, 39, 40, 42, 43, 44, 45, 46, 49, 52, 53, 54, 55, 56, 57, 59, 60, 61, 62], "gener": [2, 5, 7, 8, 14, 15, 18, 20, 21, 23, 24, 26, 35, 36, 37, 38, 39, 44, 46, 48, 51, 55, 56, 57, 58, 61], "pseudo": [2, 15, 39, 46, 48], "quasi": [2, 15, 29, 49, 51], "random": [2, 7, 9, 14, 15, 20, 21, 22, 24, 26, 38, 39, 44, 46, 48, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62], "sequenc": [2, 10, 15, 51], "number": [2, 4, 5, 6, 7, 8, 10, 13, 15, 21, 22, 24, 26, 29, 31, 33, 35, 38, 39, 44, 46, 48, 49, 53, 56, 57, 58, 60, 61], "step": [2, 4, 5, 6, 7, 8, 9, 13, 14, 15, 18, 19, 21, 22, 23, 24, 26, 27, 29, 30, 31, 35, 36, 37, 38, 39, 40, 48, 49, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62], "current": [2, 4, 5, 7, 9, 10, 13, 14, 15, 18, 19, 20, 21, 22, 23, 24, 26, 27, 31, 33, 37, 38, 39, 40, 44, 50, 55, 62], "iter": [2, 4, 7, 13, 14, 21, 22, 27, 29, 35, 36, 37, 43, 44, 49, 51, 54, 62], "float": [2, 5, 6, 7, 8, 13, 14, 15, 18, 19, 21, 23, 24, 26, 27, 29, 32, 35, 37, 38, 39, 40, 43, 44, 46, 48], "da_stat": 2, "dual_averag": [2, 28], "dualaveragingst": [2, 27], "optstat": [2, 48], "int": [2, 4, 5, 7, 8, 10, 13, 15, 21, 22, 23, 25, 26, 27, 29, 31, 33, 39, 42, 43, 46, 48, 49, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61], "base": [2, 3, 4, 5, 6, 8, 11, 15, 18, 19, 21, 23, 29, 38, 41, 44, 51, 52, 56, 59, 61], "jitter_gener": 2, "callabl": [2, 4, 5, 6, 7, 8, 9, 13, 14, 15, 18, 20, 21, 22, 23, 24, 26, 27, 29, 31, 33, 35, 36, 37, 38, 39, 40, 43, 44, 48, 49], "next_random_arg_fn": [2, 15], "gradienttransform": [2, 48, 50], "target_acceptance_r": [2, 6, 8, 57], "decay_r": 2, "tupl": [2, 4, 7, 8, 20, 22, 23, 26, 27, 29, 33, 39, 48, 49, 56], "chang": [2, 8], "estim": [2, 4, 5, 6, 7, 8, 10, 31, 33, 35, 36, 37, 48, 49, 50, 51, 58], "expect": [2, 55], "squar": [2, 4, 5, 46, 50], "trajectori": [2, 8, 11, 14, 15, 16, 17, 20, 21, 23, 51, 61], "length": [2, 14, 17, 21, 51], "dual": [2, 7, 8, 27, 51], "procedur": [2, 6, 7, 8, 9, 14, 26, 27], "size": [2, 5, 6, 7, 8, 10, 14, 15, 18, 19, 21, 22, 26, 27, 29, 35, 36, 37, 38, 39, 40, 49, 51, 55, 56, 58, 60, 61, 62], "jitter": 2, "hamiltonian": [2, 4, 5, 15, 17, 20, 22, 26, 35, 51, 61], "mont": [2, 4, 5, 35, 51, 61], "carlo": [2, 4, 5, 35, 51, 61], "kernel": [2, 5, 6, 7, 8, 9, 13, 14, 15, 18, 19, 21, 22, 24, 31, 35, 36, 37, 38, 39, 44, 49, 50, 55, 56, 57, 58, 59, 60, 62], "hrs21": [2, 51], "thi": [2, 4, 5, 6, 7, 8, 9, 10, 13, 14, 15, 18, 19, 21, 23, 24, 26, 27, 31, 32, 33, 35, 36, 37, 38, 39, 46, 50, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61], "tune": [2, 4, 5, 6, 7, 8, 51, 57, 59, 60], "i": [2, 4, 5, 6, 7, 8, 9, 10, 13, 14, 15, 17, 19, 20, 21, 22, 23, 24, 26, 27, 29, 31, 38, 39, 40, 43, 44, 46, 48, 49, 50, 51, 53, 55, 56, 57, 58, 59, 60, 61, 62], "e": [2, 6, 7, 8, 23, 31, 52, 53, 55, 56], "integr": [2, 5, 6, 8, 11, 14, 15, 16, 18, 21, 22, 26, 35, 51, 53, 55, 60, 61, 62], "statist": [2, 5, 10, 51], "collect": [2, 5, 22], "from": [2, 4, 5, 6, 7, 8, 9, 10, 13, 14, 15, 18, 21, 22, 23, 24, 26, 29, 30, 32, 33, 38, 39, 40, 44, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61], "popul": [2, 4, 5], "mani": [2, 5, 14, 52, 55], "chain": [2, 5, 6, 7, 9, 10, 13, 14, 15, 18, 19, 20, 22, 23, 24, 38, 44, 51, 60], "It": [2, 4, 5, 8, 19, 20, 23, 29, 46, 48, 55, 61, 62], "target": [2, 5, 6, 7, 8, 9, 10, 14, 22, 27, 29, 33, 38, 40, 48, 49, 50, 58], "an": [2, 4, 5, 6, 7, 8, 9, 10, 13, 15, 18, 19, 20, 21, 22, 23, 24, 27, 31, 35, 36, 37, 38, 39, 43, 44, 46, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61], "accept": [2, 5, 6, 7, 8, 13, 14, 15, 18, 19, 21, 23, 24, 27, 31, 40, 51, 56, 57, 60], "rate": [2, 6, 7, 8, 14, 18, 19, 26, 27, 57, 60], "harmon": 2, "mean": [2, 4, 13, 14, 19, 24, 46, 48, 53, 55, 57, 58, 60], "": [2, 4, 5, 7, 8, 9, 10, 14, 20, 23, 24, 29, 30, 31, 33, 48, 51, 52, 53, 55, 56, 57, 58, 59, 60, 62], "probabl": [2, 5, 6, 7, 8, 9, 15, 18, 19, 20, 22, 23, 24, 33, 38, 40, 44, 53, 54, 56, 57, 58, 59, 60, 62], "option": [2, 19, 23, 38, 40, 55], "1": [2, 4, 6, 7, 8, 10, 20, 22, 23, 24, 26, 27, 29, 31, 39, 46, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62], "given": [2, 5, 6, 7, 8, 14, 15, 19, 20, 22, 23, 24, 27, 31, 38, 44, 46, 50, 53, 54, 61], "prngkei": [2, 7, 14, 23, 37, 39, 42, 45, 46, 48, 49, 58], "propos": [2, 5, 7, 11, 13, 15, 16, 18, 19, 21, 24, 26, 39], "If": [2, 7, 23, 24, 26, 31, 43, 54, 62], "none": [2, 19, 24, 30, 39, 53, 55, 56, 57, 58, 59, 60], "halton": 2, "next": [2, 4, 9, 14, 15, 23, 35, 36, 37, 40, 55], "its": [2, 5, 14, 15, 22, 24, 27, 46, 54, 56, 58], "previou": [2, 4, 8, 13, 14, 15, 22, 23, 24, 53], "compat": [2, 50, 56, 57, 58, 59, 60, 62], "which": [2, 5, 6, 7, 8, 9, 10, 14, 15, 17, 19, 20, 21, 23, 24, 26, 31, 35, 36, 37, 39, 44, 46, 50, 52, 53, 54, 56, 57], "conform": [2, 50], "protocol": [2, 23, 50], "repres": [2, 9, 10, 22, 39, 48, 49], "how": [2, 46, 51, 52, 54, 55, 56, 57, 58, 59, 60], "much": [2, 55], "favor": [2, 23], "recent": [2, 55], "over": [2, 9, 53, 58], "earlier": [2, 7, 27, 55], "ones": [2, 17, 39, 52, 54, 55, 58, 60], "return": [2, 4, 5, 6, 7, 8, 9, 10, 13, 14, 15, 18, 19, 20, 21, 22, 23, 24, 26, 27, 29, 31, 35, 36, 37, 38, 39, 40, 43, 44, 46, 48, 49, 50, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62], "init": [2, 4, 5, 6, 7, 8, 9, 13, 14, 15, 18, 19, 21, 22, 24, 27, 31, 35, 36, 37, 38, 39, 44, 48, 49, 50, 52, 53, 54, 55, 61, 62], "initi": [2, 4, 5, 6, 7, 8, 9, 10, 13, 14, 15, 18, 19, 21, 22, 23, 24, 26, 27, 29, 35, 36, 37, 48, 55, 56, 57, 58, 59, 62], "warmup": [2, 5, 6, 7, 8, 61], "updat": [2, 4, 5, 6, 7, 8, 9, 14, 23, 26, 27, 29, 38, 39, 50, 51, 53, 59], "one": [2, 5, 6, 8, 10, 14, 15, 18, 22, 23, 24, 26, 35, 36, 37, 39, 52, 54, 55], "logdensity_fn": [2, 5, 6, 8, 13, 14, 15, 18, 19, 21, 22, 24, 40, 48, 49, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62], "num_chain": [2, 5, 55], "jitter_amount": 2, "5": [2, 7, 26, 27, 39, 48, 52, 53, 54, 55, 56, 57, 59, 60, 61], "adaptationalgorithm": [2, 5, 6, 8, 9], "algorthm": 2, "The": [2, 4, 5, 6, 7, 8, 9, 10, 13, 14, 15, 17, 18, 19, 20, 21, 22, 23, 24, 26, 27, 29, 31, 33, 35, 36, 37, 38, 39, 40, 43, 44, 46, 48, 49, 51, 52, 54, 55, 57, 58, 60, 62], "depend": [2, 5, 8, 13, 14, 20, 24, 26, 56, 62], "control": [2, 5, 7, 27, 31, 33, 37, 51], "discret": [2, 22], "onli": [2, 8, 13, 15, 18, 19, 20, 23, 35, 36, 37, 52, 54, 55, 56, 61, 62], "percentag": 2, "heurist": [2, 5, 7, 50], "ensambl": 2, "parallel": [2, 5, 14, 29, 51], "At": [2, 55], "equilibrium": 2, "aim": [2, 38], "elimin": 2, "correl": [2, 5], "between": [2, 4, 5, 7, 8, 9, 13, 14, 15, 19, 21, 23, 24, 26, 27, 31, 48, 50, 54, 55, 56], "dimens": [2, 5, 10, 14, 46], "make": [2, 5, 14, 15, 18, 26, 54, 55, 56, 57, 62], "effici": [2, 13, 14, 15, 18, 19, 31, 52, 53], "requir": [2, 8, 55, 57, 61], "uniform": [2, 23, 51, 56], "variabl": [2, 5, 13, 14, 20, 22, 24, 33, 39, 46, 53, 54, 55, 56, 57, 58], "howev": [2, 4, 53, 56], "add": [2, 15, 18, 51, 56], "anoth": [2, 15, 18, 22, 23, 55], "varianc": [2, 4, 5, 10, 48, 51, 56], "sampl": [2, 4, 5, 6, 7, 8, 9, 10, 13, 14, 15, 18, 19, 21, 22, 23, 24, 26, 29, 31, 35, 36, 37, 38, 39, 40, 44, 46, 48, 49, 50, 51, 54, 56, 57, 58, 59, 60, 61], "mai": [2, 7, 27, 52, 54, 55, 62], "slow": [2, 7, 8, 26, 27, 55], "lead": [2, 56], "suboptim": 2, "mix": [2, 5, 61], "To": [2, 9, 14, 26, 35, 36, 37, 53, 54, 55, 56, 57, 58], "allevi": 2, "rather": [2, 52], "than": [2, 5, 8, 10, 23, 31, 52, 55], "nois": [2, 14, 46], "we": [2, 5, 6, 7, 8, 13, 14, 15, 18, 19, 20, 21, 22, 24, 26, 31, 35, 36, 37, 38, 39, 40, 43, 46, 48, 49, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62], "ensur": [2, 5, 8, 22, 26, 52], "more": [2, 8, 10, 13, 14, 15, 17, 18, 20, 21, 38, 46, 55], "even": 2, "distribut": [2, 13, 20, 22, 24, 29, 33, 44, 46, 48, 49, 51, 53, 54, 55, 56, 57, 58, 60, 61], "exampl": [2, 13, 14, 15, 18, 19, 21, 22, 23, 24, 35, 36, 37, 39, 46, 52, 53, 54, 56, 57, 58, 59, 60, 61], "can": [2, 7, 8, 9, 13, 14, 15, 18, 19, 20, 21, 22, 23, 24, 31, 35, 36, 37, 39, 43, 46, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62], "learn": [2, 33, 51], "follow": [2, 4, 7, 8, 13, 14, 15, 18, 19, 20, 21, 22, 24, 26, 39, 46, 52, 53, 54, 55, 56, 57, 59, 60, 62], "code": [2, 13, 14, 15, 18, 19, 21, 22, 24, 53, 55, 62], "key_warmup": 2, "key_sampl": 2, "jax": [2, 7, 13, 14, 15, 18, 19, 20, 21, 22, 24, 30, 35, 36, 37, 39, 46, 48, 51, 53, 56, 57, 58, 59, 60, 61, 62], "split": [2, 8, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62], "rng_kei": [2, 7, 9, 13, 14, 15, 18, 19, 21, 22, 23, 24, 29, 35, 36, 37, 38, 39, 42, 44, 46, 48, 49, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62], "adam": [2, 51], "learning_r": 2, "last_stat": [2, 57, 58, 59, 60], "_": [2, 7, 10, 14, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62], "posit": [2, 5, 9, 10, 13, 14, 15, 18, 19, 20, 21, 22, 24, 29, 31, 33, 35, 36, 37, 38, 44, 46, 49, 50, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61], "pytre": [2, 5, 13, 14, 15, 18, 19, 20, 22, 24, 29, 39, 44, 46], "where": [2, 4, 7, 8, 10, 15, 22, 27, 29, 31, 44, 52, 53, 54, 55], "each": [2, 5, 8, 10, 14, 15, 20, 22, 23, 35, 36, 37, 38, 39, 46, 48, 55, 60], "leaf": [2, 5], "ha": [2, 7, 8, 21, 23, 26, 27, 33, 48, 53, 54, 55], "shape": [2, 5, 43, 46, 53, 55, 58, 59], "initial_step_s": [2, 6, 7, 8], "num_warmup_step": 2, "dynamic_hmc": [2, 15], "new_stat": [2, 13, 14, 15, 18, 19, 21, 22, 24, 37, 52, 61], "vmap": [2, 39, 56, 58], "densiti": [2, 5, 6, 8, 9, 14, 15, 18, 19, 20, 21, 22, 23, 24, 31, 33, 35, 36, 37, 38, 44, 48, 49, 50, 52, 53, 56, 58, 60, 62], "wish": [2, 5, 6, 8, 13, 14, 15, 18, 21, 22, 24, 38, 40], "cross": [2, 5, 7], "warm": [2, 5], "up": [2, 5, 7], "train": [2, 5, 58], "jitter_valu": 2, "A": [2, 4, 5, 6, 7, 8, 9, 13, 14, 15, 18, 19, 20, 21, 22, 23, 24, 26, 27, 31, 33, 35, 36, 37, 38, 40, 44, 46, 48, 49, 50, 51, 53, 54, 56], "calcul": [2, 5, 22, 29, 53], "should": [2, 8, 15, 18, 20, 21, 26, 29, 31, 46, 52, 53, 54, 55], "jit": [2, 13, 14, 15, 18, 19, 21, 22, 24, 26, 35, 36, 37, 53, 54, 55, 56, 57, 61, 62], "default": [2, 4, 7, 10, 14, 15, 18, 19, 21, 27, 35, 36, 37, 38, 40, 46, 53, 55, 61, 62], "give": [2, 5, 8, 52, 53], "equal": [2, 5, 17, 22, 23, 24, 31, 55], "weight": [2, 7, 22, 23, 27, 38, 39, 40, 42, 44, 51, 54, 58], "all": [2, 5, 9, 22, 24, 40, 42, 43, 46, 49, 53, 55, 56, 57], "histori": [2, 29, 49], "most": [2, 15, 18, 20, 35, 36, 37, 52, 53, 54], "last": [2, 5, 6, 8, 10, 26], "diagnost": [2, 5, 11, 13, 15, 18, 19, 21, 24, 55], "chees_adapt": [3, 11], "mass_matrix": [3, 8, 11], "meads_adapt": [3, 11], "pathfinder_adapt": [3, 11], "window_adapt": [3, 11, 56, 57, 58, 59, 60, 61], "mass": [4, 6, 7, 8, 15, 20, 21, 22, 46, 60, 61], "matrix": [4, 5, 6, 7, 8, 13, 15, 20, 21, 22, 24, 46, 60, 61], "famili": [4, 6, 8, 26], "geometri": 4, "stan": [4, 8, 10, 51], "manual": [4, 15, 18, 35, 36, 37, 51, 53, 54], "stab": [4, 8, 51], "veri": [4, 8, 19, 52, 55], "good": [4, 5, 7, 8, 14, 19, 52, 54, 55, 58, 60], "refer": [4, 7, 24, 26, 51, 53, 60], "automat": [4, 5, 52, 53, 55, 56], "welfordalgorithmst": 4, "carri": [4, 7, 9, 27], "through": [4, 7, 14, 27, 52], "welford": 4, "m2": 4, "sum": [4, 5, 23, 31, 52, 55, 56, 58, 61, 62], "differ": [4, 7, 14, 15, 17, 19, 21, 23, 24, 26, 27, 31, 55, 61, 62], "see": [4, 7, 8, 23, 24, 27, 46, 54, 55, 59, 61], "document": [4, 57, 58, 59, 60, 62], "welford_algorithm": 4, "explan": [4, 8, 27], "sample_s": 4, "success": [4, 8, 31], "have": [4, 7, 9, 10, 14, 19, 21, 35, 36, 37, 43, 52, 54, 55, 57, 61], "been": [4, 21, 23, 26], "comput": [4, 5, 6, 8, 9, 10, 13, 14, 15, 18, 19, 21, 22, 23, 24, 26, 38, 40, 44, 46, 48, 51, 52, 54, 55, 56, 57, 60], "also": [4, 8, 9, 14, 15, 17, 18, 20, 21, 26, 52, 54, 55, 57, 61], "massmatrixadaptationst": [4, 8], "inverse_mass_matrix": [4, 6, 7, 8, 15, 20, 21, 22, 53, 54, 62], "curent": 4, "invers": [4, 6, 7, 8, 15, 20, 21, 22, 29, 46, 49, 60, 61], "wc_state": 4, "mass_matrix_adapt": 4, "is_diagonal_matrix": 4, "bool": [4, 8, 15, 18, 19, 21, 24, 26, 38, 48], "true": [4, 8, 38, 48, 49, 52, 53, 54, 55, 57, 61], "covari": [4, 5, 8, 13, 19, 24, 49], "when": [4, 5, 9, 10, 15, 20, 21, 22, 23, 24, 29, 38, 46, 49, 53, 55], "diagon": [4, 8, 20, 29, 46], "otherwis": [4, 8, 55], "adap": [4, 8], "dens": [4, 8, 20, 46, 58], "final": [4, 6, 8, 27, 55, 56, 57, 58], "onlin": 4, "possibl": [4, 5, 8, 55], "line": [4, 29, 49, 61], "fashion": 4, "avoid": [4, 9, 52, 53, 54], "store": [4, 14, 15, 18, 20, 49], "intermedi": 4, "result": [4, 10, 29, 46, 48, 49, 55, 57], "naiv": 4, "recurr": 4, "relat": [4, 8, 40, 62], "ar": [4, 5, 6, 7, 8, 9, 14, 15, 17, 18, 22, 23, 24, 26, 27, 29, 33, 35, 36, 37, 48, 52, 53, 54, 55, 56, 60, 61, 62], "numer": [4, 46, 51, 52], "stabl": [4, 8, 38, 59], "m_": [4, 39], "2": [4, 6, 7, 8, 10, 20, 24, 26, 29, 39, 40, 46, 52, 53, 54, 56, 57, 58, 61, 62], "n": [4, 10, 19, 39, 46, 54, 56], "sum_": [4, 10], "left": [4, 7, 62], "x_i": 4, "overlin": [4, 7], "x_n": 4, "right": [4, 7, 26, 53], "relationship": 4, "might": [4, 24, 55], "seem": [4, 21], "pedant": 4, "separ": [4, 8, 53], "other": [4, 8, 9, 14, 15, 21, 29, 49, 51, 52, 53, 54, 55], "part": [4, 56], "librari": [4, 8, 14, 15, 21, 51, 54, 55, 56, 57, 58, 59, 60, 61, 62], "meadsadaptationst": 5, "mead": [5, 55], "position_sigma": 5, "contain": [5, 9, 10, 13, 14, 15, 18, 19, 22, 24, 29, 38, 39, 44, 49, 53, 58, 60], "per": [5, 29, 38, 49], "standard": [5, 14, 20, 24, 46], "deviat": [5, 24, 46], "scale": [5, 10, 32, 51, 54, 55, 61, 62], "momentum": [5, 13, 14, 15, 20, 21, 22, 25, 26, 37, 46], "alpha": [5, 14, 29, 32, 35, 37, 49, 56], "delta": [5, 7, 14, 19, 31, 38, 40, 43], "current_iter": 5, "maximum": [5, 21, 26, 29, 43, 49], "eigenvalu": 5, "damp": 5, "hs22": [5, 51], "perform": [5, 8, 13, 14, 19, 22, 23, 24, 26, 35, 36, 37, 46, 49, 53, 54, 55, 57, 59, 60], "select": [5, 39], "tunabl": 5, "determin": [5, 17, 20, 26, 31, 61], "gradient": [5, 11, 14, 15, 18, 19, 22, 29, 31, 34, 35, 36, 37, 38, 48, 49, 50, 51, 54, 56, 61], "matric": 5, "group": 5, "implement": [5, 6, 8, 9, 10, 13, 14, 15, 18, 19, 21, 22, 24, 26, 29, 31, 35, 36, 37, 38, 39, 48, 49, 50, 52, 53, 54, 56, 57, 58, 59, 60, 61, 62], "3": [5, 7, 8, 26, 29, 35, 36, 37, 39, 49, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62], "instead": [5, 14, 26, 55, 58], "ensampl": 5, "three": [5, 8, 17, 39], "element": [5, 29, 46, 53, 56], "behaviour": 5, "dynam": [5, 15, 20, 25, 26, 31, 36, 46, 51, 52, 61], "persist": [5, 14], "determinist": [5, 14], "transform": [5, 14, 22, 55, 56, 57, 58, 61], "slice": [5, 13, 14, 21, 23, 51], "non": [5, 14, 23, 24, 31, 51, 52, 57, 59, 60], "revers": [5, 14, 17, 23, 26, 52], "metropoli": [5, 7, 14, 18, 23, 24, 51], "hast": [5, 14, 23, 24, 51], "reject": [5, 14, 23, 51], "chosen": [5, 7, 15, 27, 52, 55], "stabil": [5, 7, 17, 27, 52], "veloc": [5, 14], "verlet": [5, 14], "influenc": 5, "futur": [5, 46], "decai": 5, "exponenti": 5, "properti": [5, 27, 31], "These": [5, 54], "characterist": 5, "under": 5, "simpifyng": 5, "assumpt": 5, "fundament": 5, "explor": 5, "space": [5, 14, 20, 31, 37, 38, 56, 57, 61], "efficienti": 5, "output": [5, 14, 22, 46, 52, 57, 61], "uncorrel": 5, "furthermor": 5, "singl": [5, 10, 14, 22, 33, 39, 55, 56], "lend": 5, "itself": [5, 20, 21], "fast": [5, 8, 62], "architectur": [5, 61], "maximum_eigenvalu": 5, "arrayliketre": [5, 13, 14, 15, 18, 22, 29, 31, 33, 35, 36, 37, 39, 40, 44, 45, 46, 49], "largest": 5, "unbias": 5, "ratio": 5, "input": [5, 7, 14, 46], "approxim": [5, 9, 17, 19, 29, 33, 46, 48, 49, 50, 51], "well": [5, 6, 14, 15, 18, 51, 54, 61, 62], "except": [5, 52, 54, 55], "case": [5, 8, 20, 24, 39, 52, 53, 55], "larg": [5, 7, 27, 53], "small": [5, 7, 27, 53], "significantli": 5, "larger": [5, 55], "smaller": [5, 31], "directli": [5, 15, 18, 52, 55, 57], "becaus": [5, 21, 52, 54, 55], "latter": [5, 26, 53], "batch": [5, 9, 31, 33, 35, 36, 37, 39], "first": [5, 7, 8, 19, 26, 33, 39, 51, 52, 55, 60], "everi": [5, 9, 21, 55], "flatten": [5, 20, 31, 46], "dimension": [5, 20, 24, 46], "stack": [5, 54], "vertic": 5, "row": [5, 22, 46], "pathind": 6, "pathfinderadaptationst": 6, "ss_state": [6, 8], "dualaveragingadaptationst": [6, 7, 8], "8": [6, 8, 53, 54, 55, 56, 57, 58, 59, 60], "euclidean": [6, 8, 20, 61], "manifold": [6, 8, 20, 51], "two": [6, 8, 14, 15, 17, 20, 21, 23, 24, 26, 33, 39, 53, 55, 56, 62], "pathfind": [6, 11, 29, 47, 51], "ran": 6, "subsequ": [6, 26, 54, 56], "new": [6, 8, 9, 13, 14, 15, 18, 19, 21, 22, 23, 24, 26, 38, 39, 44, 55, 61, 62], "point": [6, 7, 13, 19, 22, 26, 27, 40, 43, 49, 58, 61], "markov": [6, 9, 39, 51], "supposedli": 6, "closer": [6, 31], "typic": [6, 8, 15, 26, 33, 44], "set": [6, 8, 20, 22, 23, 26, 31, 50, 51, 52, 55], "start": [6, 7, 9, 19, 21, 26, 40, 43, 49, 52], "mcmc": [6, 7, 8, 9, 10, 11, 38, 39, 44, 51, 53, 55, 61], "so": [6, 7, 15, 18, 20, 27, 35, 36, 37, 53, 54, 55, 56, 61], "reach": [6, 7, 8, 21], "nut": [6, 8, 11, 16, 26, 54, 56, 57, 58, 59, 62], "extra_paramet": [6, 8], "fmaili": [6, 8], "whose": [6, 8, 22, 49], "being": [6, 8, 24, 26, 39, 55, 61], "dure": [6, 7, 8], "extra": [6, 8, 39, 60], "pass": [6, 8, 9, 15, 18, 23, 29, 33, 39, 49, 56], "g": [6, 8, 29, 31, 52, 55, 56], "log_step_s": [7, 27], "logarithm": [7, 19, 22, 23, 27, 40], "log_step_size_avg": 7, "time": [7, 9, 15, 17, 18, 21, 22, 26, 27, 35, 36, 37, 38, 53, 54, 55, 58, 60, 61], "taken": [7, 21, 23, 27, 48, 52], "far": [7, 27], "avg_err": [7, 27], "quantiti": [7, 9, 27], "h_t": [7, 27], "mu": [7, 27, 46, 48, 57, 59, 60], "arbitrari": [7, 14, 15, 21, 27, 53], "shrunk": [7, 27], "toward": [7, 27], "chose": [7, 27], "10": [7, 26, 27, 29, 31, 49, 54, 55, 56, 57, 59, 60, 61, 62], "epsilon_0": [7, 27], "context": [7, 27, 33, 58, 61], "find_reasonable_step_s": [7, 27], "avg_error": [7, 27], "dual_averaging_adapt": 7, "t0": [7, 27], "gamma": [7, 27, 29, 49], "05": [7, 27, 29, 49], "kappa": [7, 27], "75": [7, 8, 27], "order": [7, 15, 18, 19, 20, 24, 26, 48, 52, 54], "achiev": [7, 52, 53, 54], "desir": 7, "let": [7, 24, 52, 54, 55, 56, 57, 59, 60, 61], "u": [7, 21, 25, 26, 31, 51, 52, 56, 57, 59, 60], "note": [7, 10, 17, 24, 46, 49, 52, 57, 61], "epsilon": 7, "alpha_t": 7, "t": [7, 10, 20, 23, 24, 39, 53, 55, 56], "acept": 7, "defin": [7, 14, 20, 23, 29, 54, 56, 58, 61], "error": [7, 17, 52], "would": [7, 53, 56], "like": [7, 8, 13, 26, 35, 55, 56, 58], "find": [7, 8, 27, 38, 40, 60], "h": [7, 23, 52, 56], "x": [7, 19, 23, 29, 31, 44, 50, 52, 53, 54, 55, 56, 58, 61, 62], "mathbb": 7, "nes09": [7, 27, 51], "author": [7, 27], "hg": [7, 21, 26, 27, 51], "14": [7, 21, 26, 27, 32, 51, 53, 56], "guarante": [7, 17], "converg": [7, 8, 10, 27, 51, 55], "_t": [7, 10], "section": [7, 24, 51], "detail": [7, 8, 23, 24, 26, 27, 46, 55], "discuss": [7, 53, 55], "free": [7, 14, 27, 39, 51], "down": [7, 27], "introduc": [7, 27], "speed": [7, 8, 21, 27], "recommend": [7, 27, 54, 62], "past": [7, 27], "quickli": [7, 27], "forget": [7, 27], "kernel_gener": 7, "reference_st": 7, "hmcstate": [7, 15, 52, 53, 54, 61], "target_accept": 7, "65": 7, "reason": [7, 8, 50, 52, 54, 55, 61], "while": [7, 26, 52, 53, 54], "ani": [7, 31, 52, 53, 55, 56, 57, 59, 60], "choos": [7, 21, 50], "doubl": [7, 8, 21, 26], "halv": 7, "until": [7, 26], "kei": [7, 9, 15, 20, 39, 46, 48, 49, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62], "take": [7, 9, 13, 14, 15, 18, 19, 22, 23, 24, 26, 31, 33, 35, 36, 37, 38, 39, 40, 44, 46, 48, 49, 55, 58, 59, 60, 61], "correspond": [7, 9, 20, 23, 33, 39, 46, 53, 62], "reference_hmc_st": 7, "locat": [7, 49], "must": [7, 10, 24, 39, 43, 53], "found": [7, 14, 15, 21, 54, 59], "never": 7, "advanc": [7, 51, 61], "rel": [7, 40, 57], "search": [7, 29, 43, 49], "onc": 7, "windowadaptationst": 8, "imm_stat": 8, "is_mass_matrix_diagon": 8, "schedul": [8, 35, 36, 37, 38], "match": [8, 20, 31, 38, 46, 61], "close": [8, 52], "unlik": [8, 23, 46], "sever": [8, 26, 33, 52, 55], "phase": 8, "explicitli": [8, 61], "better": [8, 13, 14, 19, 22, 24, 31, 48, 52, 54, 55, 56], "modular": [8, 61], "doe": [8, 14, 24, 26, 52, 57, 61], "affect": [8, 46], "allow": [8, 9, 52, 55, 58], "user": [8, 13, 14, 15, 18, 21, 22, 24, 31, 35, 36, 37, 38, 46, 49, 50, 53, 54, 55, 58, 61, 62], "own": [8, 52], "thei": [8, 9, 22, 23, 24, 33, 52, 53, 62], "want": [8, 22, 48, 49, 52, 55, 62], "decoupl": 8, "consist": [8, 21, 53], "window": [8, 57, 58, 59, 60], "nesterov": [8, 51], "both": [8, 17, 23, 55], "recomput": 8, "end": [8, 13, 43, 52, 56], "re": [8, 14, 51, 52, 54, 58], "schemat": 8, "In": [8, 9, 15, 18, 20, 23, 24, 26, 39, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61], "still": [8, 10, 61], "creat": [8, 9, 22, 38, 44, 58], "progress_bar": 8, "least": 8, "link": [8, 15, 19, 24, 59], "metric": [8, 11, 16, 29, 49, 51, 61], "import": [8, 15, 20, 21, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62], "especi": [8, 52], "extrem": 8, "ineffici": 8, "wrong": 8, "provid": [8, 15, 18, 24, 35, 36, 37, 52, 55, 61, 62], "purpos": [8, 53], "origin": [8, 15, 18, 19, 21, 24, 55, 57, 60, 61], "evolv": 8, "improv": 8, "qualiti": 8, "whether": [8, 15, 18, 19, 20, 21, 24, 26, 29, 48, 54], "displai": 8, "progress": [8, 26, 30, 51], "bar": [8, 30, 51], "object": [8, 27, 29, 39, 53], "build_schedul": 8, "num_step": [8, 26, 61], "initial_buffer_s": 8, "final_buffer_s": 8, "50": [8, 19, 39, 56, 58], "first_window_s": 8, "25": [8, 53, 56, 57], "list": [8, 22], "below": [8, 53, 56], "intend": 8, "period": [8, 22], "stage": [8, 17], "interv": [8, 17, 43, 56], "global": 8, "inform": [8, 9, 13, 14, 15, 18, 19, 21, 22, 24, 38, 39, 44, 49, 51, 55, 60], "seri": [8, 51], "expand": 8, "memori": [8, 26], "twice": [8, 9, 53], "preced": 8, "distinct": [8, 55], "come": 8, "common": 8, "accur": 8, "label": [8, 58], "initial_buff": 8, "width": 8, "window_label": 8, "is_middle_window_end": 8, "initfn": 9, "often": [9, 26], "need": [9, 13, 15, 18, 19, 21, 35, 36, 37, 49, 52, 54, 55, 56, 57, 58, 59, 60, 61, 62], "some": [9, 23, 24, 55], "same": [9, 14, 22, 26, 53, 55, 61], "therefor": 9, "do": [9, 15, 18, 21, 26, 35, 36, 37, 52, 53, 54, 56, 61], "oper": [9, 29, 55, 56], "themselv": 9, "function": [9, 53, 55, 56, 57, 58, 59, 60, 61, 62], "samplingalgorithm": [9, 13, 14, 15, 18, 19, 21, 22, 24, 31, 35, 36, 37, 38, 50], "updatefn": 9, "transit": [9, 13, 14, 15, 18, 19, 21, 22, 23, 24, 26, 38, 39, 44, 55, 57, 61], "pure": [9, 53, 54, 62], "idempot": 9, "necessarili": 9, "about": [9, 13, 14, 15, 18, 19, 22, 24, 38, 39, 44, 54, 60], "transtion": 9, "simplifi": 9, "yet": [9, 10, 14, 15, 21, 54], "univers": [9, 55], "interfac": [9, 13, 14, 15, 18, 21, 22, 24, 31, 35, 36, 37, 38, 49, 50, 55, 61], "essenc": [9, 21], "what": [9, 54, 55], "rng": 9, "possibli": 9, "data": [9, 31, 33, 35, 36, 37, 51, 55, 58], "pair": [9, 14], "As": [9, 54, 55, 57, 60], "intern": [9, 51, 52, 55, 57, 59], "save": [9, 53], "addit": [9, 13, 15, 18, 19, 21, 23, 24, 26, 39], "call": [9, 22, 39, 50, 53, 55], "vialgorithm": [9, 48], "variat": [9, 29, 33, 39, 48, 49, 51], "infer": [9, 29, 48, 49, 51, 56, 57, 58, 59, 60], "potenti": [9, 10, 22, 39, 51, 57, 61, 62], "guid": 9, "draw": [9, 10, 14, 15, 18, 21, 22, 24, 29, 38, 49, 51, 53], "build": [9, 13, 14, 15, 18, 19, 21, 22, 23, 24, 26, 33, 38, 40, 44, 46, 52, 54, 56, 58, 62], "runfn": 9, "adapt": [9, 11, 15, 18, 30, 38, 43, 51, 55, 56, 57, 58, 59, 60], "potential_scale_reduct": 10, "input_arrai": 10, "arraylik": 10, "chain_axi": 10, "sample_axi": 10, "gelman": [10, 51], "rubin": [10, 51], "1992": [10, 51], "reduct": [10, 51], "multipl": [10, 46, 51, 58], "axi": [10, 54, 58], "indic": [10, 13, 22, 26, 29, 46, 54], "ndarrai": [10, 40], "r": [10, 51, 55, 56], "hat": [10, 55], "squeez": 10, "frac": [10, 20, 52, 56], "v": [10, 44, 51, 53, 56, 57, 59, 60], "w": [10, 14, 56, 58], "within": [10, 54], "posterior": [10, 13, 22, 39, 49, 53, 54, 55, 56, 57, 58, 59, 60, 61], "pool": 10, "trace": 10, "factor": [10, 29, 49], "uniti": 10, "greater": 10, "stac": [10, 51], "gr92": [10, 51], "effective_sample_s": 10, "effect": [10, 38, 40, 51, 55, 57, 60], "ess": [10, 11, 38, 41], "basic": [10, 13, 14, 15, 18, 21, 22, 24, 31, 35, 36, 37, 38, 49, 50, 61], "n_": 10, "mathit": 10, "eff": 10, "mn": 10, "tau": [10, 57, 59], "k": [10, 20, 29, 49, 51, 53, 56, 57, 59, 60], "p": [10, 20, 24, 44, 52, 53, 56], "m": [10, 19, 20, 31, 39, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61], "rho": [10, 48], "_autocorrel": 10, "lag": 10, "integ": [10, 22], "2k": 10, "staa": [10, 51], "gcsr95": [10, 51], "similar": [10, 14, 21, 55], "geyer": [10, 51], "monoton": 10, "gey92": [10, 51], "gey11": [10, 51], "diffus": [11, 16, 34], "elliptical_slic": [11, 16], "ghmc": [11, 16], "mala": [11, 16], "marginal_latent_gaussian": [11, 16], "periodic_orbit": [11, 16], "random_walk": [11, 16, 53], "termin": [11, 16, 26, 29, 49], "sgmcmc": 11, "csgld": [11, 34], "sghmc": [11, 32, 34], "sgld": [11, 31, 34], "sgnht": [11, 32, 34], "smc": [11, 15, 18], "adaptive_temp": [11, 41], "resampl": [11, 38, 39, 41, 44], "solver": [11, 12, 32, 38, 40, 41], "temper": [11, 38, 39, 40, 41, 43], "vi": [11, 51], "meanfield_vi": [11, 47], "svgd": [11, 47], "langevin": [12, 18, 31, 32, 36, 51], "overdamped_langevin": [12, 32], "logdensity_grad_fn": [12, 54], "euler": [12, 32], "overdamp": [12, 32], "ellipt": [13, 51], "ellipslicest": 13, "logdens": [13, 14, 15, 18, 19, 22, 24, 31, 52, 53, 55, 57, 61], "evalu": [13, 22, 52, 54, 55, 61], "ellipsliceinfo": 13, "debug": [13, 15, 18, 19, 21, 24, 56], "latent": [13, 19, 22], "theta": [13, 56, 57, 59], "2pi": 13, "identifi": 13, "ellipsi": 13, "drawn": [13, 49, 53, 61], "positon": 13, "mommentum": 13, "subit": 13, "sub": [13, 56], "subiter": 13, "less": [13, 53, 54, 56], "build_kernel": [13, 14, 15, 18, 21, 22, 24, 31, 35, 36, 37, 38, 44, 50, 53], "cov_matrix": 13, "mam10": [13, 51], "gaussian": [13, 19, 20, 24, 46], "prior": [13, 19, 33, 38, 44, 54, 56, 58], "along": [13, 14, 15, 18, 19, 20, 22, 23, 24, 38, 44, 49, 62], "ellip_slic": 13, "loglikelihood_fn": [13, 33, 35, 36, 37, 38, 44], "compil": [13, 14, 15, 18, 19, 21, 22, 24, 35, 36, 37, 53, 55, 56], "likelihood": [13, 19, 33, 38, 39, 44], "distributon": 13, "ghmcstate": 14, "henc": [14, 61], "thu": [14, 20, 23, 26, 52, 55, 58], "after": [14, 23, 53], "logdensity_grad": [14, 15, 18, 19, 52, 54, 61], "noise_fn": 14, "lambda": [14, 15, 53, 54, 57, 58, 61], "divergence_threshold": [14, 15, 21, 26], "1000": [14, 15, 21, 29, 49, 56, 57, 59, 60, 61], "apart": 14, "equival": [14, 26, 46, 54], "correct": [14, 24, 29, 49, 55], "atom": 14, "energi": [14, 15, 19, 20, 21, 22, 23, 24, 26, 31, 37, 61], "abov": [14, 15, 21, 26, 52, 53], "consid": [14, 15, 21, 23, 52, 54], "diverg": [14, 15, 21, 23, 26, 48, 49, 57], "mechan": [14, 26, 52], "done": [14, 19, 54, 55, 61], "determinisitc": 14, "alwai": [14, 15, 18, 21, 48, 56], "candiat": 14, "ghmc_kernel": 14, "structur": [14, 23, 46, 52], "translat": [14, 57, 59, 60], "absolut": [14, 15, 21], "sai": [14, 15, 21, 26], "commonli": [14, 15, 21], "noise_gn": 14, "dynamichmcst": 15, "hmcinfo": 15, "wa": [15, 18, 19, 21, 24, 32, 33, 52, 55, 61], "acceptance_r": [15, 18, 19, 21, 24, 57, 60], "is_accept": [15, 18, 19, 24], "is_diverg": [15, 21, 57], "exceed": [15, 21], "threshold": [15, 21], "total": [15, 23, 26, 53, 54, 55, 56, 58, 61], "includ": [15, 29], "symplect": [15, 17, 18, 21, 22, 26], "integratorst": [15, 21, 26], "init_dynam": 15, "velocity_verlet": [15, 17, 21, 22], "build_dynamic_kernel": 15, "integration_steps_fn": 15, "randomli": 15, "builder": [15, 18, 35, 36], "alia": [15, 18, 35, 36, 37], "cumbersom": [15, 18, 35, 36, 37, 61], "manipul": [15, 18, 35, 36, 37], "sinc": [15, 18, 35, 36, 37, 57], "specifi": [15, 18, 35, 36, 37, 61], "helper": [15, 18, 35, 36, 37], "special": [15, 18, 35, 36, 37], "attribut": [15, 18, 19], "etc": [15, 18, 53, 61], "you": [15, 18, 21, 35, 36, 37, 52, 54, 55, 62], "mclachlan": [15, 17, 51], "kinet": [15, 20, 21, 22, 26, 37], "befor": [15, 21, 53, 55], "palindrom": 17, "deriv": 17, "bcss14": [17, 51], "form": [17, 26, 29, 46, 52], "b1": 17, "a1": 17, "b2": 17, "choic": 17, "bound": 17, "method": [17, 26, 27, 51, 52, 53], "respect": [17, 20, 26, 39, 52, 56], "here": [17, 24, 26, 35, 36, 37, 39, 46, 53, 55], "mcl95": [17, 51], "focus": 17, "known": [17, 61], "minim": [17, 27, 29, 40, 49, 52], "norm": [17, 52, 55, 61, 62], "a2": 17, "4": [17, 26, 29, 39, 51, 54, 55, 56, 57, 58], "67": 17, "yoshida": [17, 21], "noneuclidean_leapfrog": 17, "noneuclidean_mclachlan": 17, "noneuclidean_yoshida": 17, "adjust": 18, "malast": 18, "malainfo": 18, "margin": [19, 53, 54, 56], "marginalst": 19, "rmh": [19, 24, 53], "log_p_x": 19, "model": [19, 31, 38, 49, 55, 61, 62], "grad_x": [19, 54], "u_x": 19, "auxiliari": [19, 51], "u_grad_x": 19, "marginalinfo": 19, "init_and_kernel": 19, "sampler": [19, 24, 39, 51, 52, 54, 55, 56, 57, 59, 62], "mgrad_gaussian": 19, "tp18": [19, 51], "log_likelihood": 19, "interestingli": 19, "calibr": 19, "roughli": [19, 39, 55], "q": [19, 52, 56], "exp": [19, 44, 55, 60, 61], "f": [19, 29, 40, 43, 52, 56, 57, 58, 60], "c": [19, 23, 56, 59, 62], "use_invers": 19, "zero": [19, 52, 60], "embed": 20, "particular": [20, 23, 24], "practic": [20, 51, 61], "definit": [20, 50], "fix": [20, 26, 53, 56], "independ": [20, 24, 51, 55, 58], "bblg17": [20, 51], "For": [20, 24, 46, 53, 55, 57, 59], "newtonian": 20, "relativist": [20, 51], "lph": [20, 51], "17": [20, 51, 56], "gaussian_euclidean": 20, "euclideankineticenergi": 20, "normal": [20, 24, 49, 51, 53, 54, 55, 57, 58, 59, 60, 61, 62], "bet13": [20, 51], "further": 20, "character": 20, "condit": [20, 53, 60], "pi": [20, 31], "assum": [20, 24, 35, 36, 37, 43, 61], "One": 20, "multipli": [20, 46], "tree": [20, 21, 26], "specif": [20, 24, 52, 53], "ravel_pytre": 20, "sort": 20, "dictionari": [20, 53, 56, 58], "them": [20, 39, 57, 61], "appear": [20, 33], "momentum_gener": 20, "kinetic_energi": [20, 26], "is_turn": [20, 21, 25], "turn": [20, 21, 25, 26, 51], "back": [20, 21, 54, 57, 61], "nutsinfo": 21, "trajectory_leftmost_st": 21, "leftmost": 21, "full": [21, 33, 46], "trajectory_rightmost_st": 21, "rightmost": 21, "num_trajectory_expans": 21, "subtrajectori": 21, "were": [21, 39, 57], "probabilti": 21, "across": [21, 23, 55], "entir": [21, 23], "major": 21, "samplig": 21, "multinomi": [21, 26, 42], "bet17": [21, 23, 26, 51], "expans": [21, 26], "recurs": [21, 26], "ppj19": [21, 51], "lsl": [21, 51], "20": [21, 51, 52, 55, 56, 58, 61, 62], "unusu": 21, "those": [21, 55], "familiar": [21, 24], "inde": [21, 52, 54], "conceptu": [21, 51], "construct": 21, "feel": 21, "lingo": 21, "inherit": 21, "unnecessarili": 21, "complic": 21, "hide": 21, "concept": 21, "upon": 21, "built": [21, 51, 54], "direct": [21, 22, 26], "continu": [21, 24], "simplect": 21, "max_num_doubl": 21, "obserb": 21, "occur": 21, "orbit": [22, 51], "periodicorbitalst": 22, "accord": [22, 32], "reweight": 22, "vector": [22, 24, 31, 46], "neg": [22, 31, 56], "logdensities_grad": 22, "anyth": 22, "name": [22, 54, 58, 59, 60], "either": [22, 24, 39, 46, 55], "numpi": [22, 46, 52, 53, 54, 55, 57, 58, 59, 60, 61, 62], "repeat": [22, 26], "assign": [22, 39], "biject": 22, "nw22": [22, 51], "orbital_hmc": 22, "connect": 22, "precis": [22, 46], "per_orbit": 22, "buid": 22, "trajectoryst": 23, "canon": 23, "z": [23, 29, 53, 56, 57], "sum_log_p_accept": 23, "cumul": 23, "proposal_gener": [23, 24], "associ": [23, 24, 48], "proposal_from_energy_diff": 23, "initial_energi": 23, "new_energi": 23, "flag": 23, "asymmetric_proposal_gener": 23, "transition_energy_fn": 23, "proposal_factori": 23, "account": 23, "asymmetri": 23, "keyword": [23, 29, 49], "argument": [23, 29, 33, 49, 53, 56], "samplepropos": 23, "proto": 23, "def": [23, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62], "meth": 23, "self": [23, 54], "Such": 23, "primarili": 23, "static": [23, 26, 52], "checker": 23, "recogn": [23, 55], "subtyp": 23, "duck": 23, "func": 23, "check": [23, 26, 52, 56], "pep": 23, "544": 23, "decor": [23, 30], "runtime_check": 23, "act": 23, "simpl": [23, 33, 51, 54, 61], "mind": [23, 62], "runtim": [23, 61], "presenc": 23, "ignor": [23, 40], "signatur": [23, 38], "genproto": 23, "static_binomial_sampl": [23, 24], "new_propos": 23, "lower": [23, 48, 51], "progressive_uniform_sampl": 23, "progressive_biased_sampl": 23, "bais": 23, "bias": 23, "awai": 23, "nonreversible_slice_sampl": 23, "decis": [23, 51], "nea20": [23, 51], "walk": 24, "rosenbluth": 24, "expos": 24, "conveni": [24, 52], "entri": 24, "level": [24, 48, 53, 55], "who": [24, 54, 55, 62], "simpler": [24, 56], "instanti": 24, "x_": [24, 39], "x_t": [24, 39], "newli": 24, "variant": 24, "offer": 24, "noic": 24, "additive_step": 24, "doesn": [24, 55], "t_1": 24, "irmh": 24, "symmetr": [24, 51], "That": [24, 54, 55], "without": [24, 54, 58], "proposal_logdensity_fn": 24, "asymmetr": 24, "gcsr14": [24, 51], "11": [24, 56, 57, 59, 60], "simplest": 24, "additive_step_random_walk": 24, "sigma": [24, 46, 53, 57, 59], "distanc": [24, 50], "suitabl": 24, "center": [24, 33, 57, 59, 60], "rwstate": [24, 53], "rw": 24, "log_dens": [24, 53], "rwinfo": 24, "build_additive_step": 24, "random_step": 24, "alreadi": [24, 53, 54], "compon": [24, 53], "covariance_matrix": 24, "rw_gaussian": 24, "normal_random_walk": 24, "produc": [24, 49, 52], "ad": 24, "obtain": [24, 53, 55, 58, 61], "maintain": [24, 53], "balanc": [24, 26], "classmethod": 24, "build_irmh": 24, "impli": [24, 54], "particl": [24, 38, 39, 40, 44, 50], "mutat": 24, "wan22": [24, 51], "proposal_distribut": 24, "know": [24, 52, 54], "build_rmh": 24, "appli": [24, 38, 53, 54, 55, 56, 57, 58], "build_rmh_transition_energi": 24, "rmh_propos": 24, "transition_distribut": 24, "init_propos": 24, "generate_propos": 24, "sample_propos": 24, "iterativeuturnst": 25, "momentum_sum": [25, 26], "idx_min": 25, "idx_max": 25, "iterative_uturn_numpyro": 25, "numpyro": [25, 51, 54], "style": 25, "proce": 26, "process": [26, 49, 51, 55, 61], "tradit": 26, "flip": 26, "distinguish": 26, "fall": [26, 54], "former": [26, 53], "categori": [26, 62], "There": [26, 57, 62], "faster": [26, 52, 55, 56], "risk": 26, "satur": 26, "keep": [26, 48], "discard": 26, "leftmost_st": 26, "rightmost_st": 26, "num_stat": 26, "append_to_trajectori": 26, "append": [26, 54], "reorder_trajectori": 26, "new_trajectori": 26, "merge_trajectori": 26, "left_trajectori": 26, "right_trajectori": 26, "static_integr": 26, "dynamicintegrationst": 26, "termination_st": 26, "dynamic_progressive_integr": 26, "update_termination_st": 26, "is_criterion_met": 26, "sequenti": [26, 51], "met": 26, "consecut": 26, "dynamic_recursive_integr": 26, "uturn_check_fn": 26, "use_robust_uturn_check": 26, "python": [26, 54, 59, 62], "6": [26, 39, 56, 57, 61], "implemen": 26, "mostli": 26, "valid": 26, "sure": [26, 54, 57], "actual": 26, "cannot": [26, 52, 54], "dynamicexpansionst": 26, "dynamic_multiplicative_expans": 26, "trajectory_integr": 26, "max_num_expans": 26, "pick": [26, 50], "stop": [26, 52], "prematur": 26, "els": 26, "geometr": [26, 31, 51], "why": [26, 52, 55, 61], "literatur": 26, "hmc_energi": 26, "log_x": 27, "log_x_avg": 27, "primal": [27, 51, 52], "subgradi": [27, 51], "mathemat": [27, 51], "lbfg": [28, 49], "lbfgshistori": 29, "path": [29, 49, 51], "l": [29, 49, 56], "bfg": [29, 40, 49, 52], "hessian": [29, 49], "update_mask": 29, "xi": [29, 37], "paper": 29, "minimize_lbfg": 29, "fun": [29, 38, 43, 52], "x0": 29, "maxit": [29, 49], "30": [29, 49, 51, 57], "maxcor": [29, 49], "gtol": [29, 49], "1e": [29, 35, 36, 37, 49, 52, 53, 54, 55, 61, 62], "08": [29, 49, 61], "ftol": [29, 49], "maxl": [29, 49], "lbfgs_kwarg": [29, 49], "jaxopt": [29, 49], "optstep": 29, "real": [29, 61], "scalar": [29, 37, 46, 56], "compos": [29, 51], "vjp": [29, 54], "guess": 29, "f_k": [29, 49], "f_": [29, 49], "g_k": [29, 49], "_norm": [29, 49], "lbfgs_inverse_hessian_factor": 29, "represent": 29, "formula": 29, "ii": 29, "newton": [29, 49, 51], "lu": [29, 51], "zhang": [29, 51], "et": [29, 56], "al": 29, "arxiv": [29, 51], "2108": 29, "03782": 29, "lbfgs_inverse_hessian_formula_1": 29, "beta": [29, 32, 35, 37, 49, 56], "lbfgs_inverse_hessian_formula_2": 29, "bfgs_sampl": 29, "num_sampl": [29, 30, 42, 48, 49, 53, 55, 56, 57, 58, 59, 60, 61], "grad_posit": [29, 49], "jeremi": [30, 51], "coullon": [30, 51], "blog": 30, "post": [30, 61], "cou": [30, 51], "progress_bar_scan": 30, "print_rat": 30, "scan": [30, 51, 52, 53, 55, 56, 57, 58, 59, 60, 61], "contour": [31, 51], "stochast": [31, 35, 36, 37, 51], "dlh": [31, 51], "22": [31, 51, 56], "dll20": [31, 51], "contoursgldst": 31, "energy_pdf": 31, "th": 31, "int_": 31, "s_1": 31, "mathrm": [31, 44], "d": [31, 44, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61], "s_i": 31, "partit": 31, "energy_idx": 31, "index": [31, 39, 46, 56], "belong": 31, "num_partit": 31, "512": 31, "energy_gap": 31, "min_energi": 31, "divid": 31, "landscap": 31, "rang": [31, 54, 57, 62], "rough": 31, "minimum": 31, "dataset": [31, 33, 56, 58, 61], "strictli": 31, "exact": [31, 51], "3456": 31, "enough": [31, 52], "gap": 31, "logdensity_estim": [31, 33], "gradient_estim": 31, "zeta": 31, "hyperparamet": 31, "reduc": 31, "temperatur": [31, 38], "port": [32, 33], "cn22": [32, 33, 51], "01": [32, 35, 37, 51, 55, 61], "equat": [32, 51, 52], "cfg14": [32, 51], "mcf15": [32, 51], "dfb": [32, 51], "logprior_fn": [33, 35, 36, 37, 38, 44], "data_s": [33, 35, 36, 37], "rm51": [33, 51], "instanc": [33, 46, 55], "supervis": 33, "item": [33, 53, 54, 56, 57, 60], "grad_estim": [33, 35, 36, 37], "control_vari": 33, "logdensity_grad_estim": 33, "centering_posit": 33, "bffn19": [33, 51], "map": [33, 46], "constant": [35, 36, 37], "now": [35, 36, 37, 52, 53, 54, 55, 57, 58, 59, 60], "yield": [35, 36, 37, 49, 60], "minibatch": [35, 36, 37], "new_posit": [35, 36], "grad_fn": 36, "nos\u00e9": 37, "hoover": 37, "thermostat": [37, 51], "sgnhtstate": 37, "mcmc_step_fn": [38, 44], "mcmc_init_fn": [38, 44], "resampling_fn": [38, 44], "target_ess": [38, 40], "root_solv": [38, 40], "dichotomi": [38, 43], "mcmc_kernel_factori": 38, "make_mcmc_st": 38, "delta_0": 38, "min_delta": [38, 40, 43], "max_delta": [38, 40, 43], "use_log_ess": 38, "solv": [38, 40, 43], "usual": [38, 46, 61], "temperedsmcst": [38, 44], "adaptive_tempered_smc": 38, "mcmc_paramet": 38, "num_mcmc_step": 38, "smcstate": [39, 40], "leav": 39, "n_particl": 39, "univari": [39, 55, 61], "bivari": 39, "51": [39, 61], "55": 39, "second": [39, 55], "52": [39, 56, 58, 61], "54": [39, 61], "60": [39, 61], "70": 39, "smcinfo": 39, "ancestor": 39, "log_likelihood_incr": 39, "increment": [39, 40, 43], "due": 39, "update_info": 39, "update_fn": 39, "weigh_fn": 39, "resample_fn": 39, "num_resampl": 39, "g_t": 39, "weigh": 39, "pmap": 39, "feynman": 39, "kac": 39, "term": [39, 44, 48], "goe": [39, 57], "m_t": 39, "r_t": 39, "idx": 39, "x_tm1": 39, "wast": [39, 51], "dc20": [39, 51], "charg": 39, "new_particl": 39, "thing": [40, 42, 43, 55], "log_weight": 40, "log_ess": 40, "np": [40, 54, 55, 57, 59, 60, 61, 62], "ess_solv": 40, "smc_state": 40, "max": [40, 51, 53], "root": [40, 43], "delta0": 40, "min": 40, "systemat": 42, "stratifi": 42, "residu": 42, "_delta0": 43, "ep": [43, 56], "0001": 43, "max_it": 43, "100": [43, 53, 54, 57, 58], "decreas": 43, "toler": 43, "b": [43, 46, 51, 56, 58], "lmbda": 44, "propto": 44, "p_0": 44, "easi": [44, 61], "unnorm": [44, 62], "pointwis": 44, "num_mcmc_iter": 44, "linear_map": 46, "diag_or_dense_a": 46, "highest": [46, 49], "linear": [46, 55], "y": [46, 50, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61], "ax": [46, 61], "dispatch": [46, 52], "jnp": [46, 52, 53, 55, 58, 59, 60, 61, 62], "dot": [46, 58], "dtype": [46, 52, 53, 54, 56, 60, 61], "2nd": 46, "1d": 46, "2d": 46, "googl": 46, "7859": 46, "lax": [46, 52, 53, 55, 56, 57, 58, 59, 60, 61], "dot_gener": 46, "generate_gaussian_nois": 46, "generate_unit_vector": 46, "unit": 46, "pytree_s": 46, "index_pytre": 46, "input_pytre": 46, "flat": 46, "variou": 46, "somehow": 46, "simul": [46, 51], "hamilonian": 46, "unclear": [46, 55], "interact": [46, 51], "demonstr": [46, 52, 55], "wai": [46, 53], "tell": [46, 56], "column": 46, "individu": 46, "arang": [46, 55], "mfvistat": 48, "opt_stat": 48, "mfviinfo": 48, "elbo": [48, 49], "stl_estim": 48, "field": 48, "init_st": 48, "kullback": [48, 49], "leibler": [48, 49], "stick": [48, 51], "land": [48, 51], "stl": 48, "rwd17": [48, 51], "remov": 48, "score": 48, "suggest": 48, "asd20": [48, 51], "high": 48, "generate_meanfield_logdens": 48, "pathfinderst": 49, "local": 49, "inter": 49, "fo": 49, "wrt": 49, "rappresent": 49, "initial_posit": [49, 55, 57, 60, 61, 62], "200": [49, 55], "pathfinderinfo": 49, "zcgv22": [49, 51], "prpng": 49, "un": 49, "densifi": 49, "routin": 49, "lgbf": 49, "travers": 49, "union": 49, "prng": [49, 53], "lowest": 49, "kl": 49, "heavi": 49, "just": [49, 55, 62], "visamplingalgorithm": 49, "rbf_kernel": 50, "length_scal": 50, "update_median_heurist": 50, "svgdstate": 50, "median": 50, "bandwidth": 50, "rbf": 50, "middl": 50, "ground": 50, "empir": 50, "strategi": 50, "grad_logdensity_fn": 50, "semi": 50, "update_kernel_paramet": 50, "url": 51, "http": [51, 59], "mc": 51, "org": 51, "doc": 51, "2_27": 51, "html": [51, 59], "2_20": 51, "notat": 51, "abhinav": 51, "agraw": 51, "daniel": 51, "sheldon": 51, "justin": 51, "domk": 51, "black": 51, "box": [51, 53, 55], "flow": 51, "neural": [51, 58], "system": 51, "33": [51, 56], "17358": 51, "17369": 51, "2020": 51, "jack": 51, "baker": 51, "paul": 51, "fearnhead": 51, "emili": 51, "fox": 51, "christoph": 51, "nemeth": 51, "29": [51, 61], "599": 51, "615": 51, "2019": 51, "michael": 51, "betancourt": 51, "riemannian": 51, "scienc": 51, "confer": 51, "gsi": 51, "2013": 51, "pari": 51, "franc": 51, "august": 51, "28": [51, 57, 59, 60], "proceed": 51, "327": 51, "334": 51, "springer": 51, "introduct": [51, 55], "preprint": 51, "1701": 51, "02434": 51, "2017": 51, "simon": 51, "byrn": 51, "sam": 51, "livingston": 51, "mark": 51, "girolami": 51, "foundat": 51, "1410": 51, "5110": 51, "sergio": 51, "blane": 51, "fernando": 51, "casa": 51, "je": 51, "\u00fa": 51, "mar": 51, "\u0131": 51, "sanz": 51, "serna": 51, "hybrid": 51, "siam": 51, "journal": 51, "scientif": 51, "36": 51, "a1556": 51, "a1580": 51, "2014": 51, "tianqi": 51, "chen": 51, "guestrin": 51, "machin": [51, 54, 55], "1683": 51, "1691": 51, "pmlr": 51, "loop": [51, 52, 54], "www": 51, "jeremiecoullon": 51, "com": 51, "2021": 51, "jax_progress_bar": 51, "sgmcmcjax": 51, "lightweight": 51, "open": 51, "softwar": 51, "7": [51, 56, 57, 58, 59, 60, 61], "72": 51, "4113": 51, "2022": 51, "hai": 51, "dang": 51, "dau": 51, "nicola": 51, "chopin": 51, "2011": 51, "02328": 51, "wei": 51, "deng": 51, "siqi": 51, "liang": 51, "botao": 51, "hao": 51, "guang": 51, "lin": 51, "fame": 51, "2202": 51, "09867": 51, "multi": [51, 58], "modal": 51, "15725": 51, "15736": 51, "nan": 51, "ding": 51, "youhan": 51, "fang": 51, "ryan": 51, "babbush": 51, "changyou": 51, "robert": 51, "skeel": 51, "hartmut": 51, "neven": 51, "bayesian": [51, 58, 61], "andrew": 51, "john": 51, "carlin": 51, "hal": 51, "stern": 51, "donald": 51, "analysi": 51, "chapman": 51, "hall": 51, "crc": 51, "1995": 51, "page": 51, "457": 51, "472": 51, "charl": 51, "j": [51, 56, 57, 59], "473": 51, "483": 51, "handbook": 51, "20116022": 51, "45": 51, "matthew": 51, "hoffman": 51, "alexei": 51, "radul": 51, "pavel": 51, "sountsov": 51, "artifici": 51, "intellig": 51, "3907": 51, "3915": 51, "mach": 51, "15": [51, 53, 55, 56, 57, 59, 60, 61], "1593": 51, "1623": 51, "7799": 51, "7813": 51, "junpeng": 51, "lao": 51, "suter": 51, "ian": 51, "langmor": 51, "cyril": 51, "chimisov": 51, "ashish": 51, "saxena": 51, "dave": 51, "moor": 51, "rif": 51, "saurou": 51, "joshua": 51, "dillon": 51, "tfp": [51, 58, 59], "modern": [51, 55], "tool": 51, "hardwar": [51, 62], "2002": 51, "01184": 51, "xiaoyu": 51, "valerio": 51, "perron": 51, "leonard": 51, "hasenclev": 51, "yee": 51, "whye": 51, "teh": 51, "sebastian": 51, "vollmer": 51, "1236": 51, "1245": 51, "yi": 51, "ma": 51, "complet": [51, 55], "recip": 51, "2015": 51, "On": 51, "ordinari": 51, "differenti": [51, 52, 54, 56], "composit": 51, "16": [51, 56, 57, 59, 60], "151": 51, "168": 51, "iain": 51, "murrai": 51, "david": 51, "mackai": 51, "thirteenth": 51, "541": 51, "548": 51, "jmlr": 51, "workshop": 51, "2010": 51, "radford": 51, "neal": 51, "reversibli": 51, "2001": 51, "11950": 51, "kiril": 51, "neklyudov": 51, "5790": 51, "5814": 51, "yurii": 51, "convex": [51, 52], "problem": [51, 53, 55, 59, 60], "program": [51, 55, 58], "120": 51, "221": 51, "259": 51, "2009": 51, "du": [51, 56], "phan": 51, "neeraj": 51, "pradhan": 51, "martin": 51, "jankowiak": 51, "flexibl": 51, "acceler": [51, 62], "probabilist": [51, 55, 58], "1912": 51, "11554": 51, "herbert": 51, "robbin": 51, "sutton": 51, "monro": 51, "annal": 51, "400": 51, "407": 51, "1951": 51, "geoffrei": 51, "roeder": 51, "yuhuai": 51, "wu": 51, "duvenaud": 51, "michali": 51, "titsia": 51, "omiro": 51, "papaspiliopoulo": 51, "royal": 51, "societi": 51, "methodologi": 51, "80": 51, "749": 51, "767": 51, "2018": 51, "guanyang": 51, "wang": 51, "bernoulli": 51, "2012": 51, "2033": 51, "bob": 51, "carpent": 51, "aki": 51, "vehtari": 51, "research": [51, 55, 62], "23": [51, 56, 57], "306": 51, "49": [51, 56], "primit": [52, 53, 55, 57, 59, 60], "whenev": 52, "applic": 52, "sometim": 52, "your": [52, 54, 55, 62], "express": 52, "forward": [52, 55, 58], "mode": [52, 54, 56], "backward": 52, "despair": 52, "cover": [52, 55], "academ": 52, "understand": [52, 53, 55], "min_": 52, "tediou": 52, "minimis": 52, "happen": [52, 53, 55], "show": [52, 54, 56, 57, 58, 59, 60], "modifi": [52, 53], "overal": 52, "our": [52, 53, 56, 58], "theori": 52, "conjug": [52, 56], "transport": 52, "begin": [52, 53, 56, 58], "align": [52, 56], "langl": 52, "rangl": 52, "qquad": 52, "And": [52, 54, 55, 56, 59, 60], "min_i": 52, "datetim": [52, 53, 54, 55, 56, 57, 58, 59, 60, 61], "date": [52, 53, 54, 55, 56, 57, 58, 59, 60, 61], "todai": [52, 53, 54, 55, 56, 57, 58, 59, 60, 61], "strftime": [52, 53, 54, 55, 56, 57, 58, 59, 60, 61], "scipi": [52, 53, 55, 61, 62], "out": [52, 53, 54, 55], "ab": [52, 53], "later": 52, "undefin": 52, "rais": [52, 54], "has_aux": 52, "print": [52, 54, 55, 56, 57, 58, 60], "work": [52, 53, 55, 56, 62], "while_loop": 52, "fori_loop": 52, "leverag": 52, "asum": 52, "uniqu": 52, "df": [52, 56], "dx": [52, 56], "bigg": 52, "dh": [52, 56], "dy": [52, 56], "argmin": 52, "custom_vjp": [52, 54], "functool": [52, 58], "partial": [52, 58], "custom_jvp": 52, "nondiff_argnum": 52, "f_with_gradi": 52, "defjvp": 52, "f_jac_vec_prod": 52, "tangent": 52, "x_dot": 52, "fact": [52, 55], "f_out": 52, "31415": 52, "560483": 52, "float32": [52, 53, 54, 60, 61], "weak_typ": [52, 54, 61], "previous": 52, "obvious": [52, 54], "true_f": 52, "56049085": 52, "compar": [52, 54, 55], "5604908466339111": 52, "5604829788208008": 52, "word": 52, "suffic": 52, "toi": 52, "stat": [52, 53, 55, 61, 62], "logpdf": [52, 53, 55, 61, 62], "step_kei": 52, "5600991": 52, "6666666": 52, "0519627": 52, "6498852": 52, "7897134": 52, "techniqu": 53, "joint": [53, 56, 58], "newcommand": 53, "xx": 53, "boldsymbol": 53, "yy": 53, "altern": 53, "sim": [53, 56], "mid": 53, "ideal": 53, "analyt": 53, "appropri": 53, "hand": [53, 54], "idea": [53, 55], "block": [53, 62], "xx_1": 53, "ldot": 53, "xx_n": 53, "simplic": 53, "notebook": [53, 55, 57, 58, 59, 60, 61], "focu": 53, "jsp": 53, "suppos": 53, "multivari": 53, "operatornam": [53, 56], "mvnormal": 53, "pdf": 53, "concaten": 53, "multivariate_norm": 53, "zeros_lik": 53, "cov": 53, "illustr": [53, 55], "ll": [53, 55, 61], "write": [53, 54], "mwg_kernel": 53, "hold": 53, "correctli": [53, 54, 57], "issu": [53, 55], "algorithmst": 53, "class": [53, 58], "log_prob": 53, "xx_": 53, "yy_": 53, "via": [53, 54, 55, 62], "xx_t": 53, "yy_t": 53, "cdot": 53, "approach": [53, 58], "readi": [53, 54, 55, 61], "paramt": 53, "mwg_init_x": 53, "mwg_init_i": 53, "mwg_step_fn_x": 53, "mwg_step_fn_i": 53, "rmcstate": 53, "step_fn": 53, "rng_key_x": 53, "rng_key_i": 53, "num": 53, "copi": 53, "logdensity_x": 53, "logdensity_i": 53, "transition_gener": 53, "ey": 53, "initial_st": [53, 55, 56, 57, 58, 59, 60, 61], "sampling_loop": 53, "one_step": [53, 55, 56, 57, 58, 59, 60, 61], "sample_kei": [53, 55, 56, 57, 58, 59, 60, 61], "10_000": [53, 61], "cpu": [53, 54, 58, 61, 62], "13": [53, 56, 61], "sy": [53, 54, 55, 58, 61], "wall": [53, 54, 55, 58, 61], "matplotlib": [53, 57, 59, 60, 61], "pyplot": [53, 57, 59, 60, 61], "plt": [53, 57, 59, 60, 61], "arviz": [53, 56, 57, 59, 60], "az": [53, 56, 57, 59, 60], "idata": [53, 56, 57, 59, 60], "from_dict": [53, 56, 57, 59, 60], "plot_pair": 53, "kind": [53, 54, 56], "hexbin": 53, "tight_layout": [53, 57, 59, 60], "attempt": 53, "mwg_kernel_gener": 53, "elemtn": 53, "chi": 53, "len": [53, 55, 56, 58], "zip": [53, 58, 59], "logdensity_k": 53, "kwarg": 53, "_k": 53, "sampling_loop_gener": 53, "positions_gener": 53, "reus": 53, "799": 53, "814": 53, "804": 53, "tree_map": 53, "But": 53, "ve": [53, 55], "could": [53, 55], "ourselv": 53, "cost": 53, "expens": [53, 55], "deeper": 53, "low": 53, "generaliz": 53, "prototyp": 53, "adequ": 53, "get": [53, 54, 58, 59, 61], "unrol": 53, "caus": 53, "long": [53, 55, 56, 57, 59, 60, 62], "mitig": 53, "replac": 53, "sake": 53, "fulli": 53, "blackjax": [54, 55, 56, 57, 58, 59, 60], "oryx": 54, "pymc": 54, "tensorflow": [54, 60], "nevertheless": 54, "incompat": 54, "compatibl": 54, "judg": [54, 55], "hint": 54, "solut": [54, 55], "tensor": [54, 56, 59], "srng": [54, 56], "randomstream": [54, 56], "loc": [54, 55, 61, 62], "n_rv": 54, "i_rv": 54, "categor": [54, 58], "y_rv": [54, 56], "opt": [54, 59], "hostedtoolcach": [54, 59], "9": [54, 55, 56, 57, 59, 60, 61], "18": [54, 55, 56, 57, 58, 59, 60], "x64": [54, 59], "lib": [54, 59], "python3": [54, 59], "site": [54, 59], "packag": [54, 59], "distutil": 54, "system_info": 54, "py": [54, 59], "2159": 54, "userwarn": 54, "vendor": 54, "bla": [54, 59], "netlib": 54, "wors": 54, "easili": [54, 55, 56, 57], "gain": 54, "switch": [54, 56], "_calc_info": 54, "predict": [54, 56, 58], "sampling_fn": 54, "516455713134264": 54, "1609480326942554": 54, "care": 54, "graph": [54, 56], "aeppl": [54, 56], "joint_logprob": [54, 56], "y_vv": [54, 56], "clone": 54, "i_vv": 54, "as_tensor": 54, "int64": [54, 56], "component_logdens": 54, "realiz": 54, "total_logdens": 54, "logsumexp": 54, "15039347": 54, "written": [54, 58, 62], "grad": [54, 56], "try": [54, 55], "luckili": 54, "total_logdensity_grad": 54, "44711512": 54, "abl": [54, 55, 56], "wrapper": 54, "hcb": 54, "numba_logpdf": 54, "arg": 54, "result_shap": 54, "call_grad": 54, "vjp_fwd": 54, "vjp_bwd": 54, "y_bar": 54, "defvjp": 54, "complain": [54, 56], "1503935": 54, "init_kei": [54, 56, 57, 58], "nuts_kei": [54, 62], "5826156": 54, "9966314": 54, "32264766": 54, "notic": 54, "quit": 54, "slowli": 54, "lot": 54, "forth": 54, "execut": [54, 55], "100_000": 54, "76": 54, "04": 54, "1_000": [54, 55, 61, 62], "128": 54, "overhead": 54, "workload": 54, "spent": 54, "becom": [54, 55], "irrelev": 54, "callback": 54, "bit": 54, "scope": [54, 58], "tutori": [54, 55, 62], "inspir": 54, "triton": 54, "regist": 54, "few": 55, "ubiquit": 55, "trend": 55, "emerg": 55, "thousand": 55, "whatev": 55, "thank": 55, "gpu": 55, "tpu": [55, 62], "best": [55, 56], "simd": 55, "rememb": 55, "instruct": [55, 56, 57, 59, 60], "intruct": 55, "unexpect": 55, "consequ": 55, "higher": 55, "abstract": 55, "core": [55, 58], "walkthrough": 55, "invit": 55, "read": 55, "regress": 55, "inher": 55, "limit": 55, "observ": [55, 56, 59, 60, 61, 62], "log_scal": [55, 61], "inference_loop": [55, 56, 57, 58, 59, 60, 61], "dramat": 55, "poorli": 55, "inv_mass_matrix": [55, 61], "put": 55, "foot": 55, "multiprocess": 55, "cpu_count": 55, "newcom": 55, "immedi": 55, "benefit": 55, "awesom": 55, "insid": 55, "inference_loop_multiple_chain": 55, "prepar": 55, "again": 55, "in_ax": 55, "2_000": 55, "block_until_readi": [55, 57, 61], "27": [55, 56], "obviou": 55, "said": 55, "1024": 55, "wait": 55, "slowest": 55, "think": 55, "unfortun": 55, "although": 55, "grant": 55, "led": 55, "synchron": 55, "truli": 55, "help": 55, "prove": 55, "treat": 55, "regardless": 55, "avail": 55, "topic": 55, "xla_flag": 55, "environment": 55, "o": [55, 56], "environ": 55, "xla_force_host_platform_device_count": 55, "format": 55, "advis": 55, "confirm": 55, "successfuli": 55, "command": 55, "ask": 55, "snippet": 55, "fn": 55, "parallel_fn": 55, "logic": 55, "xla": [55, 62], "num_replica": 55, "question": 55, "workaround": 55, "involv": 55, "nest": [55, 58], "against": 55, "count": 55, "side": 55, "static_broadcasted_argnum": 55, "wouldn": 55, "prefer": 55, "highlight": 55, "pmap_stat": 55, "97": 55, "wow": 55, "intuit": 55, "transpos": 55, "2000": [55, 57], "minut": 55, "saw": 55, "mention": 55, "someth": 55, "design": [55, 61, 62], "altogeth": 55, "exhibit": 55, "behind": 55, "activ": [55, 58], "area": 55, "languag": [56, 57, 58, 59, 60], "pleas": [56, 57, 59, 60], "instal": [56, 57, 59, 60], "repositori": [56, 57, 59, 60], "binomi": 56, "respons": 56, "rat": 56, "tumor": 56, "peopl": [56, 62], "test": [56, 61, 62], "group_siz": 56, "19": 56, "24": 56, "46": 56, "47": 56, "48": 56, "positv": 56, "n_of_posit": 56, "12": [56, 57, 59, 60], "n_rat_tumor": 56, "improp": 56, "a_vv": 56, "b_vv": 56, "logprior": 56, "theta_rv": 56, "prior_predictive_fn": 56, "35": 56, "41": 56, "44": 56, "a_rv": 56, "b_rv": 56, "loglikelihood": 56, "theta_vv": 56, "logprob": 56, "unbound": 56, "odd": 56, "transformvaluesrewrit": 56, "logoddstransform": 56, "transforms_op": 56, "extra_rewrit": 56, "aemcmc": 56, "dprint": 56, "elemwis": 56, "no_inplac": 56, "id": 56, "mul": 56, "tensorconst": 56, "acc_dtyp": 56, "float64": 56, "makevector": 56, "and_": 56, "le": 56, "inplacedimshuffl": 56, "tensortyp": 56, "71": 56, "gammaln": 56, "ba": 56, "bb": 56, "bc": 56, "bd": 56, "BE": 56, "bf": 56, "bg": 56, "bh": 56, "bi": 56, "bj": 56, "bk": 56, "eq": 56, "bl": 56, "sigmoid": 56, "bm": 56, "val_clon": 56, "tran": 56, "bn": 56, "bo": 56, "bp": 56, "bq": 56, "br": 56, "bt": 56, "bu": 56, "bv": 56, "bw": 56, "inf": 56, "bx": 56, "BY": 56, "bz": 56, "ca": 56, "cb": 56, "cc": 56, "cd": 56, "ce": 56, "cf": 56, "cg": 56, "ch": 56, "ci": 56, "cj": 56, "ck": 56, "cl": 56, "cm": 56, "cn": 56, "co": 56, "cp": 56, "cq": 56, "cr": 56, "ct": 56, "cu": 56, "cv": 56, "cw": 56, "cx": 56, "cy": 56, "cz": 56, "da": 56, "db": 56, "dc": 56, "dd": 56, "de": 56, "dg": 56, "ge": 56, "di": 56, "dj": 56, "dk": 56, "dl": 56, "dm": 56, "dn": 56, "dp": 56, "dq": 56, "dr": 56, "dt": 56, "dv": 56, "dw": 56, "dz": 56, "ea": 56, "eb": 56, "ec": 56, "ed": 56, "ee": 56, "ef": 56, "eg": 56, "eh": 56, "ei": 56, "ej": 56, "log1p": 56, "ek": 56, "el": 56, "em": 56, "en": [56, 59], "eo": 56, "er": 56, "eu": 56, "ev": 56, "gt": 56, "ew": 56, "ex": 56, "ez": 56, "fa": 56, "fb": 56, "fc": 56, "fd": 56, "fe": 56, "ff": 56, "ipykernel": 56, "iostream": 56, "outstream": 56, "0x7f58b96dc2e0": 56, "backend": 56, "logprob_jax": 56, "vm": 56, "jit_fn": 56, "wrap": 56, "life": 56, "flat_posit": 56, "go": [56, 62], "init_param_fn": 56, "seed": 56, "key1": 56, "key2": 56, "key3": 56, "minval": 56, "maxval": 56, "init_posit": [56, 59], "n_adapt": 56, "3000": 56, "n_sampl": 56, "warmup_kei": [56, 57, 58, 59, 60, 61], "plot_trac": [56, 57, 59, 60], "togeth": [57, 58], "reproduc": [57, 58, 59, 60], "eight": [57, 59, 60], "school": [57, 59, 60], "credit": 57, "team": 57, "hierarch": [57, 59, 60], "dist": 57, "reparam": 57, "transformreparam": 57, "eight_schools_noncent": 57, "halfcauchi": [57, 59], "plate": 57, "handler": 57, "config": 57, "transformeddistribut": 57, "affinetransform": 57, "ob": [57, 59], "initialize_model": 57, "init_param": 57, "potential_fn_gen": 57, "model_arg": 57, "dynamic_arg": 57, "num_warmup": 57, "num_diverg": 57, "2f": [57, 60], "87": 57, "00": 57, "plot": [57, 60, 61], "treatment": [57, 60], "plot_posterior": 57, "var_nam": [57, 60], "theta_bas": 57, "compact": [57, 60], "06": 57, "26": 57, "nativ": 58, "network": 58, "bnn": 58, "iri": 58, "sklearn": 58, "load_iri": 58, "featur": 58, "num_featur": 58, "num_class": 58, "target_nam": 58, "150": 58, "aesara": 58, "layer": 58, "random_vari": 58, "ppl": [58, 62], "tensorflow_prob": [58, 60], "substrat": [58, 60], "tfd": [58, 60], "dim_out": 58, "nn": 58, "relu": 58, "dim_in": 58, "w_kei": 58, "b_kei": 58, "sample_shap": 58, "perceptron": 58, "tag": 58, "collis": 58, "mlp": 58, "hidden_s": 58, "num_hidden": 58, "subkei": 58, "enumer": 58, "layer_": 58, "logit": 58, "mlp_kei": 58, "label_kei": 58, "joint_sampl": 58, "initial_weight": 58, "dict_kei": 58, "layer_1": 58, "layer_2": 58, "layer_3": 58, "num_paramet": 58, "tree_util": 58, "tree_flatten": 58, "2953": 58, "joint_log_prob": 58, "142": 58, "accuraci": 58, "interven": 58, "inject": 58, "posterior_weight": 58, "pred_kei": 58, "output_logit": 58, "output_prob": 58, "softmax": 58, "argmax": 58, "bma": 58, "9794667": 58, "9866667": 58, "descript": [59, 60], "pm": 59, "theta_1": 59, "warn": 59, "pytensor": 59, "get_jaxified_logp": 59, "sampling_jax": 59, "rv": 59, "value_var": 59, "tqdm": 59, "auto": 59, "21": 59, "tqdmwarn": 59, "iprogress": 59, "jupyt": 59, "ipywidget": 59, "readthedoc": 59, "io": 59, "user_instal": 59, "autonotebook": 59, "notebook_tqdm": 59, "init_position_dict": 59, "initial_point": 59, "50_000": [59, 60], "num_school": 60, "treatment_effect": 60, "treatment_stddev": 60, "se": 60, "jdc": 60, "jointdistributioncoroutineautobatch": 60, "avg_effect": 60, "log_tau": 60, "avg_stddev": 60, "theta_prim": 60, "school_effects_standard": 60, "yhat": 60, "pinned_model": 60, "experimental_pin": 60, "unnormalized_log_prob": 60, "92": 60, "dictionnari": 60, "school_effects_sampl": 60, "newaxi": 60, "correpond": 60, "showcas": 61, "recov": 61, "unconstrain": 61, "decent": 61, "000": 61, "34004": 61, "82": 61, "1183": 61, "2325": 61, "63171": 61, "773": 61, "hmc_kernel": 61, "mcmc_sampl": 61, "53": 61, "fig": 61, "ax1": 61, "subplot": 61, "ncol": 61, "figsiz": 61, "set_xlabel": 61, "set_ylabel": 61, "4_000": 61, "69": 61, "68": 61, "31": 61, "main": 62, "branch": 62, "latest": 62, "releas": 62, "art": 62, "robust": 62, "realli": 62, "menu": 62, "pip": 62, "nightli": 62, "conda": 62, "forg": 62, "By": 62, "relev": 62, "support": 62}, "objects": {"": [[11, 0, 0, "-", "blackjax"]], "blackjax": [[0, 0, 0, "-", "_version"], [3, 0, 0, "-", "adaptation"], [9, 0, 0, "-", "base"], [10, 0, 0, "-", "diagnostics"], [16, 0, 0, "-", "mcmc"], [28, 0, 0, "-", "optimizers"], [30, 0, 0, "-", "progress_bar"], [34, 0, 0, "-", "sgmcmc"], [41, 0, 0, "-", "smc"], [45, 0, 0, "-", "types"], [46, 0, 0, "-", "util"], [47, 0, 0, "-", "vi"]], "blackjax._version": [[0, 1, 1, "", "TYPE_CHECKING"], [0, 1, 1, "", "VERSION_TUPLE"], [0, 1, 1, "", "version"], [0, 1, 1, "", "version_tuple"]], "blackjax.adaptation": [[1, 0, 0, "-", "base"], [2, 0, 0, "-", "chees_adaptation"], [4, 0, 0, "-", "mass_matrix"], [5, 0, 0, "-", "meads_adaptation"], [6, 0, 0, "-", "pathfinder_adaptation"], [7, 0, 0, "-", "step_size"], [8, 0, 0, "-", "window_adaptation"]], "blackjax.adaptation.base": [[1, 2, 1, "", "AdaptationInfo"], [1, 2, 1, "", "AdaptationResults"]], "blackjax.adaptation.base.AdaptationInfo": [[1, 3, 1, "", "adaptation_state"], [1, 3, 1, "", "info"], [1, 3, 1, "", "state"]], "blackjax.adaptation.base.AdaptationResults": [[1, 3, 1, "", "parameters"], [1, 3, 1, "", "state"]], "blackjax.adaptation.chees_adaptation": [[2, 2, 1, "", "ChEESAdaptationState"], [2, 1, 1, "", "OPTIMAL_TARGET_ACCEPTANCE_RATE"], [2, 4, 1, "", "base"], [2, 4, 1, "", "chees_adaptation"]], "blackjax.adaptation.chees_adaptation.ChEESAdaptationState": [[2, 3, 1, "", "da_state"], [2, 3, 1, "", "log_step_size_moving_average"], [2, 3, 1, "", "log_trajectory_length_moving_average"], [2, 3, 1, "", "optim_state"], [2, 3, 1, "", "random_generator_arg"], [2, 3, 1, "", "step"], [2, 3, 1, "", "step_size"], [2, 3, 1, "", "trajectory_length"]], "blackjax.adaptation.mass_matrix": [[4, 2, 1, "", "MassMatrixAdaptationState"], [4, 2, 1, "", "WelfordAlgorithmState"], [4, 4, 1, "", "mass_matrix_adaptation"], [4, 4, 1, "", "welford_algorithm"]], "blackjax.adaptation.mass_matrix.MassMatrixAdaptationState": [[4, 3, 1, "", "inverse_mass_matrix"], [4, 3, 1, "", "wc_state"]], "blackjax.adaptation.mass_matrix.WelfordAlgorithmState": [[4, 3, 1, "", "m2"], [4, 3, 1, "", "mean"], [4, 3, 1, "", "sample_size"]], "blackjax.adaptation.meads_adaptation": [[5, 2, 1, "", "MEADSAdaptationState"], [5, 4, 1, "", "base"], [5, 4, 1, "", "maximum_eigenvalue"], [5, 4, 1, "", "meads_adaptation"]], "blackjax.adaptation.meads_adaptation.MEADSAdaptationState": [[5, 3, 1, "", "alpha"], [5, 3, 1, "", "current_iteration"], [5, 3, 1, "", "delta"], [5, 3, 1, "", "position_sigma"], [5, 3, 1, "", "step_size"]], "blackjax.adaptation.pathfinder_adaptation": [[6, 2, 1, "", "PathfinderAdaptationState"], [6, 4, 1, "", "base"], [6, 4, 1, "", "pathfinder_adaptation"]], "blackjax.adaptation.pathfinder_adaptation.PathfinderAdaptationState": [[6, 3, 1, "", "inverse_mass_matrix"], [6, 3, 1, "", "ss_state"], [6, 3, 1, "", "step_size"]], "blackjax.adaptation.step_size": [[7, 2, 1, "", "DualAveragingAdaptationState"], [7, 4, 1, "", "dual_averaging_adaptation"], [7, 4, 1, "", "find_reasonable_step_size"]], "blackjax.adaptation.step_size.DualAveragingAdaptationState": [[7, 3, 1, "", "avg_error"], [7, 3, 1, "", "log_step_size"], [7, 3, 1, "", "log_step_size_avg"], [7, 3, 1, "", "mu"], [7, 3, 1, "", "step"]], "blackjax.adaptation.window_adaptation": [[8, 2, 1, "", "WindowAdaptationState"], [8, 4, 1, "", "base"], [8, 4, 1, "", "build_schedule"], [8, 4, 1, "", "window_adaptation"]], "blackjax.adaptation.window_adaptation.WindowAdaptationState": [[8, 3, 1, "", "imm_state"], [8, 3, 1, "", "inverse_mass_matrix"], [8, 3, 1, "", "ss_state"], [8, 3, 1, "", "step_size"]], "blackjax.base": [[9, 2, 1, "", "AdaptationAlgorithm"], [9, 1, 1, "", "Info"], [9, 2, 1, "", "InitFn"], [9, 1, 1, "", "Position"], [9, 2, 1, "", "RunFn"], [9, 2, 1, "", "SamplingAlgorithm"], [9, 1, 1, "", "State"], [9, 2, 1, "", "UpdateFn"], [9, 2, 1, "", "VIAlgorithm"]], "blackjax.base.AdaptationAlgorithm": [[9, 3, 1, "", "run"]], "blackjax.base.SamplingAlgorithm": [[9, 3, 1, "", "init"], [9, 3, 1, "", "step"]], "blackjax.base.VIAlgorithm": [[9, 3, 1, "", "init"], [9, 3, 1, "", "sample"], [9, 3, 1, "", "step"]], "blackjax.diagnostics": [[10, 4, 1, "", "effective_sample_size"], [10, 4, 1, "", "potential_scale_reduction"]], "blackjax.mcmc": [[12, 0, 0, "-", "diffusions"], [13, 0, 0, "-", "elliptical_slice"], [14, 0, 0, "-", "ghmc"], [15, 0, 0, "-", "hmc"], [17, 0, 0, "-", "integrators"], [18, 0, 0, "-", "mala"], [19, 0, 0, "-", "marginal_latent_gaussian"], [20, 0, 0, "-", "metrics"], [21, 0, 0, "-", "nuts"], [22, 0, 0, "-", "periodic_orbital"], [23, 0, 0, "-", "proposal"], [24, 0, 0, "-", "random_walk"], [25, 0, 0, "-", "termination"], [26, 0, 0, "-", "trajectory"]], "blackjax.mcmc.diffusions": [[12, 4, 1, "", "overdamped_langevin"]], "blackjax.mcmc.elliptical_slice": [[13, 2, 1, "", "EllipSliceInfo"], [13, 2, 1, "", "EllipSliceState"], [13, 4, 1, "", "build_kernel"], [13, 2, 1, "", "elliptical_slice"], [13, 4, 1, "", "init"]], "blackjax.mcmc.elliptical_slice.EllipSliceInfo": [[13, 3, 1, "", "momentum"], [13, 3, 1, "", "subiter"], [13, 3, 1, "", "theta"]], "blackjax.mcmc.elliptical_slice.EllipSliceState": [[13, 3, 1, "", "logdensity"], [13, 3, 1, "", "position"]], "blackjax.mcmc.elliptical_slice.elliptical_slice": [[13, 3, 1, "", "build_kernel"], [13, 3, 1, "", "init"]], "blackjax.mcmc.ghmc": [[14, 2, 1, "", "GHMCState"], [14, 4, 1, "", "build_kernel"], [14, 2, 1, "", "ghmc"], [14, 4, 1, "", "init"]], "blackjax.mcmc.ghmc.GHMCState": [[14, 3, 1, "", "logdensity"], [14, 3, 1, "", "logdensity_grad"], [14, 3, 1, "", "momentum"], [14, 3, 1, "", "position"], [14, 3, 1, "", "slice"]], "blackjax.mcmc.ghmc.ghmc": [[14, 3, 1, "", "build_kernel"], [14, 3, 1, "", "init"]], "blackjax.mcmc.hmc": [[15, 2, 1, "", "DynamicHMCState"], [15, 2, 1, "", "HMCInfo"], [15, 2, 1, "", "HMCState"], [15, 4, 1, "", "build_dynamic_kernel"], [15, 4, 1, "", "build_kernel"], [15, 2, 1, "", "dynamic_hmc"], [15, 2, 1, "", "hmc"], [15, 4, 1, "", "init"], [15, 4, 1, "", "init_dynamic"]], "blackjax.mcmc.hmc.DynamicHMCState": [[15, 3, 1, "", "logdensity"], [15, 3, 1, "", "logdensity_grad"], [15, 3, 1, "", "position"], [15, 3, 1, "", "random_generator_arg"]], "blackjax.mcmc.hmc.HMCInfo": [[15, 3, 1, "", "acceptance_rate"], [15, 3, 1, "", "energy"], [15, 3, 1, "", "is_accepted"], [15, 3, 1, "", "is_divergent"], [15, 3, 1, "", "momentum"], [15, 3, 1, "", "num_integration_steps"], [15, 3, 1, "", "proposal"]], "blackjax.mcmc.hmc.HMCState": [[15, 3, 1, "", "logdensity"], [15, 3, 1, "", "logdensity_grad"], [15, 3, 1, "", "position"]], "blackjax.mcmc.hmc.dynamic_hmc": [[15, 3, 1, "", "build_kernel"], [15, 3, 1, "", "init"]], "blackjax.mcmc.hmc.hmc": [[15, 3, 1, "", "build_kernel"], [15, 3, 1, "", "init"]], "blackjax.mcmc.integrators": [[17, 1, 1, "", "mclachlan"], [17, 1, 1, "", "noneuclidean_leapfrog"], [17, 1, 1, "", "noneuclidean_mclachlan"], [17, 1, 1, "", "noneuclidean_yoshida"], [17, 1, 1, "", "velocity_verlet"], [17, 1, 1, "", "yoshida"]], "blackjax.mcmc.mala": [[18, 2, 1, "", "MALAInfo"], [18, 2, 1, "", "MALAState"], [18, 4, 1, "", "build_kernel"], [18, 4, 1, "", "init"], [18, 2, 1, "", "mala"]], "blackjax.mcmc.mala.MALAInfo": [[18, 3, 1, "", "acceptance_rate"], [18, 3, 1, "", "is_accepted"]], "blackjax.mcmc.mala.MALAState": [[18, 3, 1, "", "logdensity"], [18, 3, 1, "", "logdensity_grad"], [18, 3, 1, "", "position"]], "blackjax.mcmc.mala.mala": [[18, 3, 1, "", "build_kernel"], [18, 3, 1, "", "init"]], "blackjax.mcmc.marginal_latent_gaussian": [[19, 2, 1, "", "MarginalInfo"], [19, 2, 1, "", "MarginalState"], [19, 4, 1, "", "init_and_kernel"], [19, 2, 1, "", "mgrad_gaussian"]], "blackjax.mcmc.marginal_latent_gaussian.MarginalInfo": [[19, 3, 1, "", "acceptance_rate"], [19, 3, 1, "", "is_accepted"], [19, 3, 1, "", "proposal"]], "blackjax.mcmc.marginal_latent_gaussian.MarginalState": [[19, 3, 1, "", "U_grad_x"], [19, 3, 1, "", "U_x"], [19, 3, 1, "", "logdensity"], [19, 3, 1, "", "logdensity_grad"], [19, 3, 1, "", "position"]], "blackjax.mcmc.metrics": [[20, 4, 1, "", "gaussian_euclidean"]], "blackjax.mcmc.nuts": [[21, 2, 1, "", "NUTSInfo"], [21, 4, 1, "", "build_kernel"], [21, 1, 1, "", "init"], [21, 2, 1, "", "nuts"]], "blackjax.mcmc.nuts.NUTSInfo": [[21, 3, 1, "", "acceptance_rate"], [21, 3, 1, "", "energy"], [21, 3, 1, "", "is_divergent"], [21, 3, 1, "", "is_turning"], [21, 3, 1, "", "momentum"], [21, 3, 1, "", "num_integration_steps"], [21, 3, 1, "", "num_trajectory_expansions"], [21, 3, 1, "", "trajectory_leftmost_state"], [21, 3, 1, "", "trajectory_rightmost_state"]], "blackjax.mcmc.nuts.nuts": [[21, 3, 1, "", "build_kernel"], [21, 3, 1, "", "init"]], "blackjax.mcmc.periodic_orbital": [[22, 2, 1, "", "PeriodicOrbitalState"], [22, 4, 1, "", "build_kernel"], [22, 4, 1, "", "init"], [22, 2, 1, "", "orbital_hmc"]], "blackjax.mcmc.periodic_orbital.PeriodicOrbitalState": [[22, 3, 1, "", "directions"], [22, 3, 1, "", "logdensities"], [22, 3, 1, "", "logdensities_grad"], [22, 3, 1, "", "positions"], [22, 3, 1, "", "weights"]], "blackjax.mcmc.periodic_orbital.orbital_hmc": [[22, 3, 1, "", "build_kernel"], [22, 3, 1, "", "init"]], "blackjax.mcmc.proposal": [[23, 2, 1, "", "Proposal"], [23, 2, 1, "", "SampleProposal"], [23, 1, 1, "", "TrajectoryState"], [23, 4, 1, "", "asymmetric_proposal_generator"], [23, 4, 1, "", "nonreversible_slice_sampling"], [23, 4, 1, "", "progressive_biased_sampling"], [23, 4, 1, "", "progressive_uniform_sampling"], [23, 4, 1, "", "proposal_from_energy_diff"], [23, 4, 1, "", "proposal_generator"], [23, 4, 1, "", "static_binomial_sampling"]], "blackjax.mcmc.proposal.Proposal": [[23, 3, 1, "", "energy"], [23, 3, 1, "", "state"], [23, 3, 1, "", "sum_log_p_accept"], [23, 3, 1, "", "weight"]], "blackjax.mcmc.random_walk": [[24, 2, 1, "", "RWInfo"], [24, 2, 1, "", "RWState"], [24, 2, 1, "", "additive_step_random_walk"], [24, 4, 1, "", "build_additive_step"], [24, 4, 1, "", "build_irmh"], [24, 4, 1, "", "build_rmh"], [24, 4, 1, "", "build_rmh_transition_energy"], [24, 2, 1, "", "irmh"], [24, 4, 1, "", "normal"], [24, 2, 1, "", "rmh"], [24, 4, 1, "", "rmh_proposal"]], "blackjax.mcmc.random_walk.RWInfo": [[24, 3, 1, "", "acceptance_rate"], [24, 3, 1, "", "is_accepted"], [24, 3, 1, "", "proposal"]], "blackjax.mcmc.random_walk.RWState": [[24, 3, 1, "", "logdensity"], [24, 3, 1, "", "position"]], "blackjax.mcmc.random_walk.additive_step_random_walk": [[24, 3, 1, "", "build_kernel"], [24, 3, 1, "", "init"], [24, 5, 1, "", "normal_random_walk"]], "blackjax.mcmc.random_walk.irmh": [[24, 3, 1, "", "build_kernel"], [24, 3, 1, "", "init"]], "blackjax.mcmc.random_walk.rmh": [[24, 3, 1, "", "build_kernel"], [24, 3, 1, "", "init"]], "blackjax.mcmc.termination": [[25, 2, 1, "", "IterativeUTurnState"], [25, 4, 1, "", "iterative_uturn_numpyro"]], "blackjax.mcmc.termination.IterativeUTurnState": [[25, 3, 1, "", "idx_max"], [25, 3, 1, "", "idx_min"], [25, 3, 1, "", "momentum"], [25, 3, 1, "", "momentum_sum"]], "blackjax.mcmc.trajectory": [[26, 2, 1, "", "DynamicExpansionState"], [26, 2, 1, "", "DynamicIntegrationState"], [26, 2, 1, "", "Trajectory"], [26, 4, 1, "", "append_to_trajectory"], [26, 4, 1, "", "dynamic_multiplicative_expansion"], [26, 4, 1, "", "dynamic_progressive_integration"], [26, 4, 1, "", "dynamic_recursive_integration"], [26, 4, 1, "", "hmc_energy"], [26, 4, 1, "", "merge_trajectories"], [26, 4, 1, "", "reorder_trajectories"], [26, 4, 1, "", "static_integration"]], "blackjax.mcmc.trajectory.DynamicExpansionState": [[26, 3, 1, "", "proposal"], [26, 3, 1, "", "step"], [26, 3, 1, "", "termination_state"], [26, 3, 1, "", "trajectory"]], "blackjax.mcmc.trajectory.DynamicIntegrationState": [[26, 3, 1, "", "proposal"], [26, 3, 1, "", "step"], [26, 3, 1, "", "termination_state"], [26, 3, 1, "", "trajectory"]], "blackjax.mcmc.trajectory.Trajectory": [[26, 3, 1, "", "leftmost_state"], [26, 3, 1, "", "momentum_sum"], [26, 3, 1, "", "num_states"], [26, 3, 1, "", "rightmost_state"]], "blackjax.optimizers": [[27, 0, 0, "-", "dual_averaging"], [29, 0, 0, "-", "lbfgs"]], "blackjax.optimizers.dual_averaging": [[27, 2, 1, "", "DualAveragingState"], [27, 4, 1, "", "dual_averaging"]], "blackjax.optimizers.dual_averaging.DualAveragingState": [[27, 3, 1, "", "avg_error"], [27, 3, 1, "", "log_x"], [27, 3, 1, "", "log_x_avg"], [27, 3, 1, "", "mu"], [27, 3, 1, "", "step"]], "blackjax.optimizers.lbfgs": [[29, 2, 1, "", "LBFGSHistory"], [29, 4, 1, "", "bfgs_sample"], [29, 4, 1, "", "lbfgs_inverse_hessian_factors"], [29, 4, 1, "", "lbfgs_inverse_hessian_formula_1"], [29, 4, 1, "", "lbfgs_inverse_hessian_formula_2"], [29, 4, 1, "", "minimize_lbfgs"]], "blackjax.optimizers.lbfgs.LBFGSHistory": [[29, 3, 1, "", "alpha"], [29, 3, 1, "", "f"], [29, 3, 1, "", "g"], [29, 3, 1, "", "update_mask"], [29, 3, 1, "", "x"]], "blackjax.progress_bar": [[30, 4, 1, "", "progress_bar_scan"]], "blackjax.sgmcmc": [[31, 0, 0, "-", "csgld"], [32, 0, 0, "-", "diffusions"], [33, 0, 0, "-", "gradients"], [35, 0, 0, "-", "sghmc"], [36, 0, 0, "-", "sgld"], [37, 0, 0, "-", "sgnht"]], "blackjax.sgmcmc.csgld": [[31, 2, 1, "", "ContourSGLDState"], [31, 4, 1, "", "build_kernel"], [31, 2, 1, "", "csgld"], [31, 4, 1, "", "init"]], "blackjax.sgmcmc.csgld.ContourSGLDState": [[31, 3, 1, "", "energy_idx"], [31, 3, 1, "", "energy_pdf"], [31, 3, 1, "", "position"]], "blackjax.sgmcmc.csgld.csgld": [[31, 3, 1, "", "build_kernel"], [31, 3, 1, "", "init"]], "blackjax.sgmcmc.diffusions": [[32, 4, 1, "", "overdamped_langevin"], [32, 4, 1, "", "sghmc"], [32, 4, 1, "", "sgnht"]], "blackjax.sgmcmc.gradients": [[33, 4, 1, "", "control_variates"], [33, 4, 1, "", "grad_estimator"], [33, 4, 1, "", "logdensity_estimator"]], "blackjax.sgmcmc.sghmc": [[35, 4, 1, "", "build_kernel"], [35, 4, 1, "", "init"], [35, 2, 1, "", "sghmc"]], "blackjax.sgmcmc.sghmc.sghmc": [[35, 3, 1, "", "build_kernel"], [35, 3, 1, "", "init"]], "blackjax.sgmcmc.sgld": [[36, 4, 1, "", "build_kernel"], [36, 4, 1, "", "init"], [36, 2, 1, "", "sgld"]], "blackjax.sgmcmc.sgld.sgld": [[36, 3, 1, "", "build_kernel"], [36, 3, 1, "", "init"]], "blackjax.sgmcmc.sgnht": [[37, 2, 1, "", "SGNHTState"], [37, 4, 1, "", "build_kernel"], [37, 4, 1, "", "init"], [37, 2, 1, "", "sgnht"]], "blackjax.sgmcmc.sgnht.SGNHTState": [[37, 3, 1, "", "momentum"], [37, 3, 1, "", "position"], [37, 3, 1, "", "xi"]], "blackjax.sgmcmc.sgnht.sgnht": [[37, 3, 1, "", "build_kernel"], [37, 3, 1, "", "init"]], "blackjax.smc": [[38, 0, 0, "-", "adaptive_tempered"], [39, 0, 0, "-", "base"], [40, 0, 0, "-", "ess"], [42, 0, 0, "-", "resampling"], [43, 0, 0, "-", "solver"], [44, 0, 0, "-", "tempered"]], "blackjax.smc.adaptive_tempered": [[38, 2, 1, "", "adaptive_tempered_smc"], [38, 4, 1, "", "build_kernel"]], "blackjax.smc.adaptive_tempered.adaptive_tempered_smc": [[38, 3, 1, "", "build_kernel"], [38, 3, 1, "", "init"]], "blackjax.smc.base": [[39, 2, 1, "", "SMCInfo"], [39, 2, 1, "", "SMCState"], [39, 4, 1, "", "init"], [39, 4, 1, "", "step"]], "blackjax.smc.base.SMCInfo": [[39, 3, 1, "", "ancestors"], [39, 3, 1, "", "log_likelihood_increment"], [39, 3, 1, "", "update_info"]], "blackjax.smc.base.SMCState": [[39, 3, 1, "", "particles"], [39, 3, 1, "", "weights"]], "blackjax.smc.ess": [[40, 4, 1, "", "ess"], [40, 4, 1, "", "ess_solver"], [40, 4, 1, "", "log_ess"]], "blackjax.smc.resampling": [[42, 4, 1, "", "multinomial"], [42, 4, 1, "", "residual"], [42, 4, 1, "", "stratified"], [42, 4, 1, "", "systematic"]], "blackjax.smc.solver": [[43, 4, 1, "", "dichotomy"]], "blackjax.smc.tempered": [[44, 2, 1, "", "TemperedSMCState"], [44, 4, 1, "", "build_kernel"], [44, 4, 1, "", "init"]], "blackjax.smc.tempered.TemperedSMCState": [[44, 3, 1, "", "lmbda"], [44, 3, 1, "", "particles"], [44, 3, 1, "", "weights"]], "blackjax.types": [[45, 1, 1, "", "Array"], [45, 1, 1, "", "ArrayLikeTree"], [45, 1, 1, "", "ArrayTree"], [45, 1, 1, "", "PRNGKey"]], "blackjax.util": [[46, 4, 1, "", "generate_gaussian_noise"], [46, 4, 1, "", "generate_unit_vector"], [46, 4, 1, "", "index_pytree"], [46, 4, 1, "", "linear_map"], [46, 4, 1, "", "pytree_size"]], "blackjax.vi": [[48, 0, 0, "-", "meanfield_vi"], [49, 0, 0, "-", "pathfinder"], [50, 0, 0, "-", "svgd"]], "blackjax.vi.meanfield_vi": [[48, 2, 1, "", "MFVIInfo"], [48, 2, 1, "", "MFVIState"], [48, 4, 1, "", "generate_meanfield_logdensity"], [48, 2, 1, "", "meanfield_vi"], [48, 4, 1, "", "sample"], [48, 4, 1, "", "step"]], "blackjax.vi.meanfield_vi.MFVIInfo": [[48, 3, 1, "", "elbo"]], "blackjax.vi.meanfield_vi.MFVIState": [[48, 3, 1, "", "mu"], [48, 3, 1, "", "opt_state"], [48, 3, 1, "", "rho"]], "blackjax.vi.meanfield_vi.meanfield_vi": [[48, 3, 1, "", "init"], [48, 3, 1, "", "sample"], [48, 3, 1, "", "step"]], "blackjax.vi.pathfinder": [[49, 2, 1, "", "PathfinderState"], [49, 4, 1, "", "approximate"], [49, 2, 1, "", "pathfinder"], [49, 4, 1, "", "sample"]], "blackjax.vi.pathfinder.PathfinderState": [[49, 3, 1, "", "alpha"], [49, 3, 1, "", "beta"], [49, 3, 1, "", "elbo"], [49, 3, 1, "", "gamma"], [49, 3, 1, "", "grad_position"], [49, 3, 1, "", "position"]], "blackjax.vi.pathfinder.pathfinder": [[49, 3, 1, "", "approximate"], [49, 3, 1, "", "sample"]], "blackjax.vi.svgd": [[50, 4, 1, "", "rbf_kernel"], [50, 2, 1, "", "svgd"], [50, 4, 1, "", "update_median_heuristic"]], "blackjax.vi.svgd.svgd": [[50, 3, 1, "", "build_kernel"], [50, 3, 1, "", "init"]]}, "objtypes": {"0": "py:module", "1": "py:data", "2": "py:class", "3": "py:attribute", "4": "py:function", "5": "py:method"}, "objnames": {"0": ["py", "module", "Python module"], "1": ["py", "data", "Python data"], "2": ["py", "class", "Python class"], "3": ["py", "attribute", "Python attribute"], "4": ["py", "function", "Python function"], "5": ["py", "method", "Python method"]}, "titleterms": {"blackjax": [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 52, 53, 61, 62], "_version": 0, "modul": [0, 1, 2, 4, 5, 6, 7, 8, 9, 10, 12, 13, 14, 15, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 29, 30, 31, 32, 33, 35, 36, 37, 38, 39, 40, 42, 43, 44, 45, 46, 48, 49, 50], "content": [0, 1, 2, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 42, 43, 44, 45, 46, 48, 49, 50], "adapt": [1, 2, 3, 4, 5, 6, 7, 8, 61], "base": [1, 9, 39], "class": [1, 2, 4, 5, 6, 7, 8, 9, 11, 13, 14, 15, 18, 19, 21, 22, 23, 24, 25, 26, 27, 29, 31, 35, 36, 37, 38, 39, 44, 48, 49, 50], "chees_adapt": 2, "function": [2, 4, 5, 6, 7, 8, 10, 11, 12, 13, 14, 15, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 42, 43, 44, 46, 48, 49, 50, 52, 54], "attribut": [2, 9, 11, 21, 23], "submodul": [3, 11, 16, 28, 34, 41, 47], "mass_matrix": 4, "meads_adapt": 5, "pathfinder_adapt": 6, "step_siz": 7, "window_adapt": 8, "diagnost": 10, "subpackag": 11, "packag": [11, 34], "mcmc": [12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26], "diffus": [12, 32], "elliptical_slic": 13, "ghmc": 14, "hmc": [15, 61], "integr": 17, "mala": 18, "marginal_latent_gaussian": 19, "metric": 20, "nut": [21, 55, 61], "periodic_orbit": 22, "propos": 23, "random_walk": 24, "paramet": [24, 53, 61], "termin": 25, "trajectori": 26, "optim": [27, 28, 29], "dual_averag": 27, "lbfg": 29, "progress_bar": 30, "sgmcmc": [31, 32, 33, 34, 35, 36, 37], "csgld": 31, "gradient": [33, 52], "sghmc": 35, "sgld": 36, "sgnht": 37, "smc": [38, 39, 40, 41, 42, 43, 44], "adaptive_temp": 38, "ess": 40, "resampl": 42, "solver": 43, "temper": 44, "type": 45, "util": 46, "vi": [47, 48, 49, 50], "meanfield_vi": 48, "pathfind": 49, "svgd": 50, "bibliographi": 51, "us": [52, 54, 55, 56, 57, 58, 59, 60, 61], "custom": [52, 54], "defin": 52, "minimum": 52, "anoth": 52, "try": 52, "different": 52, "jax": [52, 54, 55], "grad": 52, "deriv": 52, "mathemat": 52, "tell": 52, "make": 52, "sure": 52, "result": [52, 53], "i": [52, 54], "correct": 52, "how": 53, "build": [53, 61], "metropoli": 53, "within": 53, "gibb": 53, "sampler": [53, 61], "The": [53, 61], "model": [53, 54, 56, 57, 58, 59, 60], "mwg": 53, "sampl": [53, 55], "construct": 53, "kernel": [53, 61], "set": [53, 61], "initi": [53, 61], "state": [53, 61], "each": 53, "algorithm": 53, "loop": [53, 61], "gener": 53, "check": 53, "develop": 53, "note": [53, 55], "logdens": 54, "compat": 54, "": [54, 61], "primit": 54, "aesara": [54, 56], "compil": 54, "numba": 54, "experiment": 54, "host_callback": 54, "call": 54, "xla": 54, "faster": 54, "multipl": 55, "chain": 55, "parallel": 55, "vector": 55, "v": 55, "vmap": 55, "pmap": 55, "A": [55, 61], "cpu": 55, "choos": 55, "number": 55, "devic": 55, "back": 55, "our": 55, "exampl": 55, "conclus": 55, "befor": [56, 57, 59, 60], "you": [56, 57, 59, 60], "start": [56, 57, 59, 60], "numpyro": 57, "oryx": 58, "pymc": 59, "tfp": 60, "quick": 61, "introduct": 61, "problem": 61, "infer": 61, "stan": 61, "window": 61, "welcom": 62, "hello": 62, "world": 62, "instal": 62, "gpu": 62, "instruct": 62}, "envversion": {"sphinx.domains.c": 2, "sphinx.domains.changeset": 1, "sphinx.domains.citation": 1, "sphinx.domains.cpp": 8, "sphinx.domains.index": 1, "sphinx.domains.javascript": 2, "sphinx.domains.math": 2, "sphinx.domains.python": 3, "sphinx.domains.rst": 2, "sphinx.domains.std": 2, "sphinx.ext.intersphinx": 1, "sphinx.ext.viewcode": 1, "sphinxcontrib.bibtex": 9, "sphinx": 57}, "alltitles": {"blackjax._version": [[0, "module-blackjax._version"]], "Module Contents": [[0, "module-contents"], [1, "module-contents"], [2, "module-contents"], [4, "module-contents"], [5, "module-contents"], [6, "module-contents"], [7, "module-contents"], [8, "module-contents"], [9, "module-contents"], [10, "module-contents"], [12, "module-contents"], [13, "module-contents"], [14, "module-contents"], [15, "module-contents"], [17, "module-contents"], [18, "module-contents"], [19, "module-contents"], [20, "module-contents"], [21, "module-contents"], [22, "module-contents"], [23, "module-contents"], [24, "module-contents"], [25, "module-contents"], [26, "module-contents"], [27, "module-contents"], [29, "module-contents"], [30, "module-contents"], [31, "module-contents"], [32, "module-contents"], [33, "module-contents"], [35, "module-contents"], [36, "module-contents"], [37, "module-contents"], [38, "module-contents"], [39, "module-contents"], [40, "module-contents"], [42, "module-contents"], [43, "module-contents"], [44, "module-contents"], [45, "module-contents"], [46, "module-contents"], [48, "module-contents"], [49, "module-contents"], [50, "module-contents"]], "blackjax.adaptation.base": [[1, "module-blackjax.adaptation.base"]], "Classes": [[1, "classes"], [2, "classes"], [4, "classes"], [5, "classes"], [6, "classes"], [7, "classes"], [8, "classes"], [9, "classes"], [11, "classes"], [13, "classes"], [14, "classes"], [15, "classes"], [18, "classes"], [19, "classes"], [21, "classes"], [22, "classes"], [23, "classes"], [24, "classes"], [25, "classes"], [26, "classes"], [27, "classes"], [29, "classes"], [31, "classes"], [35, "classes"], [36, "classes"], [37, "classes"], [38, "classes"], [39, "classes"], [44, "classes"], [48, "classes"], [49, "classes"], [50, "classes"]], "blackjax.adaptation.chees_adaptation": [[2, "module-blackjax.adaptation.chees_adaptation"]], "Functions": [[2, "functions"], [4, "functions"], [5, "functions"], [6, "functions"], [7, "functions"], [8, "functions"], [10, "functions"], [11, "functions"], [12, "functions"], [13, "functions"], [14, "functions"], [15, "functions"], [18, "functions"], [19, "functions"], [20, "functions"], [21, "functions"], [22, "functions"], [23, "functions"], [24, "functions"], [25, "functions"], [26, "functions"], [27, "functions"], [29, "functions"], [30, "functions"], [31, "functions"], [32, "functions"], [33, "functions"], [34, "functions"], [35, "functions"], [36, "functions"], [37, "functions"], [38, "functions"], [39, "functions"], [40, "functions"], [42, "functions"], [43, "functions"], [44, "functions"], [46, "functions"], [48, "functions"], [49, "functions"], [50, "functions"]], "Attributes": [[2, "attributes"], [9, "attributes"], [11, "attributes"], [21, "attributes"], [23, "attributes"]], "blackjax.adaptation": [[3, "module-blackjax.adaptation"]], "Submodules": [[3, "submodules"], [11, "submodules"], [16, "submodules"], [28, "submodules"], [34, "submodules"], [41, "submodules"], [47, "submodules"]], "blackjax.adaptation.mass_matrix": [[4, "module-blackjax.adaptation.mass_matrix"]], "blackjax.adaptation.meads_adaptation": [[5, "module-blackjax.adaptation.meads_adaptation"]], "blackjax.adaptation.pathfinder_adaptation": [[6, "module-blackjax.adaptation.pathfinder_adaptation"]], "blackjax.adaptation.step_size": [[7, "module-blackjax.adaptation.step_size"]], "blackjax.adaptation.window_adaptation": [[8, "module-blackjax.adaptation.window_adaptation"]], "blackjax.base": [[9, "module-blackjax.base"]], "blackjax.diagnostics": [[10, "module-blackjax.diagnostics"]], "blackjax": [[11, "module-blackjax"]], "Subpackages": [[11, "subpackages"]], "Package Contents": [[11, "package-contents"], [34, "package-contents"]], "blackjax.mcmc.diffusions": [[12, "module-blackjax.mcmc.diffusions"]], "blackjax.mcmc.elliptical_slice": [[13, "module-blackjax.mcmc.elliptical_slice"]], "blackjax.mcmc.ghmc": [[14, "module-blackjax.mcmc.ghmc"]], "blackjax.mcmc.hmc": [[15, "module-blackjax.mcmc.hmc"]], "blackjax.mcmc": [[16, "module-blackjax.mcmc"]], "blackjax.mcmc.integrators": [[17, "module-blackjax.mcmc.integrators"]], "blackjax.mcmc.mala": [[18, "module-blackjax.mcmc.mala"]], "blackjax.mcmc.marginal_latent_gaussian": [[19, "module-blackjax.mcmc.marginal_latent_gaussian"]], "blackjax.mcmc.metrics": [[20, "module-blackjax.mcmc.metrics"]], "blackjax.mcmc.nuts": [[21, "module-blackjax.mcmc.nuts"]], "blackjax.mcmc.periodic_orbital": [[22, "module-blackjax.mcmc.periodic_orbital"]], "blackjax.mcmc.proposal": [[23, "module-blackjax.mcmc.proposal"]], "blackjax.mcmc.random_walk": [[24, "module-blackjax.mcmc.random_walk"]], "Parameter": [[24, "parameter"]], "blackjax.mcmc.termination": [[25, "module-blackjax.mcmc.termination"]], "blackjax.mcmc.trajectory": [[26, "module-blackjax.mcmc.trajectory"]], "blackjax.optimizers.dual_averaging": [[27, "module-blackjax.optimizers.dual_averaging"]], "blackjax.optimizers": [[28, "module-blackjax.optimizers"]], "blackjax.optimizers.lbfgs": [[29, "module-blackjax.optimizers.lbfgs"]], "blackjax.progress_bar": [[30, "module-blackjax.progress_bar"]], "blackjax.sgmcmc.csgld": [[31, "module-blackjax.sgmcmc.csgld"]], "blackjax.sgmcmc.diffusions": [[32, "module-blackjax.sgmcmc.diffusions"]], "blackjax.sgmcmc.gradients": [[33, "module-blackjax.sgmcmc.gradients"]], "blackjax.sgmcmc": [[34, "module-blackjax.sgmcmc"]], "blackjax.sgmcmc.sghmc": [[35, "module-blackjax.sgmcmc.sghmc"]], "blackjax.sgmcmc.sgld": [[36, "module-blackjax.sgmcmc.sgld"]], "blackjax.sgmcmc.sgnht": [[37, "module-blackjax.sgmcmc.sgnht"]], "blackjax.smc.adaptive_tempered": [[38, "module-blackjax.smc.adaptive_tempered"]], "blackjax.smc.base": [[39, "module-blackjax.smc.base"]], "blackjax.smc.ess": [[40, "module-blackjax.smc.ess"]], "blackjax.smc": [[41, "module-blackjax.smc"]], "blackjax.smc.resampling": [[42, "module-blackjax.smc.resampling"]], "blackjax.smc.solver": [[43, "module-blackjax.smc.solver"]], "blackjax.smc.tempered": [[44, "module-blackjax.smc.tempered"]], "blackjax.types": [[45, "module-blackjax.types"]], "blackjax.util": [[46, "module-blackjax.util"]], "blackjax.vi": [[47, "module-blackjax.vi"]], "blackjax.vi.meanfield_vi": [[48, "module-blackjax.vi.meanfield_vi"]], "blackjax.vi.pathfinder": [[49, "module-blackjax.vi.pathfinder"]], "blackjax.vi.svgd": [[50, "module-blackjax.vi.svgd"]], "Bibliography": [[51, "bibliography"]], "Use custom gradients": [[52, "use-custom-gradients"]], "Functions defined as the minimum of another function": [[52, "functions-defined-as-the-minimum-of-another-function"]], "Trying to differentate the function with jax.grad": [[52, "trying-to-differentate-the-function-with-jax-grad"]], "Deriving the gradient mathematically": [[52, "deriving-the-gradient-mathematically"]], "Telling JAX to use a custom gradient": [[52, "telling-jax-to-use-a-custom-gradient"]], "Making sure the result is correct": [[52, "making-sure-the-result-is-correct"]], "Using the function with Blackjax": [[52, "using-the-function-with-blackjax"]], "How to build a Metropolis-Within-Gibbs sampler?": [[53, "how-to-build-a-metropolis-within-gibbs-sampler"]], "The Model": [[53, "the-model"]], "MWG Sampling in BlackJAX": [[53, "mwg-sampling-in-blackjax"]], "Construct the MWG Kernel": [[53, "construct-the-mwg-kernel"]], "Sampler Parameters": [[53, "sampler-parameters"], [61, "sampler-parameters"]], "Set the Initial State of Each Algorithm": [[53, "set-the-initial-state-of-each-algorithm"]], "Build the Sampling Loop": [[53, "build-the-sampling-loop"], [53, "id1"]], "Sampling": [[53, "sampling"], [53, "id2"]], "General MWG Kernel": [[53, "general-mwg-kernel"]], "Check Result": [[53, "check-result"]], "Developer Notes": [[53, "developer-notes"]], "Use a logdensity function that is not compatible with JAX\u2019s primitives": [[54, "use-a-logdensity-function-that-is-not-compatible-with-jax-s-primitives"]], "Aesara model compiled to Numba": [[54, "aesara-model-compiled-to-numba"]], "Use jax.experimental.host_callback to call Numba functions": [[54, "use-jax-experimental-host-callback-to-call-numba-functions"]], "Use custom XLA calls to call Numba functions faster": [[54, "use-custom-xla-calls-to-call-numba-functions-faster"]], "Sample with multiple chains in parallel": [[55, "sample-with-multiple-chains-in-parallel"]], "Vectorization vs parallelization": [[55, "vectorization-vs-parallelization"]], "NUTS in parallel": [[55, "nuts-in-parallel"]], "Using jax.vmap": [[55, "using-jax-vmap"]], "Using jax.pmap": [[55, "using-jax-pmap"]], "A note on using jax.pmap on CPU": [[55, "a-note-on-using-jax-pmap-on-cpu"]], "Choosing the number of devices": [[55, "choosing-the-number-of-devices"]], "Back to our example": [[55, "back-to-our-example"]], "Conclusions": [[55, "conclusions"]], "Use with Aesara models": [[56, "use-with-aesara-models"]], "Before you start": [[56, null], [57, null], [59, null], [60, null]], "Use with Numpyro models": [[57, "use-with-numpyro-models"]], "Use with Oryx models": [[58, "use-with-oryx-models"]], "Use with PyMC models": [[59, "use-with-pymc-models"]], "Use with TFP models": [[60, "use-with-tfp-models"]], "A Quick Introduction to Blackjax": [[61, "a-quick-introduction-to-blackjax"]], "The Problem": [[61, "the-problem"]], "HMC": [[61, "hmc"]], "Set the Initial State": [[61, "set-the-initial-state"]], "Build the Kernel and Inference Loop": [[61, "build-the-kernel-and-inference-loop"]], "Inference": [[61, "inference"]], "NUTS": [[61, "nuts"]], "Use Stan\u2019s Window Adaptation": [[61, "use-stan-s-window-adaptation"]], "Welcome to Blackjax!": [[62, "welcome-to-blackjax"]], "Hello World": [[62, "hello-world"]], "Installation": [[62, "installation"]], "GPU instructions": [[62, null]]}, "indexentries": {"type_checking (in module blackjax._version)": [[0, "blackjax._version.TYPE_CHECKING"]], "version_tuple (in module blackjax._version)": [[0, "blackjax._version.VERSION_TUPLE"], [0, "blackjax._version.version_tuple"]], "blackjax._version": [[0, "module-blackjax._version"]], "module": [[0, "module-blackjax._version"], [1, "module-blackjax.adaptation.base"], [2, "module-blackjax.adaptation.chees_adaptation"], [3, "module-blackjax.adaptation"], [4, "module-blackjax.adaptation.mass_matrix"], [5, "module-blackjax.adaptation.meads_adaptation"], [6, "module-blackjax.adaptation.pathfinder_adaptation"], [7, "module-blackjax.adaptation.step_size"], [8, "module-blackjax.adaptation.window_adaptation"], [9, "module-blackjax.base"], [10, "module-blackjax.diagnostics"], [11, "module-blackjax"], [12, "module-blackjax.mcmc.diffusions"], [13, "module-blackjax.mcmc.elliptical_slice"], [14, "module-blackjax.mcmc.ghmc"], [15, "module-blackjax.mcmc.hmc"], [16, "module-blackjax.mcmc"], [17, "module-blackjax.mcmc.integrators"], [18, "module-blackjax.mcmc.mala"], [19, "module-blackjax.mcmc.marginal_latent_gaussian"], [20, "module-blackjax.mcmc.metrics"], [21, "module-blackjax.mcmc.nuts"], [22, "module-blackjax.mcmc.periodic_orbital"], [23, "module-blackjax.mcmc.proposal"], [24, "module-blackjax.mcmc.random_walk"], [25, "module-blackjax.mcmc.termination"], [26, "module-blackjax.mcmc.trajectory"], [27, "module-blackjax.optimizers.dual_averaging"], [28, "module-blackjax.optimizers"], [29, "module-blackjax.optimizers.lbfgs"], [30, "module-blackjax.progress_bar"], [31, "module-blackjax.sgmcmc.csgld"], [32, "module-blackjax.sgmcmc.diffusions"], [33, "module-blackjax.sgmcmc.gradients"], [34, "module-blackjax.sgmcmc"], [35, "module-blackjax.sgmcmc.sghmc"], [36, "module-blackjax.sgmcmc.sgld"], [37, "module-blackjax.sgmcmc.sgnht"], [38, "module-blackjax.smc.adaptive_tempered"], [39, "module-blackjax.smc.base"], [40, "module-blackjax.smc.ess"], [41, "module-blackjax.smc"], [42, "module-blackjax.smc.resampling"], [43, "module-blackjax.smc.solver"], [44, "module-blackjax.smc.tempered"], [45, "module-blackjax.types"], [46, "module-blackjax.util"], [47, "module-blackjax.vi"], [48, "module-blackjax.vi.meanfield_vi"], [49, "module-blackjax.vi.pathfinder"], [50, "module-blackjax.vi.svgd"]], "version (in module blackjax._version)": [[0, "blackjax._version.version"]], "adaptationinfo (class in blackjax.adaptation.base)": [[1, "blackjax.adaptation.base.AdaptationInfo"]], "adaptationresults (class in blackjax.adaptation.base)": [[1, "blackjax.adaptation.base.AdaptationResults"]], "adaptation_state (adaptationinfo attribute)": [[1, "blackjax.adaptation.base.AdaptationInfo.adaptation_state"]], "blackjax.adaptation.base": [[1, "module-blackjax.adaptation.base"]], "info (adaptationinfo attribute)": [[1, "blackjax.adaptation.base.AdaptationInfo.info"]], "parameters (adaptationresults attribute)": [[1, "blackjax.adaptation.base.AdaptationResults.parameters"]], "state (adaptationinfo attribute)": [[1, "blackjax.adaptation.base.AdaptationInfo.state"]], "state (adaptationresults attribute)": [[1, "blackjax.adaptation.base.AdaptationResults.state"]], "cheesadaptationstate (class in blackjax.adaptation.chees_adaptation)": [[2, "blackjax.adaptation.chees_adaptation.ChEESAdaptationState"]], "optimal_target_acceptance_rate (in module blackjax.adaptation.chees_adaptation)": [[2, "blackjax.adaptation.chees_adaptation.OPTIMAL_TARGET_ACCEPTANCE_RATE"]], "base() (in module blackjax.adaptation.chees_adaptation)": [[2, "blackjax.adaptation.chees_adaptation.base"]], "blackjax.adaptation.chees_adaptation": [[2, "module-blackjax.adaptation.chees_adaptation"]], "chees_adaptation() (in module blackjax.adaptation.chees_adaptation)": [[2, "blackjax.adaptation.chees_adaptation.chees_adaptation"]], "da_state (cheesadaptationstate attribute)": [[2, "blackjax.adaptation.chees_adaptation.ChEESAdaptationState.da_state"]], "log_step_size_moving_average (cheesadaptationstate attribute)": [[2, "blackjax.adaptation.chees_adaptation.ChEESAdaptationState.log_step_size_moving_average"]], "log_trajectory_length_moving_average (cheesadaptationstate attribute)": [[2, "blackjax.adaptation.chees_adaptation.ChEESAdaptationState.log_trajectory_length_moving_average"]], "optim_state (cheesadaptationstate attribute)": [[2, "blackjax.adaptation.chees_adaptation.ChEESAdaptationState.optim_state"]], "random_generator_arg (cheesadaptationstate attribute)": [[2, "blackjax.adaptation.chees_adaptation.ChEESAdaptationState.random_generator_arg"]], "step (cheesadaptationstate attribute)": [[2, "blackjax.adaptation.chees_adaptation.ChEESAdaptationState.step"]], "step_size (cheesadaptationstate attribute)": [[2, "blackjax.adaptation.chees_adaptation.ChEESAdaptationState.step_size"]], "trajectory_length (cheesadaptationstate attribute)": [[2, "blackjax.adaptation.chees_adaptation.ChEESAdaptationState.trajectory_length"]], "blackjax.adaptation": [[3, "module-blackjax.adaptation"]], "massmatrixadaptationstate (class in blackjax.adaptation.mass_matrix)": [[4, "blackjax.adaptation.mass_matrix.MassMatrixAdaptationState"]], "welfordalgorithmstate (class in blackjax.adaptation.mass_matrix)": [[4, "blackjax.adaptation.mass_matrix.WelfordAlgorithmState"]], "blackjax.adaptation.mass_matrix": [[4, "module-blackjax.adaptation.mass_matrix"]], "inverse_mass_matrix (massmatrixadaptationstate attribute)": [[4, "blackjax.adaptation.mass_matrix.MassMatrixAdaptationState.inverse_mass_matrix"]], "m2 (welfordalgorithmstate attribute)": [[4, "blackjax.adaptation.mass_matrix.WelfordAlgorithmState.m2"]], "mass_matrix_adaptation() (in module blackjax.adaptation.mass_matrix)": [[4, "blackjax.adaptation.mass_matrix.mass_matrix_adaptation"]], "mean (welfordalgorithmstate attribute)": [[4, "blackjax.adaptation.mass_matrix.WelfordAlgorithmState.mean"]], "sample_size (welfordalgorithmstate attribute)": [[4, "blackjax.adaptation.mass_matrix.WelfordAlgorithmState.sample_size"]], "wc_state (massmatrixadaptationstate attribute)": [[4, "blackjax.adaptation.mass_matrix.MassMatrixAdaptationState.wc_state"]], "welford_algorithm() (in module blackjax.adaptation.mass_matrix)": [[4, "blackjax.adaptation.mass_matrix.welford_algorithm"]], "meadsadaptationstate (class in blackjax.adaptation.meads_adaptation)": [[5, "blackjax.adaptation.meads_adaptation.MEADSAdaptationState"]], "alpha (meadsadaptationstate attribute)": [[5, "blackjax.adaptation.meads_adaptation.MEADSAdaptationState.alpha"]], "base() (in module blackjax.adaptation.meads_adaptation)": [[5, "blackjax.adaptation.meads_adaptation.base"]], "blackjax.adaptation.meads_adaptation": [[5, "module-blackjax.adaptation.meads_adaptation"]], "current_iteration (meadsadaptationstate attribute)": [[5, "blackjax.adaptation.meads_adaptation.MEADSAdaptationState.current_iteration"]], "delta (meadsadaptationstate attribute)": [[5, "blackjax.adaptation.meads_adaptation.MEADSAdaptationState.delta"]], "maximum_eigenvalue() (in module blackjax.adaptation.meads_adaptation)": [[5, "blackjax.adaptation.meads_adaptation.maximum_eigenvalue"]], "meads_adaptation() (in module blackjax.adaptation.meads_adaptation)": [[5, "blackjax.adaptation.meads_adaptation.meads_adaptation"]], "position_sigma (meadsadaptationstate attribute)": [[5, "blackjax.adaptation.meads_adaptation.MEADSAdaptationState.position_sigma"]], "step_size (meadsadaptationstate attribute)": [[5, "blackjax.adaptation.meads_adaptation.MEADSAdaptationState.step_size"]], "pathfinderadaptationstate (class in blackjax.adaptation.pathfinder_adaptation)": [[6, "blackjax.adaptation.pathfinder_adaptation.PathfinderAdaptationState"]], "base() (in module blackjax.adaptation.pathfinder_adaptation)": [[6, "blackjax.adaptation.pathfinder_adaptation.base"]], "blackjax.adaptation.pathfinder_adaptation": [[6, "module-blackjax.adaptation.pathfinder_adaptation"]], "inverse_mass_matrix (pathfinderadaptationstate attribute)": [[6, "blackjax.adaptation.pathfinder_adaptation.PathfinderAdaptationState.inverse_mass_matrix"]], "pathfinder_adaptation() (in module blackjax.adaptation.pathfinder_adaptation)": [[6, "blackjax.adaptation.pathfinder_adaptation.pathfinder_adaptation"]], "ss_state (pathfinderadaptationstate attribute)": [[6, "blackjax.adaptation.pathfinder_adaptation.PathfinderAdaptationState.ss_state"]], "step_size (pathfinderadaptationstate attribute)": [[6, "blackjax.adaptation.pathfinder_adaptation.PathfinderAdaptationState.step_size"]], "dualaveragingadaptationstate (class in blackjax.adaptation.step_size)": [[7, "blackjax.adaptation.step_size.DualAveragingAdaptationState"]], "avg_error (dualaveragingadaptationstate attribute)": [[7, "blackjax.adaptation.step_size.DualAveragingAdaptationState.avg_error"]], "blackjax.adaptation.step_size": [[7, "module-blackjax.adaptation.step_size"]], "dual_averaging_adaptation() (in module blackjax.adaptation.step_size)": [[7, "blackjax.adaptation.step_size.dual_averaging_adaptation"]], "find_reasonable_step_size() (in module blackjax.adaptation.step_size)": [[7, "blackjax.adaptation.step_size.find_reasonable_step_size"]], "log_step_size (dualaveragingadaptationstate attribute)": [[7, "blackjax.adaptation.step_size.DualAveragingAdaptationState.log_step_size"]], "log_step_size_avg (dualaveragingadaptationstate attribute)": [[7, "blackjax.adaptation.step_size.DualAveragingAdaptationState.log_step_size_avg"]], "mu (dualaveragingadaptationstate attribute)": [[7, "blackjax.adaptation.step_size.DualAveragingAdaptationState.mu"]], "step (dualaveragingadaptationstate attribute)": [[7, "blackjax.adaptation.step_size.DualAveragingAdaptationState.step"]], "windowadaptationstate (class in blackjax.adaptation.window_adaptation)": [[8, "blackjax.adaptation.window_adaptation.WindowAdaptationState"]], "base() (in module blackjax.adaptation.window_adaptation)": [[8, "blackjax.adaptation.window_adaptation.base"]], "blackjax.adaptation.window_adaptation": [[8, "module-blackjax.adaptation.window_adaptation"]], "build_schedule() (in module blackjax.adaptation.window_adaptation)": [[8, "blackjax.adaptation.window_adaptation.build_schedule"]], "imm_state (windowadaptationstate attribute)": [[8, "blackjax.adaptation.window_adaptation.WindowAdaptationState.imm_state"]], "inverse_mass_matrix (windowadaptationstate attribute)": [[8, "blackjax.adaptation.window_adaptation.WindowAdaptationState.inverse_mass_matrix"]], "ss_state (windowadaptationstate attribute)": [[8, "blackjax.adaptation.window_adaptation.WindowAdaptationState.ss_state"]], "step_size (windowadaptationstate attribute)": [[8, "blackjax.adaptation.window_adaptation.WindowAdaptationState.step_size"]], "window_adaptation() (in module blackjax.adaptation.window_adaptation)": [[8, "blackjax.adaptation.window_adaptation.window_adaptation"]], "adaptationalgorithm (class in blackjax.base)": [[9, "blackjax.base.AdaptationAlgorithm"]], "info (in module blackjax.base)": [[9, "blackjax.base.Info"]], "initfn (class in blackjax.base)": [[9, "blackjax.base.InitFn"]], "position (in module blackjax.base)": [[9, "blackjax.base.Position"]], "runfn (class in blackjax.base)": [[9, "blackjax.base.RunFn"]], "samplingalgorithm (class in blackjax.base)": [[9, "blackjax.base.SamplingAlgorithm"]], "state (in module blackjax.base)": [[9, "blackjax.base.State"]], "updatefn (class in blackjax.base)": [[9, "blackjax.base.UpdateFn"]], "vialgorithm (class in blackjax.base)": [[9, "blackjax.base.VIAlgorithm"]], "blackjax.base": [[9, "module-blackjax.base"]], "init (samplingalgorithm attribute)": [[9, "blackjax.base.SamplingAlgorithm.init"]], "init (vialgorithm attribute)": [[9, "blackjax.base.VIAlgorithm.init"]], "run (adaptationalgorithm attribute)": [[9, "blackjax.base.AdaptationAlgorithm.run"]], "sample (vialgorithm attribute)": [[9, "blackjax.base.VIAlgorithm.sample"]], "step (samplingalgorithm attribute)": [[9, "blackjax.base.SamplingAlgorithm.step"]], "step (vialgorithm attribute)": [[9, "blackjax.base.VIAlgorithm.step"]], "blackjax.diagnostics": [[10, "module-blackjax.diagnostics"]], "effective_sample_size() (in module blackjax.diagnostics)": [[10, "blackjax.diagnostics.effective_sample_size"]], "potential_scale_reduction() (in module blackjax.diagnostics)": [[10, "blackjax.diagnostics.potential_scale_reduction"]], "blackjax": [[11, "module-blackjax"]], "blackjax.mcmc.diffusions": [[12, "module-blackjax.mcmc.diffusions"]], "overdamped_langevin() (in module blackjax.mcmc.diffusions)": [[12, "blackjax.mcmc.diffusions.overdamped_langevin"]], "ellipsliceinfo (class in blackjax.mcmc.elliptical_slice)": [[13, "blackjax.mcmc.elliptical_slice.EllipSliceInfo"]], "ellipslicestate (class in blackjax.mcmc.elliptical_slice)": [[13, "blackjax.mcmc.elliptical_slice.EllipSliceState"]], "blackjax.mcmc.elliptical_slice": [[13, "module-blackjax.mcmc.elliptical_slice"]], "build_kernel (elliptical_slice attribute)": [[13, "blackjax.mcmc.elliptical_slice.elliptical_slice.build_kernel"]], "build_kernel() (in module blackjax.mcmc.elliptical_slice)": [[13, "blackjax.mcmc.elliptical_slice.build_kernel"]], "elliptical_slice (class in blackjax.mcmc.elliptical_slice)": [[13, "blackjax.mcmc.elliptical_slice.elliptical_slice"]], "init (elliptical_slice attribute)": [[13, "blackjax.mcmc.elliptical_slice.elliptical_slice.init"]], "init() (in module blackjax.mcmc.elliptical_slice)": [[13, "blackjax.mcmc.elliptical_slice.init"]], "logdensity (ellipslicestate attribute)": [[13, "blackjax.mcmc.elliptical_slice.EllipSliceState.logdensity"]], "momentum (ellipsliceinfo attribute)": [[13, "blackjax.mcmc.elliptical_slice.EllipSliceInfo.momentum"]], "position (ellipslicestate attribute)": [[13, "blackjax.mcmc.elliptical_slice.EllipSliceState.position"]], "subiter (ellipsliceinfo attribute)": [[13, "blackjax.mcmc.elliptical_slice.EllipSliceInfo.subiter"]], "theta (ellipsliceinfo attribute)": [[13, "blackjax.mcmc.elliptical_slice.EllipSliceInfo.theta"]], "ghmcstate (class in blackjax.mcmc.ghmc)": [[14, "blackjax.mcmc.ghmc.GHMCState"]], "blackjax.mcmc.ghmc": [[14, "module-blackjax.mcmc.ghmc"]], "build_kernel (ghmc attribute)": [[14, "blackjax.mcmc.ghmc.ghmc.build_kernel"]], "build_kernel() (in module blackjax.mcmc.ghmc)": [[14, "blackjax.mcmc.ghmc.build_kernel"]], "ghmc (class in blackjax.mcmc.ghmc)": [[14, "blackjax.mcmc.ghmc.ghmc"]], "init (ghmc attribute)": [[14, "blackjax.mcmc.ghmc.ghmc.init"]], "init() (in module blackjax.mcmc.ghmc)": [[14, "blackjax.mcmc.ghmc.init"]], "logdensity (ghmcstate attribute)": [[14, "blackjax.mcmc.ghmc.GHMCState.logdensity"]], "logdensity_grad (ghmcstate attribute)": [[14, "blackjax.mcmc.ghmc.GHMCState.logdensity_grad"]], "momentum (ghmcstate attribute)": [[14, "blackjax.mcmc.ghmc.GHMCState.momentum"]], "position (ghmcstate attribute)": [[14, "blackjax.mcmc.ghmc.GHMCState.position"]], "slice (ghmcstate attribute)": [[14, "blackjax.mcmc.ghmc.GHMCState.slice"]], "dynamichmcstate (class in blackjax.mcmc.hmc)": [[15, "blackjax.mcmc.hmc.DynamicHMCState"]], "hmcinfo (class in blackjax.mcmc.hmc)": [[15, "blackjax.mcmc.hmc.HMCInfo"]], "hmcstate (class in blackjax.mcmc.hmc)": [[15, "blackjax.mcmc.hmc.HMCState"]], "acceptance_rate (hmcinfo attribute)": [[15, "blackjax.mcmc.hmc.HMCInfo.acceptance_rate"]], "blackjax.mcmc.hmc": [[15, "module-blackjax.mcmc.hmc"]], "build_dynamic_kernel() (in module blackjax.mcmc.hmc)": [[15, "blackjax.mcmc.hmc.build_dynamic_kernel"]], "build_kernel (dynamic_hmc attribute)": [[15, "blackjax.mcmc.hmc.dynamic_hmc.build_kernel"]], "build_kernel (hmc attribute)": [[15, "blackjax.mcmc.hmc.hmc.build_kernel"]], "build_kernel() (in module blackjax.mcmc.hmc)": [[15, "blackjax.mcmc.hmc.build_kernel"]], "dynamic_hmc (class in blackjax.mcmc.hmc)": [[15, "blackjax.mcmc.hmc.dynamic_hmc"]], "energy (hmcinfo attribute)": [[15, "blackjax.mcmc.hmc.HMCInfo.energy"]], "hmc (class in blackjax.mcmc.hmc)": [[15, "blackjax.mcmc.hmc.hmc"]], "init (dynamic_hmc attribute)": [[15, "blackjax.mcmc.hmc.dynamic_hmc.init"]], "init (hmc attribute)": [[15, "blackjax.mcmc.hmc.hmc.init"]], "init() (in module blackjax.mcmc.hmc)": [[15, "blackjax.mcmc.hmc.init"]], "init_dynamic() (in module blackjax.mcmc.hmc)": [[15, "blackjax.mcmc.hmc.init_dynamic"]], "is_accepted (hmcinfo attribute)": [[15, "blackjax.mcmc.hmc.HMCInfo.is_accepted"]], "is_divergent (hmcinfo attribute)": [[15, "blackjax.mcmc.hmc.HMCInfo.is_divergent"]], "logdensity (dynamichmcstate attribute)": [[15, "blackjax.mcmc.hmc.DynamicHMCState.logdensity"]], "logdensity (hmcstate attribute)": [[15, "blackjax.mcmc.hmc.HMCState.logdensity"]], "logdensity_grad (dynamichmcstate attribute)": [[15, "blackjax.mcmc.hmc.DynamicHMCState.logdensity_grad"]], "logdensity_grad (hmcstate attribute)": [[15, "blackjax.mcmc.hmc.HMCState.logdensity_grad"]], "momentum (hmcinfo attribute)": [[15, "blackjax.mcmc.hmc.HMCInfo.momentum"]], "num_integration_steps (hmcinfo attribute)": [[15, "blackjax.mcmc.hmc.HMCInfo.num_integration_steps"]], "position (dynamichmcstate attribute)": [[15, "blackjax.mcmc.hmc.DynamicHMCState.position"]], "position (hmcstate attribute)": [[15, "blackjax.mcmc.hmc.HMCState.position"]], "proposal (hmcinfo attribute)": [[15, "blackjax.mcmc.hmc.HMCInfo.proposal"]], "random_generator_arg (dynamichmcstate attribute)": [[15, "blackjax.mcmc.hmc.DynamicHMCState.random_generator_arg"]], "blackjax.mcmc": [[16, "module-blackjax.mcmc"]], "blackjax.mcmc.integrators": [[17, "module-blackjax.mcmc.integrators"]], "mclachlan (in module blackjax.mcmc.integrators)": [[17, "blackjax.mcmc.integrators.mclachlan"]], "noneuclidean_leapfrog (in module blackjax.mcmc.integrators)": [[17, "blackjax.mcmc.integrators.noneuclidean_leapfrog"]], "noneuclidean_mclachlan (in module blackjax.mcmc.integrators)": [[17, "blackjax.mcmc.integrators.noneuclidean_mclachlan"]], "noneuclidean_yoshida (in module blackjax.mcmc.integrators)": [[17, "blackjax.mcmc.integrators.noneuclidean_yoshida"]], "velocity_verlet (in module blackjax.mcmc.integrators)": [[17, "blackjax.mcmc.integrators.velocity_verlet"]], "yoshida (in module blackjax.mcmc.integrators)": [[17, "blackjax.mcmc.integrators.yoshida"]], "malainfo (class in blackjax.mcmc.mala)": [[18, "blackjax.mcmc.mala.MALAInfo"]], "malastate (class in blackjax.mcmc.mala)": [[18, "blackjax.mcmc.mala.MALAState"]], "acceptance_rate (malainfo attribute)": [[18, "blackjax.mcmc.mala.MALAInfo.acceptance_rate"]], "blackjax.mcmc.mala": [[18, "module-blackjax.mcmc.mala"]], "build_kernel (mala attribute)": [[18, "blackjax.mcmc.mala.mala.build_kernel"]], "build_kernel() (in module blackjax.mcmc.mala)": [[18, "blackjax.mcmc.mala.build_kernel"]], "init (mala attribute)": [[18, "blackjax.mcmc.mala.mala.init"]], "init() (in module blackjax.mcmc.mala)": [[18, "blackjax.mcmc.mala.init"]], "is_accepted (malainfo attribute)": [[18, "blackjax.mcmc.mala.MALAInfo.is_accepted"]], "logdensity (malastate attribute)": [[18, "blackjax.mcmc.mala.MALAState.logdensity"]], "logdensity_grad (malastate attribute)": [[18, "blackjax.mcmc.mala.MALAState.logdensity_grad"]], "mala (class in blackjax.mcmc.mala)": [[18, "blackjax.mcmc.mala.mala"]], "position (malastate attribute)": [[18, "blackjax.mcmc.mala.MALAState.position"]], "marginalinfo (class in blackjax.mcmc.marginal_latent_gaussian)": [[19, "blackjax.mcmc.marginal_latent_gaussian.MarginalInfo"]], "marginalstate (class in blackjax.mcmc.marginal_latent_gaussian)": [[19, "blackjax.mcmc.marginal_latent_gaussian.MarginalState"]], "u_grad_x (marginalstate attribute)": [[19, "blackjax.mcmc.marginal_latent_gaussian.MarginalState.U_grad_x"]], "u_x (marginalstate attribute)": [[19, "blackjax.mcmc.marginal_latent_gaussian.MarginalState.U_x"]], "acceptance_rate (marginalinfo attribute)": [[19, "blackjax.mcmc.marginal_latent_gaussian.MarginalInfo.acceptance_rate"]], "blackjax.mcmc.marginal_latent_gaussian": [[19, "module-blackjax.mcmc.marginal_latent_gaussian"]], "init_and_kernel() (in module blackjax.mcmc.marginal_latent_gaussian)": [[19, "blackjax.mcmc.marginal_latent_gaussian.init_and_kernel"]], "is_accepted (marginalinfo attribute)": [[19, "blackjax.mcmc.marginal_latent_gaussian.MarginalInfo.is_accepted"]], "logdensity (marginalstate attribute)": [[19, "blackjax.mcmc.marginal_latent_gaussian.MarginalState.logdensity"]], "logdensity_grad (marginalstate attribute)": [[19, "blackjax.mcmc.marginal_latent_gaussian.MarginalState.logdensity_grad"]], "mgrad_gaussian (class in blackjax.mcmc.marginal_latent_gaussian)": [[19, "blackjax.mcmc.marginal_latent_gaussian.mgrad_gaussian"]], "position (marginalstate attribute)": [[19, "blackjax.mcmc.marginal_latent_gaussian.MarginalState.position"]], "proposal (marginalinfo attribute)": [[19, "blackjax.mcmc.marginal_latent_gaussian.MarginalInfo.proposal"]], "blackjax.mcmc.metrics": [[20, "module-blackjax.mcmc.metrics"]], "gaussian_euclidean() (in module blackjax.mcmc.metrics)": [[20, "blackjax.mcmc.metrics.gaussian_euclidean"]], "nutsinfo (class in blackjax.mcmc.nuts)": [[21, "blackjax.mcmc.nuts.NUTSInfo"]], "acceptance_rate (nutsinfo attribute)": [[21, "blackjax.mcmc.nuts.NUTSInfo.acceptance_rate"]], "blackjax.mcmc.nuts": [[21, "module-blackjax.mcmc.nuts"]], "build_kernel (nuts attribute)": [[21, "blackjax.mcmc.nuts.nuts.build_kernel"]], "build_kernel() (in module blackjax.mcmc.nuts)": [[21, "blackjax.mcmc.nuts.build_kernel"]], "energy (nutsinfo attribute)": [[21, "blackjax.mcmc.nuts.NUTSInfo.energy"]], "init (in module blackjax.mcmc.nuts)": [[21, "blackjax.mcmc.nuts.init"]], "init (nuts attribute)": [[21, "blackjax.mcmc.nuts.nuts.init"]], "is_divergent (nutsinfo attribute)": [[21, "blackjax.mcmc.nuts.NUTSInfo.is_divergent"]], "is_turning (nutsinfo attribute)": [[21, "blackjax.mcmc.nuts.NUTSInfo.is_turning"]], "momentum (nutsinfo attribute)": [[21, "blackjax.mcmc.nuts.NUTSInfo.momentum"]], "num_integration_steps (nutsinfo attribute)": [[21, "blackjax.mcmc.nuts.NUTSInfo.num_integration_steps"]], "num_trajectory_expansions (nutsinfo attribute)": [[21, "blackjax.mcmc.nuts.NUTSInfo.num_trajectory_expansions"]], "nuts (class in blackjax.mcmc.nuts)": [[21, "blackjax.mcmc.nuts.nuts"]], "trajectory_leftmost_state (nutsinfo attribute)": [[21, "blackjax.mcmc.nuts.NUTSInfo.trajectory_leftmost_state"]], "trajectory_rightmost_state (nutsinfo attribute)": [[21, "blackjax.mcmc.nuts.NUTSInfo.trajectory_rightmost_state"]], "periodicorbitalstate (class in blackjax.mcmc.periodic_orbital)": [[22, "blackjax.mcmc.periodic_orbital.PeriodicOrbitalState"]], "blackjax.mcmc.periodic_orbital": [[22, "module-blackjax.mcmc.periodic_orbital"]], "build_kernel (orbital_hmc attribute)": [[22, "blackjax.mcmc.periodic_orbital.orbital_hmc.build_kernel"]], "build_kernel() (in module blackjax.mcmc.periodic_orbital)": [[22, "blackjax.mcmc.periodic_orbital.build_kernel"]], "directions (periodicorbitalstate attribute)": [[22, "blackjax.mcmc.periodic_orbital.PeriodicOrbitalState.directions"]], "init (orbital_hmc attribute)": [[22, "blackjax.mcmc.periodic_orbital.orbital_hmc.init"]], "init() (in module blackjax.mcmc.periodic_orbital)": [[22, "blackjax.mcmc.periodic_orbital.init"]], "logdensities (periodicorbitalstate attribute)": [[22, "blackjax.mcmc.periodic_orbital.PeriodicOrbitalState.logdensities"]], "logdensities_grad (periodicorbitalstate attribute)": [[22, "blackjax.mcmc.periodic_orbital.PeriodicOrbitalState.logdensities_grad"]], "orbital_hmc (class in blackjax.mcmc.periodic_orbital)": [[22, "blackjax.mcmc.periodic_orbital.orbital_hmc"]], "positions (periodicorbitalstate attribute)": [[22, "blackjax.mcmc.periodic_orbital.PeriodicOrbitalState.positions"]], "weights (periodicorbitalstate attribute)": [[22, "blackjax.mcmc.periodic_orbital.PeriodicOrbitalState.weights"]], "proposal (class in blackjax.mcmc.proposal)": [[23, "blackjax.mcmc.proposal.Proposal"]], "sampleproposal (class in blackjax.mcmc.proposal)": [[23, "blackjax.mcmc.proposal.SampleProposal"]], "trajectorystate (in module blackjax.mcmc.proposal)": [[23, "blackjax.mcmc.proposal.TrajectoryState"]], "asymmetric_proposal_generator() (in module blackjax.mcmc.proposal)": [[23, "blackjax.mcmc.proposal.asymmetric_proposal_generator"]], "blackjax.mcmc.proposal": [[23, "module-blackjax.mcmc.proposal"]], "energy (proposal attribute)": [[23, "blackjax.mcmc.proposal.Proposal.energy"]], "nonreversible_slice_sampling() (in module blackjax.mcmc.proposal)": [[23, "blackjax.mcmc.proposal.nonreversible_slice_sampling"]], "progressive_biased_sampling() (in module blackjax.mcmc.proposal)": [[23, "blackjax.mcmc.proposal.progressive_biased_sampling"]], "progressive_uniform_sampling() (in module blackjax.mcmc.proposal)": [[23, "blackjax.mcmc.proposal.progressive_uniform_sampling"]], "proposal_from_energy_diff() (in module blackjax.mcmc.proposal)": [[23, "blackjax.mcmc.proposal.proposal_from_energy_diff"]], "proposal_generator() (in module blackjax.mcmc.proposal)": [[23, "blackjax.mcmc.proposal.proposal_generator"]], "state (proposal attribute)": [[23, "blackjax.mcmc.proposal.Proposal.state"]], "static_binomial_sampling() (in module blackjax.mcmc.proposal)": [[23, "blackjax.mcmc.proposal.static_binomial_sampling"]], "sum_log_p_accept (proposal attribute)": [[23, "blackjax.mcmc.proposal.Proposal.sum_log_p_accept"]], "weight (proposal attribute)": [[23, "blackjax.mcmc.proposal.Proposal.weight"]], "rwinfo (class in blackjax.mcmc.random_walk)": [[24, "blackjax.mcmc.random_walk.RWInfo"]], "rwstate (class in blackjax.mcmc.random_walk)": [[24, "blackjax.mcmc.random_walk.RWState"]], "acceptance_rate (rwinfo attribute)": [[24, "blackjax.mcmc.random_walk.RWInfo.acceptance_rate"]], "additive_step_random_walk (class in blackjax.mcmc.random_walk)": [[24, "blackjax.mcmc.random_walk.additive_step_random_walk"]], "blackjax.mcmc.random_walk": [[24, "module-blackjax.mcmc.random_walk"]], "build_additive_step() (in module blackjax.mcmc.random_walk)": [[24, "blackjax.mcmc.random_walk.build_additive_step"]], "build_irmh() (in module blackjax.mcmc.random_walk)": [[24, "blackjax.mcmc.random_walk.build_irmh"]], "build_kernel (additive_step_random_walk attribute)": [[24, "blackjax.mcmc.random_walk.additive_step_random_walk.build_kernel"]], "build_kernel (irmh attribute)": [[24, "blackjax.mcmc.random_walk.irmh.build_kernel"]], "build_kernel (rmh attribute)": [[24, "blackjax.mcmc.random_walk.rmh.build_kernel"]], "build_rmh() (in module blackjax.mcmc.random_walk)": [[24, "blackjax.mcmc.random_walk.build_rmh"]], "build_rmh_transition_energy() (in module blackjax.mcmc.random_walk)": [[24, "blackjax.mcmc.random_walk.build_rmh_transition_energy"]], "init (additive_step_random_walk attribute)": [[24, "blackjax.mcmc.random_walk.additive_step_random_walk.init"]], "init (irmh attribute)": [[24, "blackjax.mcmc.random_walk.irmh.init"]], "init (rmh attribute)": [[24, "blackjax.mcmc.random_walk.rmh.init"]], "irmh (class in blackjax.mcmc.random_walk)": [[24, "blackjax.mcmc.random_walk.irmh"]], "is_accepted (rwinfo attribute)": [[24, "blackjax.mcmc.random_walk.RWInfo.is_accepted"]], "logdensity (rwstate attribute)": [[24, "blackjax.mcmc.random_walk.RWState.logdensity"]], "normal() (in module blackjax.mcmc.random_walk)": [[24, "blackjax.mcmc.random_walk.normal"]], "normal_random_walk() (additive_step_random_walk class method)": [[24, "blackjax.mcmc.random_walk.additive_step_random_walk.normal_random_walk"]], "position (rwstate attribute)": [[24, "blackjax.mcmc.random_walk.RWState.position"]], "proposal (rwinfo attribute)": [[24, "blackjax.mcmc.random_walk.RWInfo.proposal"]], "rmh (class in blackjax.mcmc.random_walk)": [[24, "blackjax.mcmc.random_walk.rmh"]], "rmh_proposal() (in module blackjax.mcmc.random_walk)": [[24, "blackjax.mcmc.random_walk.rmh_proposal"]], "iterativeuturnstate (class in blackjax.mcmc.termination)": [[25, "blackjax.mcmc.termination.IterativeUTurnState"]], "blackjax.mcmc.termination": [[25, "module-blackjax.mcmc.termination"]], "idx_max (iterativeuturnstate attribute)": [[25, "blackjax.mcmc.termination.IterativeUTurnState.idx_max"]], "idx_min (iterativeuturnstate attribute)": [[25, "blackjax.mcmc.termination.IterativeUTurnState.idx_min"]], "iterative_uturn_numpyro() (in module blackjax.mcmc.termination)": [[25, "blackjax.mcmc.termination.iterative_uturn_numpyro"]], "momentum (iterativeuturnstate attribute)": [[25, "blackjax.mcmc.termination.IterativeUTurnState.momentum"]], "momentum_sum (iterativeuturnstate attribute)": [[25, "blackjax.mcmc.termination.IterativeUTurnState.momentum_sum"]], "dynamicexpansionstate (class in blackjax.mcmc.trajectory)": [[26, "blackjax.mcmc.trajectory.DynamicExpansionState"]], "dynamicintegrationstate (class in blackjax.mcmc.trajectory)": [[26, "blackjax.mcmc.trajectory.DynamicIntegrationState"]], "trajectory (class in blackjax.mcmc.trajectory)": [[26, "blackjax.mcmc.trajectory.Trajectory"]], "append_to_trajectory() (in module blackjax.mcmc.trajectory)": [[26, "blackjax.mcmc.trajectory.append_to_trajectory"]], "blackjax.mcmc.trajectory": [[26, "module-blackjax.mcmc.trajectory"]], "dynamic_multiplicative_expansion() (in module blackjax.mcmc.trajectory)": [[26, "blackjax.mcmc.trajectory.dynamic_multiplicative_expansion"]], "dynamic_progressive_integration() (in module blackjax.mcmc.trajectory)": [[26, "blackjax.mcmc.trajectory.dynamic_progressive_integration"]], "dynamic_recursive_integration() (in module blackjax.mcmc.trajectory)": [[26, "blackjax.mcmc.trajectory.dynamic_recursive_integration"]], "hmc_energy() (in module blackjax.mcmc.trajectory)": [[26, "blackjax.mcmc.trajectory.hmc_energy"]], "leftmost_state (trajectory attribute)": [[26, "blackjax.mcmc.trajectory.Trajectory.leftmost_state"]], "merge_trajectories() (in module blackjax.mcmc.trajectory)": [[26, "blackjax.mcmc.trajectory.merge_trajectories"]], "momentum_sum (trajectory attribute)": [[26, "blackjax.mcmc.trajectory.Trajectory.momentum_sum"]], "num_states (trajectory attribute)": [[26, "blackjax.mcmc.trajectory.Trajectory.num_states"]], "proposal (dynamicexpansionstate attribute)": [[26, "blackjax.mcmc.trajectory.DynamicExpansionState.proposal"]], "proposal (dynamicintegrationstate attribute)": [[26, "blackjax.mcmc.trajectory.DynamicIntegrationState.proposal"]], "reorder_trajectories() (in module blackjax.mcmc.trajectory)": [[26, "blackjax.mcmc.trajectory.reorder_trajectories"]], "rightmost_state (trajectory attribute)": [[26, "blackjax.mcmc.trajectory.Trajectory.rightmost_state"]], "static_integration() (in module blackjax.mcmc.trajectory)": [[26, "blackjax.mcmc.trajectory.static_integration"]], "step (dynamicexpansionstate attribute)": [[26, "blackjax.mcmc.trajectory.DynamicExpansionState.step"]], "step (dynamicintegrationstate attribute)": [[26, "blackjax.mcmc.trajectory.DynamicIntegrationState.step"]], "termination_state (dynamicexpansionstate attribute)": [[26, "blackjax.mcmc.trajectory.DynamicExpansionState.termination_state"]], "termination_state (dynamicintegrationstate attribute)": [[26, "blackjax.mcmc.trajectory.DynamicIntegrationState.termination_state"]], "trajectory (dynamicexpansionstate attribute)": [[26, "blackjax.mcmc.trajectory.DynamicExpansionState.trajectory"]], "trajectory (dynamicintegrationstate attribute)": [[26, "blackjax.mcmc.trajectory.DynamicIntegrationState.trajectory"]], "dualaveragingstate (class in blackjax.optimizers.dual_averaging)": [[27, "blackjax.optimizers.dual_averaging.DualAveragingState"]], "avg_error (dualaveragingstate attribute)": [[27, "blackjax.optimizers.dual_averaging.DualAveragingState.avg_error"]], "blackjax.optimizers.dual_averaging": [[27, "module-blackjax.optimizers.dual_averaging"]], "dual_averaging() (in module blackjax.optimizers.dual_averaging)": [[27, "blackjax.optimizers.dual_averaging.dual_averaging"]], "log_x (dualaveragingstate attribute)": [[27, "blackjax.optimizers.dual_averaging.DualAveragingState.log_x"]], "log_x_avg (dualaveragingstate attribute)": [[27, "blackjax.optimizers.dual_averaging.DualAveragingState.log_x_avg"]], "mu (dualaveragingstate attribute)": [[27, "blackjax.optimizers.dual_averaging.DualAveragingState.mu"]], "step (dualaveragingstate attribute)": [[27, "blackjax.optimizers.dual_averaging.DualAveragingState.step"]], "blackjax.optimizers": [[28, "module-blackjax.optimizers"]], "lbfgshistory (class in blackjax.optimizers.lbfgs)": [[29, "blackjax.optimizers.lbfgs.LBFGSHistory"]], "alpha (lbfgshistory attribute)": [[29, "blackjax.optimizers.lbfgs.LBFGSHistory.alpha"]], "bfgs_sample() (in module blackjax.optimizers.lbfgs)": [[29, "blackjax.optimizers.lbfgs.bfgs_sample"]], "blackjax.optimizers.lbfgs": [[29, "module-blackjax.optimizers.lbfgs"]], "f (lbfgshistory attribute)": [[29, "blackjax.optimizers.lbfgs.LBFGSHistory.f"]], "g (lbfgshistory attribute)": [[29, "blackjax.optimizers.lbfgs.LBFGSHistory.g"]], "lbfgs_inverse_hessian_factors() (in module blackjax.optimizers.lbfgs)": [[29, "blackjax.optimizers.lbfgs.lbfgs_inverse_hessian_factors"]], "lbfgs_inverse_hessian_formula_1() (in module blackjax.optimizers.lbfgs)": [[29, "blackjax.optimizers.lbfgs.lbfgs_inverse_hessian_formula_1"]], "lbfgs_inverse_hessian_formula_2() (in module blackjax.optimizers.lbfgs)": [[29, "blackjax.optimizers.lbfgs.lbfgs_inverse_hessian_formula_2"]], "minimize_lbfgs() (in module blackjax.optimizers.lbfgs)": [[29, "blackjax.optimizers.lbfgs.minimize_lbfgs"]], "update_mask (lbfgshistory attribute)": [[29, "blackjax.optimizers.lbfgs.LBFGSHistory.update_mask"]], "x (lbfgshistory attribute)": [[29, "blackjax.optimizers.lbfgs.LBFGSHistory.x"]], "blackjax.progress_bar": [[30, "module-blackjax.progress_bar"]], "progress_bar_scan() (in module blackjax.progress_bar)": [[30, "blackjax.progress_bar.progress_bar_scan"]], "contoursgldstate (class in blackjax.sgmcmc.csgld)": [[31, "blackjax.sgmcmc.csgld.ContourSGLDState"]], "blackjax.sgmcmc.csgld": [[31, "module-blackjax.sgmcmc.csgld"]], "build_kernel (csgld attribute)": [[31, "blackjax.sgmcmc.csgld.csgld.build_kernel"]], "build_kernel() (in module blackjax.sgmcmc.csgld)": [[31, "blackjax.sgmcmc.csgld.build_kernel"]], "csgld (class in blackjax.sgmcmc.csgld)": [[31, "blackjax.sgmcmc.csgld.csgld"]], "energy_idx (contoursgldstate attribute)": [[31, "blackjax.sgmcmc.csgld.ContourSGLDState.energy_idx"]], "energy_pdf (contoursgldstate attribute)": [[31, "blackjax.sgmcmc.csgld.ContourSGLDState.energy_pdf"]], "init (csgld attribute)": [[31, "blackjax.sgmcmc.csgld.csgld.init"]], "init() (in module blackjax.sgmcmc.csgld)": [[31, "blackjax.sgmcmc.csgld.init"]], "position (contoursgldstate attribute)": [[31, "blackjax.sgmcmc.csgld.ContourSGLDState.position"]], "blackjax.sgmcmc.diffusions": [[32, "module-blackjax.sgmcmc.diffusions"]], "overdamped_langevin() (in module blackjax.sgmcmc.diffusions)": [[32, "blackjax.sgmcmc.diffusions.overdamped_langevin"]], "sghmc() (in module blackjax.sgmcmc.diffusions)": [[32, "blackjax.sgmcmc.diffusions.sghmc"]], "sgnht() (in module blackjax.sgmcmc.diffusions)": [[32, "blackjax.sgmcmc.diffusions.sgnht"]], "blackjax.sgmcmc.gradients": [[33, "module-blackjax.sgmcmc.gradients"]], "control_variates() (in module blackjax.sgmcmc.gradients)": [[33, "blackjax.sgmcmc.gradients.control_variates"]], "grad_estimator() (in module blackjax.sgmcmc.gradients)": [[33, "blackjax.sgmcmc.gradients.grad_estimator"]], "logdensity_estimator() (in module blackjax.sgmcmc.gradients)": [[33, "blackjax.sgmcmc.gradients.logdensity_estimator"]], "blackjax.sgmcmc": [[34, "module-blackjax.sgmcmc"]], "blackjax.sgmcmc.sghmc": [[35, "module-blackjax.sgmcmc.sghmc"]], "build_kernel (sghmc attribute)": [[35, "blackjax.sgmcmc.sghmc.sghmc.build_kernel"]], "build_kernel() (in module blackjax.sgmcmc.sghmc)": [[35, "blackjax.sgmcmc.sghmc.build_kernel"]], "init (sghmc attribute)": [[35, "blackjax.sgmcmc.sghmc.sghmc.init"]], "init() (in module blackjax.sgmcmc.sghmc)": [[35, "blackjax.sgmcmc.sghmc.init"]], "sghmc (class in blackjax.sgmcmc.sghmc)": [[35, "blackjax.sgmcmc.sghmc.sghmc"]], "blackjax.sgmcmc.sgld": [[36, "module-blackjax.sgmcmc.sgld"]], "build_kernel (sgld attribute)": [[36, "blackjax.sgmcmc.sgld.sgld.build_kernel"]], "build_kernel() (in module blackjax.sgmcmc.sgld)": [[36, "blackjax.sgmcmc.sgld.build_kernel"]], "init (sgld attribute)": [[36, "blackjax.sgmcmc.sgld.sgld.init"]], "init() (in module blackjax.sgmcmc.sgld)": [[36, "blackjax.sgmcmc.sgld.init"]], "sgld (class in blackjax.sgmcmc.sgld)": [[36, "blackjax.sgmcmc.sgld.sgld"]], "sgnhtstate (class in blackjax.sgmcmc.sgnht)": [[37, "blackjax.sgmcmc.sgnht.SGNHTState"]], "blackjax.sgmcmc.sgnht": [[37, "module-blackjax.sgmcmc.sgnht"]], "build_kernel (sgnht attribute)": [[37, "blackjax.sgmcmc.sgnht.sgnht.build_kernel"]], "build_kernel() (in module blackjax.sgmcmc.sgnht)": [[37, "blackjax.sgmcmc.sgnht.build_kernel"]], "init (sgnht attribute)": [[37, "blackjax.sgmcmc.sgnht.sgnht.init"]], "init() (in module blackjax.sgmcmc.sgnht)": [[37, "blackjax.sgmcmc.sgnht.init"]], "momentum (sgnhtstate attribute)": [[37, "blackjax.sgmcmc.sgnht.SGNHTState.momentum"]], "position (sgnhtstate attribute)": [[37, "blackjax.sgmcmc.sgnht.SGNHTState.position"]], "sgnht (class in blackjax.sgmcmc.sgnht)": [[37, "blackjax.sgmcmc.sgnht.sgnht"]], "xi (sgnhtstate attribute)": [[37, "blackjax.sgmcmc.sgnht.SGNHTState.xi"]], "adaptive_tempered_smc (class in blackjax.smc.adaptive_tempered)": [[38, "blackjax.smc.adaptive_tempered.adaptive_tempered_smc"]], "blackjax.smc.adaptive_tempered": [[38, "module-blackjax.smc.adaptive_tempered"]], "build_kernel (adaptive_tempered_smc attribute)": [[38, "blackjax.smc.adaptive_tempered.adaptive_tempered_smc.build_kernel"]], "build_kernel() (in module blackjax.smc.adaptive_tempered)": [[38, "blackjax.smc.adaptive_tempered.build_kernel"]], "init (adaptive_tempered_smc attribute)": [[38, "blackjax.smc.adaptive_tempered.adaptive_tempered_smc.init"]], "smcinfo (class in blackjax.smc.base)": [[39, "blackjax.smc.base.SMCInfo"]], "smcstate (class in blackjax.smc.base)": [[39, "blackjax.smc.base.SMCState"]], "ancestors (smcinfo attribute)": [[39, "blackjax.smc.base.SMCInfo.ancestors"]], "blackjax.smc.base": [[39, "module-blackjax.smc.base"]], "init() (in module blackjax.smc.base)": [[39, "blackjax.smc.base.init"]], "log_likelihood_increment (smcinfo attribute)": [[39, "blackjax.smc.base.SMCInfo.log_likelihood_increment"]], "particles (smcstate attribute)": [[39, "blackjax.smc.base.SMCState.particles"]], "step() (in module blackjax.smc.base)": [[39, "blackjax.smc.base.step"]], "update_info (smcinfo attribute)": [[39, "blackjax.smc.base.SMCInfo.update_info"]], "weights (smcstate attribute)": [[39, "blackjax.smc.base.SMCState.weights"]], "blackjax.smc.ess": [[40, "module-blackjax.smc.ess"]], "ess() (in module blackjax.smc.ess)": [[40, "blackjax.smc.ess.ess"]], "ess_solver() (in module blackjax.smc.ess)": [[40, "blackjax.smc.ess.ess_solver"]], "log_ess() (in module blackjax.smc.ess)": [[40, "blackjax.smc.ess.log_ess"]], "blackjax.smc": [[41, "module-blackjax.smc"]], "blackjax.smc.resampling": [[42, "module-blackjax.smc.resampling"]], "multinomial() (in module blackjax.smc.resampling)": [[42, "blackjax.smc.resampling.multinomial"]], "residual() (in module blackjax.smc.resampling)": [[42, "blackjax.smc.resampling.residual"]], "stratified() (in module blackjax.smc.resampling)": [[42, "blackjax.smc.resampling.stratified"]], "systematic() (in module blackjax.smc.resampling)": [[42, "blackjax.smc.resampling.systematic"]], "blackjax.smc.solver": [[43, "module-blackjax.smc.solver"]], "dichotomy() (in module blackjax.smc.solver)": [[43, "blackjax.smc.solver.dichotomy"]], "temperedsmcstate (class in blackjax.smc.tempered)": [[44, "blackjax.smc.tempered.TemperedSMCState"]], "blackjax.smc.tempered": [[44, "module-blackjax.smc.tempered"]], "build_kernel() (in module blackjax.smc.tempered)": [[44, "blackjax.smc.tempered.build_kernel"]], "init() (in module blackjax.smc.tempered)": [[44, "blackjax.smc.tempered.init"]], "lmbda (temperedsmcstate attribute)": [[44, "blackjax.smc.tempered.TemperedSMCState.lmbda"]], "particles (temperedsmcstate attribute)": [[44, "blackjax.smc.tempered.TemperedSMCState.particles"]], "weights (temperedsmcstate attribute)": [[44, "blackjax.smc.tempered.TemperedSMCState.weights"]], "array (in module blackjax.types)": [[45, "blackjax.types.Array"]], "arrayliketree (in module blackjax.types)": [[45, "blackjax.types.ArrayLikeTree"]], "arraytree (in module blackjax.types)": [[45, "blackjax.types.ArrayTree"]], "prngkey (in module blackjax.types)": [[45, "blackjax.types.PRNGKey"]], "blackjax.types": [[45, "module-blackjax.types"]], "blackjax.util": [[46, "module-blackjax.util"]], "generate_gaussian_noise() (in module blackjax.util)": [[46, "blackjax.util.generate_gaussian_noise"]], "generate_unit_vector() (in module blackjax.util)": [[46, "blackjax.util.generate_unit_vector"]], "index_pytree() (in module blackjax.util)": [[46, "blackjax.util.index_pytree"]], "linear_map() (in module blackjax.util)": [[46, "blackjax.util.linear_map"]], "pytree_size() (in module blackjax.util)": [[46, "blackjax.util.pytree_size"]], "blackjax.vi": [[47, "module-blackjax.vi"]], "mfviinfo (class in blackjax.vi.meanfield_vi)": [[48, "blackjax.vi.meanfield_vi.MFVIInfo"]], "mfvistate (class in blackjax.vi.meanfield_vi)": [[48, "blackjax.vi.meanfield_vi.MFVIState"]], "blackjax.vi.meanfield_vi": [[48, "module-blackjax.vi.meanfield_vi"]], "elbo (mfviinfo attribute)": [[48, "blackjax.vi.meanfield_vi.MFVIInfo.elbo"]], "generate_meanfield_logdensity() (in module blackjax.vi.meanfield_vi)": [[48, "blackjax.vi.meanfield_vi.generate_meanfield_logdensity"]], "init (meanfield_vi attribute)": [[48, "blackjax.vi.meanfield_vi.meanfield_vi.init"]], "meanfield_vi (class in blackjax.vi.meanfield_vi)": [[48, "blackjax.vi.meanfield_vi.meanfield_vi"]], "mu (mfvistate attribute)": [[48, "blackjax.vi.meanfield_vi.MFVIState.mu"]], "opt_state (mfvistate attribute)": [[48, "blackjax.vi.meanfield_vi.MFVIState.opt_state"]], "rho (mfvistate attribute)": [[48, "blackjax.vi.meanfield_vi.MFVIState.rho"]], "sample (meanfield_vi attribute)": [[48, "blackjax.vi.meanfield_vi.meanfield_vi.sample"]], "sample() (in module blackjax.vi.meanfield_vi)": [[48, "blackjax.vi.meanfield_vi.sample"]], "step (meanfield_vi attribute)": [[48, "blackjax.vi.meanfield_vi.meanfield_vi.step"]], "step() (in module blackjax.vi.meanfield_vi)": [[48, "blackjax.vi.meanfield_vi.step"]], "pathfinderstate (class in blackjax.vi.pathfinder)": [[49, "blackjax.vi.pathfinder.PathfinderState"]], "alpha (pathfinderstate attribute)": [[49, "blackjax.vi.pathfinder.PathfinderState.alpha"]], "approximate (pathfinder attribute)": [[49, "blackjax.vi.pathfinder.pathfinder.approximate"]], "approximate() (in module blackjax.vi.pathfinder)": [[49, "blackjax.vi.pathfinder.approximate"]], "beta (pathfinderstate attribute)": [[49, "blackjax.vi.pathfinder.PathfinderState.beta"]], "blackjax.vi.pathfinder": [[49, "module-blackjax.vi.pathfinder"]], "elbo (pathfinderstate attribute)": [[49, "blackjax.vi.pathfinder.PathfinderState.elbo"]], "gamma (pathfinderstate attribute)": [[49, "blackjax.vi.pathfinder.PathfinderState.gamma"]], "grad_position (pathfinderstate attribute)": [[49, "blackjax.vi.pathfinder.PathfinderState.grad_position"]], "pathfinder (class in blackjax.vi.pathfinder)": [[49, "blackjax.vi.pathfinder.pathfinder"]], "position (pathfinderstate attribute)": [[49, "blackjax.vi.pathfinder.PathfinderState.position"]], "sample (pathfinder attribute)": [[49, "blackjax.vi.pathfinder.pathfinder.sample"]], "sample() (in module blackjax.vi.pathfinder)": [[49, "blackjax.vi.pathfinder.sample"]], "blackjax.vi.svgd": [[50, "module-blackjax.vi.svgd"]], "build_kernel (svgd attribute)": [[50, "blackjax.vi.svgd.svgd.build_kernel"]], "init (svgd attribute)": [[50, "blackjax.vi.svgd.svgd.init"]], "rbf_kernel() (in module blackjax.vi.svgd)": [[50, "blackjax.vi.svgd.rbf_kernel"]], "svgd (class in blackjax.vi.svgd)": [[50, "blackjax.vi.svgd.svgd"]], "update_median_heuristic() (in module blackjax.vi.svgd)": [[50, "blackjax.vi.svgd.update_median_heuristic"]]}}) \ No newline at end of file +Search.setIndex({"docnames": ["autoapi/blackjax/_version/index", "autoapi/blackjax/adaptation/base/index", "autoapi/blackjax/adaptation/chees_adaptation/index", "autoapi/blackjax/adaptation/index", "autoapi/blackjax/adaptation/mass_matrix/index", "autoapi/blackjax/adaptation/meads_adaptation/index", "autoapi/blackjax/adaptation/pathfinder_adaptation/index", "autoapi/blackjax/adaptation/step_size/index", "autoapi/blackjax/adaptation/window_adaptation/index", "autoapi/blackjax/base/index", "autoapi/blackjax/diagnostics/index", "autoapi/blackjax/index", "autoapi/blackjax/mcmc/diffusions/index", "autoapi/blackjax/mcmc/elliptical_slice/index", "autoapi/blackjax/mcmc/ghmc/index", "autoapi/blackjax/mcmc/hmc/index", "autoapi/blackjax/mcmc/index", "autoapi/blackjax/mcmc/integrators/index", "autoapi/blackjax/mcmc/mala/index", "autoapi/blackjax/mcmc/marginal_latent_gaussian/index", "autoapi/blackjax/mcmc/metrics/index", "autoapi/blackjax/mcmc/nuts/index", "autoapi/blackjax/mcmc/periodic_orbital/index", "autoapi/blackjax/mcmc/proposal/index", "autoapi/blackjax/mcmc/random_walk/index", "autoapi/blackjax/mcmc/termination/index", "autoapi/blackjax/mcmc/trajectory/index", "autoapi/blackjax/optimizers/dual_averaging/index", "autoapi/blackjax/optimizers/index", "autoapi/blackjax/optimizers/lbfgs/index", "autoapi/blackjax/progress_bar/index", "autoapi/blackjax/sgmcmc/csgld/index", "autoapi/blackjax/sgmcmc/diffusions/index", "autoapi/blackjax/sgmcmc/gradients/index", "autoapi/blackjax/sgmcmc/index", "autoapi/blackjax/sgmcmc/sghmc/index", "autoapi/blackjax/sgmcmc/sgld/index", "autoapi/blackjax/sgmcmc/sgnht/index", "autoapi/blackjax/smc/adaptive_tempered/index", "autoapi/blackjax/smc/base/index", "autoapi/blackjax/smc/ess/index", "autoapi/blackjax/smc/index", "autoapi/blackjax/smc/resampling/index", "autoapi/blackjax/smc/solver/index", "autoapi/blackjax/smc/tempered/index", "autoapi/blackjax/types/index", "autoapi/blackjax/util/index", "autoapi/blackjax/vi/index", "autoapi/blackjax/vi/meanfield_vi/index", "autoapi/blackjax/vi/pathfinder/index", "autoapi/blackjax/vi/svgd/index", "bib", "examples/howto_custom_gradients", "examples/howto_metropolis_within_gibbs", "examples/howto_other_frameworks", "examples/howto_sample_multiple_chains", "examples/howto_use_aesara", "examples/howto_use_numpyro", "examples/howto_use_oryx", "examples/howto_use_pymc", "examples/howto_use_tfp", "examples/quickstart", "index"], "filenames": ["autoapi/blackjax/_version/index.rst", "autoapi/blackjax/adaptation/base/index.rst", "autoapi/blackjax/adaptation/chees_adaptation/index.rst", "autoapi/blackjax/adaptation/index.rst", "autoapi/blackjax/adaptation/mass_matrix/index.rst", "autoapi/blackjax/adaptation/meads_adaptation/index.rst", "autoapi/blackjax/adaptation/pathfinder_adaptation/index.rst", "autoapi/blackjax/adaptation/step_size/index.rst", "autoapi/blackjax/adaptation/window_adaptation/index.rst", "autoapi/blackjax/base/index.rst", "autoapi/blackjax/diagnostics/index.rst", "autoapi/blackjax/index.rst", "autoapi/blackjax/mcmc/diffusions/index.rst", "autoapi/blackjax/mcmc/elliptical_slice/index.rst", "autoapi/blackjax/mcmc/ghmc/index.rst", "autoapi/blackjax/mcmc/hmc/index.rst", "autoapi/blackjax/mcmc/index.rst", "autoapi/blackjax/mcmc/integrators/index.rst", "autoapi/blackjax/mcmc/mala/index.rst", "autoapi/blackjax/mcmc/marginal_latent_gaussian/index.rst", "autoapi/blackjax/mcmc/metrics/index.rst", "autoapi/blackjax/mcmc/nuts/index.rst", "autoapi/blackjax/mcmc/periodic_orbital/index.rst", "autoapi/blackjax/mcmc/proposal/index.rst", "autoapi/blackjax/mcmc/random_walk/index.rst", "autoapi/blackjax/mcmc/termination/index.rst", "autoapi/blackjax/mcmc/trajectory/index.rst", "autoapi/blackjax/optimizers/dual_averaging/index.rst", "autoapi/blackjax/optimizers/index.rst", "autoapi/blackjax/optimizers/lbfgs/index.rst", "autoapi/blackjax/progress_bar/index.rst", "autoapi/blackjax/sgmcmc/csgld/index.rst", "autoapi/blackjax/sgmcmc/diffusions/index.rst", "autoapi/blackjax/sgmcmc/gradients/index.rst", "autoapi/blackjax/sgmcmc/index.rst", "autoapi/blackjax/sgmcmc/sghmc/index.rst", "autoapi/blackjax/sgmcmc/sgld/index.rst", "autoapi/blackjax/sgmcmc/sgnht/index.rst", "autoapi/blackjax/smc/adaptive_tempered/index.rst", "autoapi/blackjax/smc/base/index.rst", "autoapi/blackjax/smc/ess/index.rst", "autoapi/blackjax/smc/index.rst", "autoapi/blackjax/smc/resampling/index.rst", "autoapi/blackjax/smc/solver/index.rst", "autoapi/blackjax/smc/tempered/index.rst", "autoapi/blackjax/types/index.rst", "autoapi/blackjax/util/index.rst", "autoapi/blackjax/vi/index.rst", "autoapi/blackjax/vi/meanfield_vi/index.rst", "autoapi/blackjax/vi/pathfinder/index.rst", "autoapi/blackjax/vi/svgd/index.rst", "bib.rst", "examples/howto_custom_gradients.md", "examples/howto_metropolis_within_gibbs.md", "examples/howto_other_frameworks.md", "examples/howto_sample_multiple_chains.md", "examples/howto_use_aesara.md", "examples/howto_use_numpyro.md", "examples/howto_use_oryx.md", "examples/howto_use_pymc.md", "examples/howto_use_tfp.md", "examples/quickstart.md", "index.md"], "titles": ["blackjax._version", "blackjax.adaptation.base", "blackjax.adaptation.chees_adaptation", "blackjax.adaptation", "blackjax.adaptation.mass_matrix", "blackjax.adaptation.meads_adaptation", "blackjax.adaptation.pathfinder_adaptation", "blackjax.adaptation.step_size", "blackjax.adaptation.window_adaptation", "blackjax.base", "blackjax.diagnostics", "blackjax", "blackjax.mcmc.diffusions", "blackjax.mcmc.elliptical_slice", "blackjax.mcmc.ghmc", "blackjax.mcmc.hmc", "blackjax.mcmc", "blackjax.mcmc.integrators", "blackjax.mcmc.mala", "blackjax.mcmc.marginal_latent_gaussian", "blackjax.mcmc.metrics", "blackjax.mcmc.nuts", "blackjax.mcmc.periodic_orbital", "blackjax.mcmc.proposal", "blackjax.mcmc.random_walk", "blackjax.mcmc.termination", "blackjax.mcmc.trajectory", "blackjax.optimizers.dual_averaging", "blackjax.optimizers", "blackjax.optimizers.lbfgs", "blackjax.progress_bar", "blackjax.sgmcmc.csgld", "blackjax.sgmcmc.diffusions", "blackjax.sgmcmc.gradients", "blackjax.sgmcmc", "blackjax.sgmcmc.sghmc", "blackjax.sgmcmc.sgld", "blackjax.sgmcmc.sgnht", "blackjax.smc.adaptive_tempered", "blackjax.smc.base", "blackjax.smc.ess", "blackjax.smc", "blackjax.smc.resampling", "blackjax.smc.solver", "blackjax.smc.tempered", "blackjax.types", "blackjax.util", "blackjax.vi", "blackjax.vi.meanfield_vi", "blackjax.vi.pathfinder", "blackjax.vi.svgd", "Bibliography", "Use custom gradients", "How to build a Metropolis-Within-Gibbs sampler?", "Use a logdensity function that is not compatible with JAX\u2019s primitives", "Sample with multiple chains in parallel", "Use with Aesara models", "Use with Numpyro models", "Use with Oryx models", "Use with PyMC models", "Use with TFP models", "A Quick Introduction to Blackjax", "Welcome to Blackjax!"], "terms": {"type_check": 0, "fals": [0, 8, 19, 26, 57, 60], "sourc": [0, 1, 2, 4, 5, 6, 7, 8, 9, 10, 12, 13, 14, 15, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 29, 30, 31, 32, 33, 35, 36, 37, 38, 39, 40, 42, 43, 44, 45, 46, 48, 49, 50, 51], "version_tupl": 0, "version": [0, 14, 19, 20, 21, 24, 57, 59, 60, 62], "str": 0, "adaptationresult": [1, 8], "state": [1, 2, 4, 5, 6, 7, 8, 9, 13, 14, 15, 18, 19, 21, 22, 23, 24, 26, 27, 31, 35, 36, 37, 38, 39, 40, 44, 48, 49, 50, 52, 54, 55, 56, 57, 58, 59, 60, 62], "type": [1, 2, 4, 5, 6, 7, 8, 10, 13, 14, 15, 18, 19, 20, 21, 22, 23, 24, 25, 26, 29, 31, 33, 35, 36, 37, 38, 39, 40, 42, 43, 44, 46, 48, 49, 50, 53, 56], "arraytre": [1, 5, 13, 14, 15, 18, 21, 22, 24, 26, 31, 37, 39, 44, 45, 46, 48, 49], "paramet": [1, 2, 4, 5, 6, 7, 8, 9, 10, 13, 14, 15, 17, 18, 19, 20, 21, 22, 23, 26, 27, 29, 31, 32, 33, 35, 36, 37, 38, 39, 40, 43, 44, 46, 48, 49, 50, 51, 55, 56, 57, 58, 59, 60], "dict": [1, 22, 53, 58], "adaptationinfo": 1, "namedtupl": [1, 26, 39, 53, 60], "info": [1, 2, 9, 13, 14, 15, 18, 19, 21, 22, 24, 35, 36, 39, 52, 54, 56, 57, 58, 59, 60], "adaptation_st": 1, "public": [2, 13, 14, 15, 18, 19, 21, 22, 31, 35, 36, 37], "api": [2, 13, 14, 15, 18, 19, 21, 22, 31, 35, 36, 37, 54, 59], "chee": [2, 55], "hmc": [2, 5, 6, 7, 8, 11, 14, 16, 26, 35, 46, 51, 52, 53, 55, 60], "optimal_target_acceptance_r": 2, "0": [2, 5, 6, 7, 8, 10, 14, 22, 23, 27, 31, 32, 35, 37, 39, 43, 46, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62], "651": 2, "cheesadaptationst": 2, "scheme": [2, 5, 6, 7, 8, 27, 51], "step_siz": [2, 3, 5, 6, 8, 11, 14, 15, 17, 18, 21, 22, 26, 35, 36, 37, 53, 54, 55, 61, 62], "valu": [2, 4, 5, 6, 7, 8, 10, 13, 14, 15, 17, 18, 19, 20, 21, 22, 23, 24, 26, 27, 29, 31, 33, 40, 44, 51, 52, 53, 56, 58, 60, 61], "algorithm": [2, 4, 5, 6, 7, 8, 9, 13, 14, 15, 18, 21, 22, 24, 26, 27, 29, 31, 32, 33, 35, 36, 37, 39, 40, 44, 46, 49, 50, 51, 55, 56, 58, 61, 62], "log_step_size_moving_averag": 2, "run": [2, 4, 6, 8, 9, 14, 15, 26, 29, 31, 54, 55, 56, 57, 58, 59, 60, 61, 62], "move": [2, 5, 6, 8, 24, 55], "averag": [2, 7, 8, 21, 27, 57, 58, 60], "log": [2, 5, 6, 7, 8, 13, 14, 15, 18, 19, 21, 22, 24, 27, 31, 33, 35, 36, 37, 38, 39, 40, 44, 48, 49, 50, 52, 53, 54, 56, 57, 58, 59, 60, 61, 62], "trajectory_length": 2, "num_integration_step": [2, 15, 21, 35, 53, 57, 60, 61], "log_trajectory_length_moving_averag": 2, "optim_st": 2, "optax": [2, 48, 50], "optim": [2, 6, 31, 48, 49, 50, 51, 52, 53, 54], "us": [2, 4, 5, 6, 7, 8, 9, 10, 13, 14, 15, 17, 18, 19, 20, 21, 22, 23, 24, 26, 27, 29, 30, 38, 39, 43, 44, 46, 48, 49, 51, 53, 62], "maxim": [2, 5], "criterion": [2, 10, 25, 26, 52], "random_generator_arg": [2, 15], "util": [2, 15, 38, 57], "arrai": [2, 4, 5, 6, 8, 10, 13, 15, 19, 20, 22, 23, 24, 25, 29, 31, 39, 40, 42, 43, 44, 45, 46, 49, 52, 53, 54, 55, 56, 57, 59, 60, 61, 62], "gener": [2, 5, 7, 8, 14, 15, 18, 20, 21, 23, 24, 26, 35, 36, 37, 38, 39, 44, 46, 48, 51, 55, 56, 57, 58, 61], "pseudo": [2, 15, 39, 46, 48], "quasi": [2, 15, 29, 49, 51], "random": [2, 7, 9, 14, 15, 20, 21, 22, 24, 26, 38, 39, 44, 46, 48, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62], "sequenc": [2, 10, 15, 51], "number": [2, 4, 5, 6, 7, 8, 10, 13, 15, 21, 22, 24, 26, 29, 31, 33, 35, 38, 39, 44, 46, 48, 49, 53, 56, 57, 58, 60, 61], "step": [2, 4, 5, 6, 7, 8, 9, 13, 14, 15, 18, 19, 21, 22, 23, 24, 26, 27, 29, 30, 31, 35, 36, 37, 38, 39, 40, 48, 49, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62], "current": [2, 4, 5, 7, 9, 10, 13, 14, 15, 18, 19, 20, 21, 22, 23, 24, 26, 27, 31, 33, 37, 38, 39, 40, 44, 50, 55, 62], "iter": [2, 4, 7, 13, 14, 21, 22, 27, 29, 35, 36, 37, 43, 44, 49, 51, 54, 62], "float": [2, 5, 6, 7, 8, 13, 14, 15, 18, 19, 21, 23, 24, 26, 27, 29, 32, 35, 37, 38, 39, 40, 43, 44, 46, 48], "da_stat": 2, "dual_averag": [2, 28], "dualaveragingst": [2, 27], "optstat": [2, 48], "int": [2, 4, 5, 7, 8, 10, 13, 15, 21, 22, 23, 25, 26, 27, 29, 31, 33, 39, 42, 43, 46, 48, 49, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61], "base": [2, 3, 4, 5, 6, 8, 11, 15, 18, 19, 21, 23, 29, 38, 41, 44, 51, 52, 56, 59, 61], "jitter_gener": 2, "callabl": [2, 4, 5, 6, 7, 8, 9, 13, 14, 15, 18, 20, 21, 22, 23, 24, 26, 27, 29, 31, 33, 35, 36, 37, 38, 39, 40, 43, 44, 48, 49], "next_random_arg_fn": [2, 15], "gradienttransform": [2, 48, 50], "target_acceptance_r": [2, 6, 8, 57], "decay_r": 2, "tupl": [2, 4, 7, 8, 20, 22, 23, 26, 27, 29, 33, 39, 48, 49, 56], "chang": [2, 8], "estim": [2, 4, 5, 6, 7, 8, 10, 31, 33, 35, 36, 37, 48, 49, 50, 51, 58], "expect": [2, 55], "squar": [2, 4, 5, 46, 50], "trajectori": [2, 8, 11, 14, 15, 16, 17, 20, 21, 23, 51, 61], "length": [2, 14, 17, 21, 51], "dual": [2, 7, 8, 27, 51], "procedur": [2, 6, 7, 8, 9, 14, 26, 27], "size": [2, 5, 6, 7, 8, 10, 14, 15, 18, 19, 21, 22, 26, 27, 29, 35, 36, 37, 38, 39, 40, 49, 51, 55, 56, 58, 60, 61, 62], "jitter": 2, "hamiltonian": [2, 4, 5, 15, 17, 20, 22, 26, 35, 51, 61], "mont": [2, 4, 5, 35, 51, 61], "carlo": [2, 4, 5, 35, 51, 61], "kernel": [2, 5, 6, 7, 8, 9, 13, 14, 15, 18, 19, 21, 22, 24, 31, 35, 36, 37, 38, 39, 44, 49, 50, 55, 56, 57, 58, 59, 60, 62], "hrs21": [2, 51], "thi": [2, 4, 5, 6, 7, 8, 9, 10, 13, 14, 15, 18, 19, 21, 23, 24, 26, 27, 31, 32, 33, 35, 36, 37, 38, 39, 46, 50, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61], "tune": [2, 4, 5, 6, 7, 8, 51, 57, 59, 60], "i": [2, 4, 5, 6, 7, 8, 9, 10, 13, 14, 15, 17, 19, 20, 21, 22, 23, 24, 26, 27, 29, 31, 38, 39, 40, 43, 44, 46, 48, 49, 50, 51, 53, 55, 56, 57, 58, 59, 60, 61, 62], "e": [2, 6, 7, 8, 23, 31, 52, 53, 55, 56], "integr": [2, 5, 6, 8, 11, 14, 15, 16, 18, 21, 22, 26, 35, 51, 53, 55, 60, 61, 62], "statist": [2, 5, 10, 51], "collect": [2, 5, 22], "from": [2, 4, 5, 6, 7, 8, 9, 10, 13, 14, 15, 18, 21, 22, 23, 24, 26, 29, 30, 32, 33, 38, 39, 40, 44, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61], "popul": [2, 4, 5], "mani": [2, 5, 14, 52, 55], "chain": [2, 5, 6, 7, 9, 10, 13, 14, 15, 18, 19, 20, 22, 23, 24, 38, 44, 51, 60], "It": [2, 4, 5, 8, 19, 20, 23, 29, 46, 48, 55, 61, 62], "target": [2, 5, 6, 7, 8, 9, 10, 14, 22, 27, 29, 33, 38, 40, 48, 49, 50, 58], "an": [2, 4, 5, 6, 7, 8, 9, 10, 13, 15, 18, 19, 20, 21, 22, 23, 24, 27, 31, 35, 36, 37, 38, 39, 43, 44, 46, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61], "accept": [2, 5, 6, 7, 8, 13, 14, 15, 18, 19, 21, 23, 24, 27, 31, 40, 51, 56, 57, 60], "rate": [2, 6, 7, 8, 14, 18, 19, 26, 27, 57, 60], "harmon": 2, "mean": [2, 4, 13, 14, 19, 24, 46, 48, 53, 55, 57, 58, 60], "": [2, 4, 5, 7, 8, 9, 10, 14, 20, 23, 24, 29, 30, 31, 33, 48, 51, 52, 53, 55, 56, 57, 58, 59, 60, 62], "probabl": [2, 5, 6, 7, 8, 9, 15, 18, 19, 20, 22, 23, 24, 33, 38, 40, 44, 53, 54, 56, 57, 58, 59, 60, 62], "option": [2, 19, 23, 38, 40, 55], "1": [2, 4, 6, 7, 8, 10, 20, 22, 23, 24, 26, 27, 29, 31, 39, 46, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62], "given": [2, 5, 6, 7, 8, 14, 15, 19, 20, 22, 23, 24, 27, 31, 38, 44, 46, 50, 53, 54, 61], "prngkei": [2, 7, 14, 23, 37, 39, 42, 45, 46, 48, 49, 58], "propos": [2, 5, 7, 11, 13, 15, 16, 18, 19, 21, 24, 26, 39], "If": [2, 7, 23, 24, 26, 31, 43, 54, 62], "none": [2, 19, 24, 30, 39, 53, 55, 56, 57, 58, 59, 60], "halton": 2, "next": [2, 4, 9, 14, 15, 23, 35, 36, 37, 40, 55], "its": [2, 5, 14, 15, 22, 24, 27, 46, 54, 56, 58], "previou": [2, 4, 8, 13, 14, 15, 22, 23, 24, 53], "compat": [2, 50, 56, 57, 58, 59, 60, 62], "which": [2, 5, 6, 7, 8, 9, 10, 14, 15, 17, 19, 20, 21, 23, 24, 26, 31, 35, 36, 37, 39, 44, 46, 50, 52, 53, 54, 56, 57], "conform": [2, 50], "protocol": [2, 23, 50], "repres": [2, 9, 10, 22, 39, 48, 49], "how": [2, 46, 51, 52, 54, 55, 56, 57, 58, 59, 60], "much": [2, 55], "favor": [2, 23], "recent": [2, 55], "over": [2, 9, 53, 58], "earlier": [2, 7, 27, 55], "ones": [2, 17, 39, 52, 54, 55, 58, 60], "return": [2, 4, 5, 6, 7, 8, 9, 10, 13, 14, 15, 18, 19, 20, 21, 22, 23, 24, 26, 27, 29, 31, 35, 36, 37, 38, 39, 40, 43, 44, 46, 48, 49, 50, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62], "init": [2, 4, 5, 6, 7, 8, 9, 13, 14, 15, 18, 19, 21, 22, 24, 27, 31, 35, 36, 37, 38, 39, 44, 48, 49, 50, 52, 53, 54, 55, 61, 62], "initi": [2, 4, 5, 6, 7, 8, 9, 10, 13, 14, 15, 18, 19, 21, 22, 23, 24, 26, 27, 29, 35, 36, 37, 48, 55, 56, 57, 58, 59, 62], "warmup": [2, 5, 6, 7, 8, 61], "updat": [2, 4, 5, 6, 7, 8, 9, 14, 23, 26, 27, 29, 38, 39, 50, 51, 53, 59], "one": [2, 5, 6, 8, 10, 14, 15, 18, 22, 23, 24, 26, 35, 36, 37, 39, 52, 54, 55], "logdensity_fn": [2, 5, 6, 8, 13, 14, 15, 18, 19, 21, 22, 24, 40, 48, 49, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62], "num_chain": [2, 5, 55], "jitter_amount": 2, "5": [2, 7, 26, 27, 39, 48, 52, 54, 55, 56, 57, 59, 60, 61], "adaptationalgorithm": [2, 5, 6, 8, 9], "algorthm": 2, "The": [2, 4, 5, 6, 7, 8, 9, 10, 13, 14, 15, 17, 18, 19, 20, 21, 22, 23, 24, 26, 27, 29, 31, 33, 35, 36, 37, 38, 39, 40, 43, 44, 46, 48, 49, 51, 52, 54, 55, 57, 58, 60, 62], "depend": [2, 5, 8, 13, 14, 20, 24, 26, 56, 62], "control": [2, 5, 7, 27, 31, 33, 37, 51], "discret": [2, 22], "onli": [2, 8, 13, 15, 18, 19, 20, 23, 35, 36, 37, 52, 54, 55, 56, 61, 62], "percentag": 2, "heurist": [2, 5, 7, 50], "ensambl": 2, "parallel": [2, 5, 14, 29, 51], "At": [2, 55], "equilibrium": 2, "aim": [2, 38], "elimin": 2, "correl": [2, 5], "between": [2, 4, 5, 7, 8, 9, 13, 14, 15, 19, 21, 23, 24, 26, 27, 31, 48, 50, 54, 55, 56], "dimens": [2, 5, 10, 14, 46], "make": [2, 5, 14, 15, 18, 26, 54, 55, 56, 57, 62], "effici": [2, 13, 14, 15, 18, 19, 31, 52, 53], "requir": [2, 8, 55, 57, 61], "uniform": [2, 23, 51, 56], "variabl": [2, 5, 13, 14, 20, 22, 24, 33, 39, 46, 53, 54, 55, 56, 57, 58], "howev": [2, 4, 53, 56], "add": [2, 15, 18, 51, 56], "anoth": [2, 15, 18, 22, 23, 55], "varianc": [2, 4, 5, 10, 48, 51, 56], "sampl": [2, 4, 5, 6, 7, 8, 9, 10, 13, 14, 15, 18, 19, 21, 22, 23, 24, 26, 29, 31, 35, 36, 37, 38, 39, 40, 44, 46, 48, 49, 50, 51, 54, 56, 57, 58, 59, 60, 61], "mai": [2, 7, 27, 52, 54, 55, 62], "slow": [2, 7, 8, 26, 27, 55], "lead": [2, 56], "suboptim": 2, "mix": [2, 5, 61], "To": [2, 9, 14, 26, 35, 36, 37, 53, 54, 55, 56, 57, 58], "allevi": 2, "rather": [2, 52], "than": [2, 5, 8, 10, 23, 31, 52, 55], "nois": [2, 14, 46], "we": [2, 5, 6, 7, 8, 13, 14, 15, 18, 19, 20, 21, 22, 24, 26, 31, 35, 36, 37, 38, 39, 40, 43, 46, 48, 49, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62], "ensur": [2, 5, 8, 22, 26, 52], "more": [2, 8, 10, 13, 14, 15, 17, 18, 20, 21, 38, 46, 55], "even": 2, "distribut": [2, 13, 20, 22, 24, 29, 33, 44, 46, 48, 49, 51, 53, 54, 55, 56, 57, 58, 60, 61], "exampl": [2, 13, 14, 15, 18, 19, 21, 22, 23, 24, 35, 36, 37, 39, 46, 52, 53, 54, 56, 57, 58, 59, 60, 61], "can": [2, 7, 8, 9, 13, 14, 15, 18, 19, 20, 21, 22, 23, 24, 31, 35, 36, 37, 39, 43, 46, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62], "learn": [2, 33, 51], "follow": [2, 4, 7, 8, 13, 14, 15, 18, 19, 20, 21, 22, 24, 26, 39, 46, 52, 53, 54, 55, 56, 57, 59, 60, 62], "code": [2, 13, 14, 15, 18, 19, 21, 22, 24, 53, 55, 62], "key_warmup": 2, "key_sampl": 2, "jax": [2, 7, 13, 14, 15, 18, 19, 20, 21, 22, 24, 30, 35, 36, 37, 39, 46, 48, 51, 53, 56, 57, 58, 59, 60, 61, 62], "split": [2, 8, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62], "rng_kei": [2, 7, 9, 13, 14, 15, 18, 19, 21, 22, 23, 24, 29, 35, 36, 37, 38, 39, 42, 44, 46, 48, 49, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62], "adam": [2, 51], "learning_r": 2, "last_stat": [2, 57, 58, 59, 60], "_": [2, 7, 10, 14, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62], "posit": [2, 5, 9, 10, 13, 14, 15, 18, 19, 20, 21, 22, 24, 29, 31, 33, 35, 36, 37, 38, 44, 46, 49, 50, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61], "pytre": [2, 5, 13, 14, 15, 18, 19, 20, 22, 24, 29, 39, 44, 46], "where": [2, 4, 7, 8, 10, 15, 22, 27, 29, 31, 44, 52, 53, 54, 55], "each": [2, 5, 8, 10, 14, 15, 20, 22, 23, 35, 36, 37, 38, 39, 46, 48, 55, 60], "leaf": [2, 5], "ha": [2, 7, 8, 21, 23, 26, 27, 33, 48, 53, 54, 55], "shape": [2, 5, 43, 46, 53, 55, 58, 59], "initial_step_s": [2, 6, 7, 8], "num_warmup_step": 2, "dynamic_hmc": [2, 15], "new_stat": [2, 13, 14, 15, 18, 19, 21, 22, 24, 37, 52, 61], "vmap": [2, 39, 56, 58], "densiti": [2, 5, 6, 8, 9, 14, 15, 18, 19, 20, 21, 22, 23, 24, 31, 33, 35, 36, 37, 38, 44, 48, 49, 50, 52, 53, 56, 58, 60, 62], "wish": [2, 5, 6, 8, 13, 14, 15, 18, 21, 22, 24, 38, 40], "cross": [2, 5, 7], "warm": [2, 5], "up": [2, 5, 7], "train": [2, 5, 58], "jitter_valu": 2, "A": [2, 4, 5, 6, 7, 8, 9, 13, 14, 15, 18, 19, 20, 21, 22, 23, 24, 26, 27, 31, 33, 35, 36, 37, 38, 40, 44, 46, 48, 49, 50, 51, 53, 54, 56], "calcul": [2, 5, 22, 29, 53], "should": [2, 8, 15, 18, 20, 21, 26, 29, 31, 46, 52, 53, 54, 55], "jit": [2, 13, 14, 15, 18, 19, 21, 22, 24, 26, 35, 36, 37, 53, 54, 55, 56, 57, 61, 62], "default": [2, 4, 7, 10, 14, 15, 18, 19, 21, 27, 35, 36, 37, 38, 40, 46, 53, 55, 61, 62], "give": [2, 5, 8, 52, 53], "equal": [2, 5, 17, 22, 23, 24, 31, 55], "weight": [2, 7, 22, 23, 27, 38, 39, 40, 42, 44, 51, 54, 58], "all": [2, 5, 9, 22, 24, 40, 42, 43, 46, 49, 53, 55, 56, 57], "histori": [2, 29, 49], "most": [2, 15, 18, 20, 35, 36, 37, 52, 53, 54], "last": [2, 5, 6, 8, 10, 26], "diagnost": [2, 5, 11, 13, 15, 18, 19, 21, 24, 55], "chees_adapt": [3, 11], "mass_matrix": [3, 8, 11], "meads_adapt": [3, 11], "pathfinder_adapt": [3, 11], "window_adapt": [3, 11, 56, 57, 58, 59, 60, 61], "mass": [4, 6, 7, 8, 15, 20, 21, 22, 46, 60, 61], "matrix": [4, 5, 6, 7, 8, 13, 15, 20, 21, 22, 24, 46, 60, 61], "famili": [4, 6, 8, 26], "geometri": 4, "stan": [4, 8, 10, 51], "manual": [4, 15, 18, 35, 36, 37, 51, 53, 54], "stab": [4, 8, 51], "veri": [4, 8, 19, 52, 55], "good": [4, 5, 7, 8, 14, 19, 52, 54, 55, 58, 60], "refer": [4, 7, 24, 26, 51, 53, 60], "automat": [4, 5, 52, 53, 55, 56], "welfordalgorithmst": 4, "carri": [4, 7, 9, 27], "through": [4, 7, 14, 27, 52], "welford": 4, "m2": 4, "sum": [4, 5, 23, 31, 52, 55, 56, 58, 61, 62], "differ": [4, 7, 14, 15, 17, 19, 21, 23, 24, 26, 27, 31, 55, 61, 62], "see": [4, 7, 8, 23, 24, 27, 46, 54, 55, 59, 61], "document": [4, 57, 58, 59, 60, 62], "welford_algorithm": 4, "explan": [4, 8, 27], "sample_s": 4, "success": [4, 8, 31], "have": [4, 7, 9, 10, 14, 19, 21, 35, 36, 37, 43, 52, 54, 55, 57, 61], "been": [4, 21, 23, 26], "comput": [4, 5, 6, 8, 9, 10, 13, 14, 15, 18, 19, 21, 22, 23, 24, 26, 38, 40, 44, 46, 48, 51, 52, 54, 55, 56, 57, 60], "also": [4, 8, 9, 14, 15, 17, 18, 20, 21, 26, 52, 54, 55, 57, 61], "massmatrixadaptationst": [4, 8], "inverse_mass_matrix": [4, 6, 7, 8, 15, 20, 21, 22, 53, 54, 62], "curent": 4, "invers": [4, 6, 7, 8, 15, 20, 21, 22, 29, 46, 49, 60, 61], "wc_state": 4, "mass_matrix_adapt": 4, "is_diagonal_matrix": 4, "bool": [4, 8, 15, 18, 19, 21, 24, 26, 38, 48], "true": [4, 8, 38, 48, 49, 52, 53, 54, 55, 57, 61], "covari": [4, 5, 8, 13, 19, 24, 49], "when": [4, 5, 9, 10, 15, 20, 21, 22, 23, 24, 29, 38, 46, 49, 53, 55], "diagon": [4, 8, 20, 29, 46], "otherwis": [4, 8, 55], "adap": [4, 8], "dens": [4, 8, 20, 46, 58], "final": [4, 6, 8, 27, 55, 56, 57, 58], "onlin": 4, "possibl": [4, 5, 8, 55], "line": [4, 29, 49, 61], "fashion": 4, "avoid": [4, 9, 52, 53, 54], "store": [4, 14, 15, 18, 20, 49], "intermedi": 4, "result": [4, 10, 29, 46, 48, 49, 55, 57], "naiv": 4, "recurr": 4, "relat": [4, 8, 40, 62], "ar": [4, 5, 6, 7, 8, 9, 14, 15, 17, 18, 22, 23, 24, 26, 27, 29, 33, 35, 36, 37, 48, 52, 53, 54, 55, 56, 60, 61, 62], "numer": [4, 46, 51, 52], "stabl": [4, 8, 38, 59], "m_": [4, 39], "2": [4, 6, 7, 8, 10, 20, 24, 26, 29, 39, 40, 46, 52, 53, 54, 56, 57, 58, 61, 62], "n": [4, 10, 19, 39, 46, 54, 56], "sum_": [4, 10], "left": [4, 7, 62], "x_i": 4, "overlin": [4, 7], "x_n": 4, "right": [4, 7, 26, 53], "relationship": 4, "might": [4, 24, 55], "seem": [4, 21], "pedant": 4, "separ": [4, 8, 53], "other": [4, 8, 9, 14, 15, 21, 29, 49, 51, 52, 53, 54, 55], "part": [4, 56], "librari": [4, 8, 14, 15, 21, 51, 54, 55, 56, 57, 58, 59, 60, 61, 62], "meadsadaptationst": 5, "mead": [5, 55], "position_sigma": 5, "contain": [5, 9, 10, 13, 14, 15, 18, 19, 22, 24, 29, 38, 39, 44, 49, 53, 58, 60], "per": [5, 29, 38, 49], "standard": [5, 14, 20, 24, 46], "deviat": [5, 24, 46], "scale": [5, 10, 32, 51, 54, 55, 61, 62], "momentum": [5, 13, 14, 15, 20, 21, 22, 25, 26, 37, 46], "alpha": [5, 14, 29, 32, 35, 37, 49, 56], "delta": [5, 7, 14, 19, 31, 38, 40, 43], "current_iter": 5, "maximum": [5, 21, 26, 29, 43, 49], "eigenvalu": 5, "damp": 5, "hs22": [5, 51], "perform": [5, 8, 13, 14, 19, 22, 23, 24, 26, 35, 36, 37, 46, 49, 53, 54, 55, 57, 59, 60], "select": [5, 39], "tunabl": 5, "determin": [5, 17, 20, 26, 31, 61], "gradient": [5, 11, 14, 15, 18, 19, 22, 29, 31, 34, 35, 36, 37, 38, 48, 49, 50, 51, 54, 56, 61], "matric": 5, "group": 5, "implement": [5, 6, 8, 9, 10, 13, 14, 15, 18, 19, 21, 22, 24, 26, 29, 31, 35, 36, 37, 38, 39, 48, 49, 50, 52, 53, 54, 56, 57, 58, 59, 60, 61, 62], "3": [5, 7, 8, 26, 29, 35, 36, 37, 39, 49, 51, 52, 54, 55, 56, 57, 58, 59, 60, 61, 62], "instead": [5, 14, 26, 55, 58], "ensampl": 5, "three": [5, 8, 17, 39], "element": [5, 29, 46, 53, 56], "behaviour": 5, "dynam": [5, 15, 20, 25, 26, 31, 36, 46, 51, 52, 61], "persist": [5, 14], "determinist": [5, 14], "transform": [5, 14, 22, 55, 56, 57, 58, 61], "slice": [5, 13, 14, 21, 23, 51], "non": [5, 14, 23, 24, 31, 51, 52, 57, 59, 60], "revers": [5, 14, 17, 23, 26, 52], "metropoli": [5, 7, 14, 18, 23, 24, 51], "hast": [5, 14, 23, 24, 51], "reject": [5, 14, 23, 51], "chosen": [5, 7, 15, 27, 52, 55], "stabil": [5, 7, 17, 27, 52], "veloc": [5, 14], "verlet": [5, 14], "influenc": 5, "futur": [5, 46], "decai": 5, "exponenti": 5, "properti": [5, 27, 31], "These": [5, 54], "characterist": 5, "under": 5, "simpifyng": 5, "assumpt": 5, "fundament": 5, "explor": 5, "space": [5, 14, 20, 31, 37, 38, 56, 57, 61], "efficienti": 5, "output": [5, 14, 22, 46, 52, 57, 61], "uncorrel": 5, "furthermor": 5, "singl": [5, 10, 14, 22, 33, 39, 55, 56], "lend": 5, "itself": [5, 20, 21], "fast": [5, 8, 62], "architectur": [5, 61], "maximum_eigenvalu": 5, "arrayliketre": [5, 13, 14, 15, 18, 22, 29, 31, 33, 35, 36, 37, 39, 40, 44, 45, 46, 49], "largest": 5, "unbias": 5, "ratio": 5, "input": [5, 7, 14, 46], "approxim": [5, 9, 17, 19, 29, 33, 46, 48, 49, 50, 51], "well": [5, 6, 14, 15, 18, 51, 54, 61, 62], "except": [5, 52, 54, 55], "case": [5, 8, 20, 24, 39, 52, 53, 55], "larg": [5, 7, 27, 53], "small": [5, 7, 27, 53], "significantli": 5, "larger": [5, 55], "smaller": [5, 31], "directli": [5, 15, 18, 52, 55, 57], "becaus": [5, 21, 52, 54, 55], "latter": [5, 26, 53], "batch": [5, 9, 31, 33, 35, 36, 37, 39], "first": [5, 7, 8, 19, 26, 33, 39, 51, 52, 55, 60], "everi": [5, 9, 21, 55], "flatten": [5, 20, 31, 46], "dimension": [5, 20, 24, 46], "stack": [5, 54], "vertic": 5, "row": [5, 22, 46], "pathind": 6, "pathfinderadaptationst": 6, "ss_state": [6, 8], "dualaveragingadaptationst": [6, 7, 8], "8": [6, 8, 53, 54, 55, 56, 57, 58, 59, 60], "euclidean": [6, 8, 20, 61], "manifold": [6, 8, 20, 51], "two": [6, 8, 14, 15, 17, 20, 21, 23, 24, 26, 33, 39, 53, 55, 56, 62], "pathfind": [6, 11, 29, 47, 51], "ran": 6, "subsequ": [6, 26, 54, 56], "new": [6, 8, 9, 13, 14, 15, 18, 19, 21, 22, 23, 24, 26, 38, 39, 44, 55, 61, 62], "point": [6, 7, 13, 19, 22, 26, 27, 40, 43, 49, 58, 61], "markov": [6, 9, 39, 51], "supposedli": 6, "closer": [6, 31], "typic": [6, 8, 15, 26, 33, 44], "set": [6, 8, 20, 22, 23, 26, 31, 50, 51, 52, 55], "start": [6, 7, 9, 19, 21, 26, 40, 43, 49, 52], "mcmc": [6, 7, 8, 9, 10, 11, 38, 39, 44, 51, 53, 55, 61], "so": [6, 7, 15, 18, 20, 27, 35, 36, 37, 53, 54, 55, 56, 61], "reach": [6, 7, 8, 21], "nut": [6, 8, 11, 16, 26, 54, 56, 57, 58, 59, 62], "extra_paramet": [6, 8], "fmaili": [6, 8], "whose": [6, 8, 22, 49], "being": [6, 8, 24, 26, 39, 55, 61], "dure": [6, 7, 8], "extra": [6, 8, 39, 60], "pass": [6, 8, 9, 15, 18, 23, 29, 33, 39, 49, 56], "g": [6, 8, 29, 31, 52, 55, 56], "log_step_s": [7, 27], "logarithm": [7, 19, 22, 23, 27, 40], "log_step_size_avg": 7, "time": [7, 9, 15, 17, 18, 21, 22, 26, 27, 35, 36, 37, 38, 53, 54, 55, 58, 60, 61], "taken": [7, 21, 23, 27, 48, 52], "far": [7, 27], "avg_err": [7, 27], "quantiti": [7, 9, 27], "h_t": [7, 27], "mu": [7, 27, 46, 48, 57, 59, 60], "arbitrari": [7, 14, 15, 21, 27, 53], "shrunk": [7, 27], "toward": [7, 27], "chose": [7, 27], "10": [7, 26, 27, 29, 31, 49, 54, 55, 56, 57, 59, 60, 61, 62], "epsilon_0": [7, 27], "context": [7, 27, 33, 58, 61], "find_reasonable_step_s": [7, 27], "avg_error": [7, 27], "dual_averaging_adapt": 7, "t0": [7, 27], "gamma": [7, 27, 29, 49], "05": [7, 27, 29, 49], "kappa": [7, 27], "75": [7, 8, 27], "order": [7, 15, 18, 19, 20, 24, 26, 48, 52, 54], "achiev": [7, 52, 53, 54], "desir": 7, "let": [7, 24, 52, 54, 55, 56, 57, 59, 60, 61], "u": [7, 21, 25, 26, 31, 51, 52, 56, 57, 59, 60], "note": [7, 10, 17, 24, 46, 49, 52, 57, 61], "epsilon": 7, "alpha_t": 7, "t": [7, 10, 20, 23, 24, 39, 53, 55, 56], "acept": 7, "defin": [7, 14, 20, 23, 29, 54, 56, 58, 61], "error": [7, 17, 52], "would": [7, 53, 56], "like": [7, 8, 13, 26, 35, 55, 56, 58], "find": [7, 8, 27, 38, 40, 60], "h": [7, 23, 52, 56], "x": [7, 19, 23, 29, 31, 44, 50, 52, 53, 54, 55, 56, 58, 61, 62], "mathbb": 7, "nes09": [7, 27, 51], "author": [7, 27], "hg": [7, 21, 26, 27, 51], "14": [7, 21, 26, 27, 32, 51, 53, 56], "guarante": [7, 17], "converg": [7, 8, 10, 27, 51, 55], "_t": [7, 10], "section": [7, 24, 51], "detail": [7, 8, 23, 24, 26, 27, 46, 55], "discuss": [7, 53, 55], "free": [7, 14, 27, 39, 51], "down": [7, 27], "introduc": [7, 27], "speed": [7, 8, 21, 27], "recommend": [7, 27, 54, 62], "past": [7, 27], "quickli": [7, 27], "forget": [7, 27], "kernel_gener": 7, "reference_st": 7, "hmcstate": [7, 15, 52, 53, 54, 61], "target_accept": 7, "65": 7, "reason": [7, 8, 50, 52, 54, 55, 61], "while": [7, 26, 52, 53, 54], "ani": [7, 31, 52, 53, 55, 56, 57, 59, 60], "choos": [7, 21, 50], "doubl": [7, 8, 21, 26], "halv": 7, "until": [7, 26], "kei": [7, 9, 15, 20, 39, 46, 48, 49, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62], "take": [7, 9, 13, 14, 15, 18, 19, 22, 23, 24, 26, 31, 33, 35, 36, 37, 38, 39, 40, 44, 46, 48, 49, 55, 58, 59, 60, 61], "correspond": [7, 9, 20, 23, 33, 39, 46, 53, 62], "reference_hmc_st": 7, "locat": [7, 49], "must": [7, 10, 24, 39, 43, 53], "found": [7, 14, 15, 21, 54, 59], "never": 7, "advanc": [7, 51, 61], "rel": [7, 40, 57], "search": [7, 29, 43, 49], "onc": 7, "windowadaptationst": 8, "imm_stat": 8, "is_mass_matrix_diagon": 8, "schedul": [8, 35, 36, 37, 38], "match": [8, 20, 31, 38, 46, 61], "close": [8, 52], "unlik": [8, 23, 46], "sever": [8, 26, 33, 52, 55], "phase": 8, "explicitli": [8, 61], "better": [8, 13, 14, 19, 22, 24, 31, 48, 52, 54, 55, 56], "modular": [8, 61], "doe": [8, 14, 24, 26, 52, 57, 61], "affect": [8, 46], "allow": [8, 9, 52, 55, 58], "user": [8, 13, 14, 15, 18, 21, 22, 24, 31, 35, 36, 37, 38, 46, 49, 50, 53, 54, 55, 58, 61, 62], "own": [8, 52], "thei": [8, 9, 22, 23, 24, 33, 52, 53, 62], "want": [8, 22, 48, 49, 52, 55, 62], "decoupl": 8, "consist": [8, 21, 53], "window": [8, 57, 58, 59, 60], "nesterov": [8, 51], "both": [8, 17, 23, 55], "recomput": 8, "end": [8, 13, 43, 52, 56], "re": [8, 14, 51, 52, 54, 58], "schemat": 8, "In": [8, 9, 15, 18, 20, 23, 24, 26, 39, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61], "still": [8, 10, 61], "creat": [8, 9, 22, 38, 44, 58], "progress_bar": 8, "least": 8, "link": [8, 15, 19, 24, 59], "metric": [8, 11, 16, 29, 49, 51, 61], "import": [8, 15, 20, 21, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62], "especi": [8, 52], "extrem": 8, "ineffici": 8, "wrong": 8, "provid": [8, 15, 18, 24, 35, 36, 37, 52, 55, 61, 62], "purpos": [8, 53], "origin": [8, 15, 18, 19, 21, 24, 55, 57, 60, 61], "evolv": 8, "improv": 8, "qualiti": 8, "whether": [8, 15, 18, 19, 20, 21, 24, 26, 29, 48, 54], "displai": 8, "progress": [8, 26, 30, 51], "bar": [8, 30, 51], "object": [8, 27, 29, 39, 53], "build_schedul": 8, "num_step": [8, 26, 61], "initial_buffer_s": 8, "final_buffer_s": 8, "50": [8, 19, 39, 56, 58], "first_window_s": 8, "25": [8, 56, 57], "list": [8, 22], "below": [8, 53, 56], "intend": 8, "period": [8, 22], "stage": [8, 17], "interv": [8, 17, 43, 56], "global": 8, "inform": [8, 9, 13, 14, 15, 18, 19, 21, 22, 24, 38, 39, 44, 49, 51, 55, 60], "seri": [8, 51], "expand": 8, "memori": [8, 26], "twice": [8, 9, 53], "preced": 8, "distinct": [8, 55], "come": 8, "common": 8, "accur": 8, "label": [8, 58], "initial_buff": 8, "width": 8, "window_label": 8, "is_middle_window_end": 8, "initfn": 9, "often": [9, 26], "need": [9, 13, 15, 18, 19, 21, 35, 36, 37, 49, 52, 54, 55, 56, 57, 58, 59, 60, 61, 62], "some": [9, 23, 24, 55], "same": [9, 14, 22, 26, 53, 55, 61], "therefor": 9, "do": [9, 15, 18, 21, 26, 35, 36, 37, 52, 53, 54, 56, 61], "oper": [9, 29, 55, 56], "themselv": 9, "function": [9, 53, 55, 56, 57, 58, 59, 60, 61, 62], "samplingalgorithm": [9, 13, 14, 15, 18, 19, 21, 22, 24, 31, 35, 36, 37, 38, 50], "updatefn": 9, "transit": [9, 13, 14, 15, 18, 19, 21, 22, 23, 24, 26, 38, 39, 44, 55, 57, 61], "pure": [9, 53, 54, 62], "idempot": 9, "necessarili": 9, "about": [9, 13, 14, 15, 18, 19, 22, 24, 38, 39, 44, 54, 60], "transtion": 9, "simplifi": 9, "yet": [9, 10, 14, 15, 21, 54], "univers": [9, 55], "interfac": [9, 13, 14, 15, 18, 21, 22, 24, 31, 35, 36, 37, 38, 49, 50, 55, 61], "essenc": [9, 21], "what": [9, 54, 55], "rng": 9, "possibli": 9, "data": [9, 31, 33, 35, 36, 37, 51, 55, 58], "pair": [9, 14], "As": [9, 54, 55, 57, 60], "intern": [9, 51, 52, 55, 57, 59], "save": [9, 53], "addit": [9, 13, 15, 18, 19, 21, 23, 24, 26, 39], "call": [9, 22, 39, 50, 53, 55], "vialgorithm": [9, 48], "variat": [9, 29, 33, 39, 48, 49, 51], "infer": [9, 29, 48, 49, 51, 56, 57, 58, 59, 60], "potenti": [9, 10, 22, 39, 51, 57, 61, 62], "guid": 9, "draw": [9, 10, 14, 15, 18, 21, 22, 24, 29, 38, 49, 51, 53], "build": [9, 13, 14, 15, 18, 19, 21, 22, 23, 24, 26, 33, 38, 40, 44, 46, 52, 54, 56, 58, 62], "runfn": 9, "adapt": [9, 11, 15, 18, 30, 38, 43, 51, 55, 56, 57, 58, 59, 60], "potential_scale_reduct": 10, "input_arrai": 10, "arraylik": 10, "chain_axi": 10, "sample_axi": 10, "gelman": [10, 51], "rubin": [10, 51], "1992": [10, 51], "reduct": [10, 51], "multipl": [10, 46, 51, 58], "axi": [10, 54, 58], "indic": [10, 13, 22, 26, 29, 46, 54], "ndarrai": [10, 40], "r": [10, 51, 55, 56], "hat": [10, 55], "squeez": 10, "frac": [10, 20, 52, 56], "v": [10, 44, 51, 53, 56, 57, 59, 60], "w": [10, 14, 56, 58], "within": [10, 54], "posterior": [10, 13, 22, 39, 49, 53, 54, 55, 56, 57, 58, 59, 60, 61], "pool": 10, "trace": 10, "factor": [10, 29, 49], "uniti": 10, "greater": 10, "stac": [10, 51], "gr92": [10, 51], "effective_sample_s": 10, "effect": [10, 38, 40, 51, 55, 57, 60], "ess": [10, 11, 38, 41], "basic": [10, 13, 14, 15, 18, 21, 22, 24, 31, 35, 36, 37, 38, 49, 50, 61], "n_": 10, "mathit": 10, "eff": 10, "mn": 10, "tau": [10, 57, 59], "k": [10, 20, 29, 49, 51, 53, 56, 57, 59, 60], "p": [10, 20, 24, 44, 52, 53, 56], "m": [10, 19, 20, 31, 39, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61], "rho": [10, 48], "_autocorrel": 10, "lag": 10, "integ": [10, 22], "2k": 10, "staa": [10, 51], "gcsr95": [10, 51], "similar": [10, 14, 21, 55], "geyer": [10, 51], "monoton": 10, "gey92": [10, 51], "gey11": [10, 51], "diffus": [11, 16, 34], "elliptical_slic": [11, 16], "ghmc": [11, 16], "mala": [11, 16], "marginal_latent_gaussian": [11, 16], "periodic_orbit": [11, 16], "random_walk": [11, 16, 53], "termin": [11, 16, 26, 29, 49], "sgmcmc": 11, "csgld": [11, 34], "sghmc": [11, 32, 34], "sgld": [11, 31, 34], "sgnht": [11, 32, 34], "smc": [11, 15, 18], "adaptive_temp": [11, 41], "resampl": [11, 38, 39, 41, 44], "solver": [11, 12, 32, 38, 40, 41], "temper": [11, 38, 39, 40, 41, 43], "vi": [11, 51], "meanfield_vi": [11, 47], "svgd": [11, 47], "langevin": [12, 18, 31, 32, 36, 51], "overdamped_langevin": [12, 32], "logdensity_grad_fn": [12, 54], "euler": [12, 32], "overdamp": [12, 32], "ellipt": [13, 51], "ellipslicest": 13, "logdens": [13, 14, 15, 18, 19, 22, 24, 31, 52, 53, 55, 57, 61], "evalu": [13, 22, 52, 54, 55, 61], "ellipsliceinfo": 13, "debug": [13, 15, 18, 19, 21, 24, 56], "latent": [13, 19, 22], "theta": [13, 56, 57, 59], "2pi": 13, "identifi": 13, "ellipsi": 13, "drawn": [13, 49, 53, 61], "positon": 13, "mommentum": 13, "subit": 13, "sub": [13, 56], "subiter": 13, "less": [13, 53, 54, 56], "build_kernel": [13, 14, 15, 18, 21, 22, 24, 31, 35, 36, 37, 38, 44, 50, 53], "cov_matrix": 13, "mam10": [13, 51], "gaussian": [13, 19, 20, 24, 46], "prior": [13, 19, 33, 38, 44, 54, 56, 58], "along": [13, 14, 15, 18, 19, 20, 22, 23, 24, 38, 44, 49, 62], "ellip_slic": 13, "loglikelihood_fn": [13, 33, 35, 36, 37, 38, 44], "compil": [13, 14, 15, 18, 19, 21, 22, 24, 35, 36, 37, 53, 55, 56], "likelihood": [13, 19, 33, 38, 39, 44], "distributon": 13, "ghmcstate": 14, "henc": [14, 61], "thu": [14, 20, 23, 26, 52, 55, 58], "after": [14, 23, 53], "logdensity_grad": [14, 15, 18, 19, 52, 54, 61], "noise_fn": 14, "lambda": [14, 15, 53, 54, 57, 58, 61], "divergence_threshold": [14, 15, 21, 26], "1000": [14, 15, 21, 29, 49, 56, 57, 59, 60, 61], "apart": 14, "equival": [14, 26, 46, 54], "correct": [14, 24, 29, 49, 55], "atom": 14, "energi": [14, 15, 19, 20, 21, 22, 23, 24, 26, 31, 37, 61], "abov": [14, 15, 21, 26, 52, 53], "consid": [14, 15, 21, 23, 52, 54], "diverg": [14, 15, 21, 23, 26, 48, 49, 57], "mechan": [14, 26, 52], "done": [14, 19, 54, 55, 61], "determinisitc": 14, "alwai": [14, 15, 18, 21, 48, 56], "candiat": 14, "ghmc_kernel": 14, "structur": [14, 23, 46, 52], "translat": [14, 57, 59, 60], "absolut": [14, 15, 21], "sai": [14, 15, 21, 26], "commonli": [14, 15, 21], "noise_gn": 14, "dynamichmcst": 15, "hmcinfo": 15, "wa": [15, 18, 19, 21, 24, 32, 33, 52, 55, 61], "acceptance_r": [15, 18, 19, 21, 24, 57, 60], "is_accept": [15, 18, 19, 24], "is_diverg": [15, 21, 57], "exceed": [15, 21], "threshold": [15, 21], "total": [15, 23, 26, 53, 54, 55, 56, 58, 61], "includ": [15, 29], "symplect": [15, 17, 18, 21, 22, 26], "integratorst": [15, 21, 26], "init_dynam": 15, "velocity_verlet": [15, 17, 21, 22], "build_dynamic_kernel": 15, "integration_steps_fn": 15, "randomli": 15, "builder": [15, 18, 35, 36], "alia": [15, 18, 35, 36, 37], "cumbersom": [15, 18, 35, 36, 37, 61], "manipul": [15, 18, 35, 36, 37], "sinc": [15, 18, 35, 36, 37, 57], "specifi": [15, 18, 35, 36, 37, 61], "helper": [15, 18, 35, 36, 37], "special": [15, 18, 35, 36, 37], "attribut": [15, 18, 19], "etc": [15, 18, 53, 61], "you": [15, 18, 21, 35, 36, 37, 52, 54, 55, 62], "mclachlan": [15, 17, 51], "kinet": [15, 20, 21, 22, 26, 37], "befor": [15, 21, 53, 55], "palindrom": 17, "deriv": 17, "bcss14": [17, 51], "form": [17, 26, 29, 46, 52], "b1": 17, "a1": 17, "b2": 17, "choic": 17, "bound": 17, "method": [17, 26, 27, 51, 52, 53], "respect": [17, 20, 26, 39, 52, 56], "here": [17, 24, 26, 35, 36, 37, 39, 46, 53, 55], "mcl95": [17, 51], "focus": 17, "known": [17, 61], "minim": [17, 27, 29, 40, 49, 52], "norm": [17, 52, 55, 61, 62], "a2": 17, "4": [17, 26, 29, 39, 51, 53, 54, 55, 56, 57, 58], "67": [17, 61], "yoshida": [17, 21], "noneuclidean_leapfrog": 17, "noneuclidean_mclachlan": 17, "noneuclidean_yoshida": 17, "adjust": 18, "malast": 18, "malainfo": 18, "margin": [19, 53, 54, 56], "marginalst": 19, "rmh": [19, 24, 53], "log_p_x": 19, "model": [19, 31, 38, 49, 55, 61, 62], "grad_x": [19, 54], "u_x": 19, "auxiliari": [19, 51], "u_grad_x": 19, "marginalinfo": 19, "init_and_kernel": 19, "sampler": [19, 24, 39, 51, 52, 54, 55, 56, 57, 59, 62], "mgrad_gaussian": 19, "tp18": [19, 51], "log_likelihood": 19, "interestingli": 19, "calibr": 19, "roughli": [19, 39, 55], "q": [19, 52, 56], "exp": [19, 44, 55, 60, 61], "f": [19, 29, 40, 43, 52, 56, 57, 58, 60], "c": [19, 23, 56, 59, 62], "use_invers": 19, "zero": [19, 52, 60], "embed": 20, "particular": [20, 23, 24], "practic": [20, 51, 61], "definit": [20, 50], "fix": [20, 26, 53, 56], "independ": [20, 24, 51, 55, 58], "bblg17": [20, 51], "For": [20, 24, 46, 53, 55, 57, 59], "newtonian": 20, "relativist": [20, 51], "lph": [20, 51], "17": [20, 51, 56], "gaussian_euclidean": 20, "euclideankineticenergi": 20, "normal": [20, 24, 49, 51, 53, 54, 55, 57, 58, 59, 60, 61, 62], "bet13": [20, 51], "further": 20, "character": 20, "condit": [20, 53, 60], "pi": [20, 31], "assum": [20, 24, 35, 36, 37, 43, 61], "One": 20, "multipli": [20, 46], "tree": [20, 21, 26], "specif": [20, 24, 52, 53], "ravel_pytre": 20, "sort": 20, "dictionari": [20, 53, 56, 58], "them": [20, 39, 57, 61], "appear": [20, 33], "momentum_gener": 20, "kinetic_energi": [20, 26], "is_turn": [20, 21, 25], "turn": [20, 21, 25, 26, 51], "back": [20, 21, 54, 57, 61], "nutsinfo": 21, "trajectory_leftmost_st": 21, "leftmost": 21, "full": [21, 33, 46], "trajectory_rightmost_st": 21, "rightmost": 21, "num_trajectory_expans": 21, "subtrajectori": 21, "were": [21, 39, 57], "probabilti": 21, "across": [21, 23, 55], "entir": [21, 23], "major": 21, "samplig": 21, "multinomi": [21, 26, 42], "bet17": [21, 23, 26, 51], "expans": [21, 26], "recurs": [21, 26], "ppj19": [21, 51], "lsl": [21, 51], "20": [21, 51, 52, 55, 56, 58, 61, 62], "unusu": 21, "those": [21, 55], "familiar": [21, 24], "inde": [21, 52, 54], "conceptu": [21, 51], "construct": 21, "feel": 21, "lingo": 21, "inherit": 21, "unnecessarili": 21, "complic": 21, "hide": 21, "concept": 21, "upon": 21, "built": [21, 51, 54], "direct": [21, 22, 26], "continu": [21, 24], "simplect": 21, "max_num_doubl": 21, "obserb": 21, "occur": 21, "orbit": [22, 51], "periodicorbitalst": 22, "accord": [22, 32], "reweight": 22, "vector": [22, 24, 31, 46], "neg": [22, 31, 56], "logdensities_grad": 22, "anyth": 22, "name": [22, 54, 58, 59, 60], "either": [22, 24, 39, 46, 55], "numpi": [22, 46, 52, 53, 54, 55, 57, 58, 59, 60, 61, 62], "repeat": [22, 26], "assign": [22, 39], "biject": 22, "nw22": [22, 51], "orbital_hmc": 22, "connect": 22, "precis": [22, 46], "per_orbit": 22, "buid": 22, "trajectoryst": 23, "canon": 23, "z": [23, 29, 53, 56, 57], "sum_log_p_accept": 23, "cumul": 23, "proposal_gener": [23, 24], "associ": [23, 24, 48], "proposal_from_energy_diff": 23, "initial_energi": 23, "new_energi": 23, "flag": 23, "asymmetric_proposal_gener": 23, "transition_energy_fn": 23, "proposal_factori": 23, "account": 23, "asymmetri": 23, "keyword": [23, 29, 49], "argument": [23, 29, 33, 49, 53, 56], "samplepropos": 23, "proto": 23, "def": [23, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62], "meth": 23, "self": [23, 54], "Such": 23, "primarili": 23, "static": [23, 26, 52], "checker": 23, "recogn": [23, 55], "subtyp": 23, "duck": 23, "func": 23, "check": [23, 26, 52, 56], "pep": 23, "544": 23, "decor": [23, 30], "runtime_check": 23, "act": 23, "simpl": [23, 33, 51, 54, 61], "mind": [23, 62], "runtim": [23, 61], "presenc": 23, "ignor": [23, 40], "signatur": [23, 38], "genproto": 23, "static_binomial_sampl": [23, 24], "new_propos": 23, "lower": [23, 48, 51], "progressive_uniform_sampl": 23, "progressive_biased_sampl": 23, "bais": 23, "bias": 23, "awai": 23, "nonreversible_slice_sampl": 23, "decis": [23, 51], "nea20": [23, 51], "walk": 24, "rosenbluth": 24, "expos": 24, "conveni": [24, 52], "entri": 24, "level": [24, 48, 53, 55], "who": [24, 54, 55, 62], "simpler": [24, 56], "instanti": 24, "x_": [24, 39], "x_t": [24, 39], "newli": 24, "variant": 24, "offer": 24, "noic": 24, "additive_step": 24, "doesn": [24, 55], "t_1": 24, "irmh": 24, "symmetr": [24, 51], "That": [24, 54, 55], "without": [24, 54, 58], "proposal_logdensity_fn": 24, "asymmetr": 24, "gcsr14": [24, 51], "11": [24, 56, 57, 59, 60], "simplest": 24, "additive_step_random_walk": 24, "sigma": [24, 46, 53, 57, 59], "distanc": [24, 50], "suitabl": 24, "center": [24, 33, 57, 59, 60], "rwstate": [24, 53], "rw": 24, "log_dens": [24, 53], "rwinfo": 24, "build_additive_step": 24, "random_step": 24, "alreadi": [24, 53, 54], "compon": [24, 53], "covariance_matrix": 24, "rw_gaussian": 24, "normal_random_walk": 24, "produc": [24, 49, 52], "ad": 24, "obtain": [24, 53, 55, 58, 61], "maintain": [24, 53], "balanc": [24, 26], "classmethod": 24, "build_irmh": 24, "impli": [24, 54], "particl": [24, 38, 39, 40, 44, 50], "mutat": 24, "wan22": [24, 51], "proposal_distribut": 24, "know": [24, 52, 54], "build_rmh": 24, "appli": [24, 38, 53, 54, 55, 56, 57, 58], "build_rmh_transition_energi": 24, "rmh_propos": 24, "transition_distribut": 24, "init_propos": 24, "generate_propos": 24, "sample_propos": 24, "iterativeuturnst": 25, "momentum_sum": [25, 26], "idx_min": 25, "idx_max": 25, "iterative_uturn_numpyro": 25, "numpyro": [25, 51, 54], "style": 25, "proce": 26, "process": [26, 49, 51, 55, 61], "tradit": 26, "flip": 26, "distinguish": 26, "fall": [26, 54], "former": [26, 53], "categori": [26, 62], "There": [26, 57, 62], "faster": [26, 52, 55, 56], "risk": 26, "satur": 26, "keep": [26, 48], "discard": 26, "leftmost_st": 26, "rightmost_st": 26, "num_stat": 26, "append_to_trajectori": 26, "append": [26, 54], "reorder_trajectori": 26, "new_trajectori": 26, "merge_trajectori": 26, "left_trajectori": 26, "right_trajectori": 26, "static_integr": 26, "dynamicintegrationst": 26, "termination_st": 26, "dynamic_progressive_integr": 26, "update_termination_st": 26, "is_criterion_met": 26, "sequenti": [26, 51], "met": 26, "consecut": 26, "dynamic_recursive_integr": 26, "uturn_check_fn": 26, "use_robust_uturn_check": 26, "python": [26, 54, 59, 62], "6": [26, 39, 54, 56, 57, 61], "implemen": 26, "mostli": 26, "valid": 26, "sure": [26, 54, 57], "actual": 26, "cannot": [26, 52, 54], "dynamicexpansionst": 26, "dynamic_multiplicative_expans": 26, "trajectory_integr": 26, "max_num_expans": 26, "pick": [26, 50], "stop": [26, 52], "prematur": 26, "els": 26, "geometr": [26, 31, 51], "why": [26, 52, 55, 61], "literatur": 26, "hmc_energi": 26, "log_x": 27, "log_x_avg": 27, "primal": [27, 51, 52], "subgradi": [27, 51], "mathemat": [27, 51], "lbfg": [28, 49], "lbfgshistori": 29, "path": [29, 49, 51], "l": [29, 49, 56], "bfg": [29, 40, 49, 52], "hessian": [29, 49], "update_mask": 29, "xi": [29, 37], "paper": 29, "minimize_lbfg": 29, "fun": [29, 38, 43, 52], "x0": 29, "maxit": [29, 49], "30": [29, 49, 51, 57], "maxcor": [29, 49], "gtol": [29, 49], "1e": [29, 35, 36, 37, 49, 52, 53, 54, 55, 61, 62], "08": [29, 49], "ftol": [29, 49], "maxl": [29, 49], "lbfgs_kwarg": [29, 49], "jaxopt": [29, 49], "optstep": 29, "real": [29, 61], "scalar": [29, 37, 46, 56], "compos": [29, 51], "vjp": [29, 54], "guess": 29, "f_k": [29, 49], "f_": [29, 49], "g_k": [29, 49], "_norm": [29, 49], "lbfgs_inverse_hessian_factor": 29, "represent": 29, "formula": 29, "ii": 29, "newton": [29, 49, 51], "lu": [29, 51], "zhang": [29, 51], "et": [29, 56], "al": 29, "arxiv": [29, 51], "2108": 29, "03782": 29, "lbfgs_inverse_hessian_formula_1": 29, "beta": [29, 32, 35, 37, 49, 56], "lbfgs_inverse_hessian_formula_2": 29, "bfgs_sampl": 29, "num_sampl": [29, 30, 42, 48, 49, 53, 55, 56, 57, 58, 59, 60, 61], "grad_posit": [29, 49], "jeremi": [30, 51], "coullon": [30, 51], "blog": 30, "post": [30, 61], "cou": [30, 51], "progress_bar_scan": 30, "print_rat": 30, "scan": [30, 51, 52, 53, 55, 56, 57, 58, 59, 60, 61], "contour": [31, 51], "stochast": [31, 35, 36, 37, 51], "dlh": [31, 51], "22": [31, 51, 56], "dll20": [31, 51], "contoursgldst": 31, "energy_pdf": 31, "th": 31, "int_": 31, "s_1": 31, "mathrm": [31, 44], "d": [31, 44, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61], "s_i": 31, "partit": 31, "energy_idx": 31, "index": [31, 39, 46, 56], "belong": 31, "num_partit": 31, "512": 31, "energy_gap": 31, "min_energi": 31, "divid": 31, "landscap": 31, "rang": [31, 54, 57, 62], "rough": 31, "minimum": 31, "dataset": [31, 33, 56, 58, 61], "strictli": 31, "exact": [31, 51], "3456": 31, "enough": [31, 52], "gap": 31, "logdensity_estim": [31, 33], "gradient_estim": 31, "zeta": 31, "hyperparamet": 31, "reduc": 31, "temperatur": [31, 38], "port": [32, 33], "cn22": [32, 33, 51], "01": [32, 35, 37, 51, 55, 61], "equat": [32, 51, 52], "cfg14": [32, 51], "mcf15": [32, 51], "dfb": [32, 51], "logprior_fn": [33, 35, 36, 37, 38, 44], "data_s": [33, 35, 36, 37], "rm51": [33, 51], "instanc": [33, 46, 55], "supervis": 33, "item": [33, 53, 54, 56, 57, 60], "grad_estim": [33, 35, 36, 37], "control_vari": 33, "logdensity_grad_estim": 33, "centering_posit": 33, "bffn19": [33, 51], "map": [33, 46], "constant": [35, 36, 37], "now": [35, 36, 37, 52, 53, 54, 55, 57, 58, 59, 60], "yield": [35, 36, 37, 49, 60], "minibatch": [35, 36, 37], "new_posit": [35, 36], "grad_fn": 36, "nos\u00e9": 37, "hoover": 37, "thermostat": [37, 51], "sgnhtstate": 37, "mcmc_step_fn": [38, 44], "mcmc_init_fn": [38, 44], "resampling_fn": [38, 44], "target_ess": [38, 40], "root_solv": [38, 40], "dichotomi": [38, 43], "mcmc_kernel_factori": 38, "make_mcmc_st": 38, "delta_0": 38, "min_delta": [38, 40, 43], "max_delta": [38, 40, 43], "use_log_ess": 38, "solv": [38, 40, 43], "usual": [38, 46, 61], "temperedsmcst": [38, 44], "adaptive_tempered_smc": 38, "mcmc_paramet": 38, "num_mcmc_step": 38, "smcstate": [39, 40], "leav": 39, "n_particl": 39, "univari": [39, 55, 61], "bivari": 39, "51": [39, 61], "55": 39, "second": [39, 55], "52": [39, 56], "54": [39, 61], "60": [39, 61], "70": 39, "smcinfo": 39, "ancestor": 39, "log_likelihood_incr": 39, "increment": [39, 40, 43], "due": 39, "update_info": 39, "update_fn": 39, "weigh_fn": 39, "resample_fn": 39, "num_resampl": 39, "g_t": 39, "weigh": 39, "pmap": 39, "feynman": 39, "kac": 39, "term": [39, 44, 48], "goe": [39, 57], "m_t": 39, "r_t": 39, "idx": 39, "x_tm1": 39, "wast": [39, 51], "dc20": [39, 51], "charg": 39, "new_particl": 39, "thing": [40, 42, 43, 55], "log_weight": 40, "log_ess": 40, "np": [40, 54, 55, 57, 59, 60, 61, 62], "ess_solv": 40, "smc_state": 40, "max": [40, 51, 53], "root": [40, 43], "delta0": 40, "min": 40, "systemat": 42, "stratifi": 42, "residu": 42, "_delta0": 43, "ep": [43, 56], "0001": 43, "max_it": 43, "100": [43, 53, 54, 57, 58], "decreas": 43, "toler": 43, "b": [43, 46, 51, 56, 58], "lmbda": 44, "propto": 44, "p_0": 44, "easi": [44, 61], "unnorm": [44, 62], "pointwis": 44, "num_mcmc_iter": 44, "linear_map": 46, "diag_or_dense_a": 46, "highest": [46, 49], "linear": [46, 55], "y": [46, 50, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61], "ax": [46, 61], "dispatch": [46, 52], "jnp": [46, 52, 53, 55, 58, 59, 60, 61, 62], "dot": [46, 58], "dtype": [46, 52, 53, 54, 56, 60, 61], "2nd": 46, "1d": 46, "2d": 46, "googl": 46, "7859": 46, "lax": [46, 52, 53, 55, 56, 57, 58, 59, 60, 61], "dot_gener": 46, "generate_gaussian_nois": 46, "generate_unit_vector": 46, "unit": 46, "pytree_s": 46, "index_pytre": 46, "input_pytre": 46, "flat": 46, "variou": 46, "somehow": 46, "simul": [46, 51], "hamilonian": 46, "unclear": [46, 55], "interact": [46, 51], "demonstr": [46, 52, 55], "wai": [46, 53], "tell": [46, 56], "column": 46, "individu": 46, "arang": [46, 55], "mfvistat": 48, "opt_stat": 48, "mfviinfo": 48, "elbo": [48, 49], "stl_estim": 48, "field": 48, "init_st": 48, "kullback": [48, 49], "leibler": [48, 49], "stick": [48, 51], "land": [48, 51], "stl": 48, "rwd17": [48, 51], "remov": 48, "score": 48, "suggest": 48, "asd20": [48, 51], "high": 48, "generate_meanfield_logdens": 48, "pathfinderst": 49, "local": 49, "inter": 49, "fo": 49, "wrt": 49, "rappresent": 49, "initial_posit": [49, 55, 57, 60, 61, 62], "200": [49, 55], "pathfinderinfo": 49, "zcgv22": [49, 51], "prpng": 49, "un": 49, "densifi": 49, "routin": 49, "lgbf": 49, "travers": 49, "union": 49, "prng": [49, 53], "lowest": 49, "kl": 49, "heavi": 49, "just": [49, 55, 62], "visamplingalgorithm": 49, "rbf_kernel": 50, "length_scal": 50, "update_median_heurist": 50, "svgdstate": 50, "median": 50, "bandwidth": 50, "rbf": 50, "middl": 50, "ground": 50, "empir": 50, "strategi": 50, "grad_logdensity_fn": 50, "semi": 50, "update_kernel_paramet": 50, "url": 51, "http": [51, 59], "mc": 51, "org": 51, "doc": 51, "2_27": 51, "html": [51, 59], "2_20": 51, "notat": 51, "abhinav": 51, "agraw": 51, "daniel": 51, "sheldon": 51, "justin": 51, "domk": 51, "black": 51, "box": [51, 53, 55], "flow": 51, "neural": [51, 58], "system": 51, "33": [51, 56], "17358": 51, "17369": 51, "2020": 51, "jack": 51, "baker": 51, "paul": 51, "fearnhead": 51, "emili": 51, "fox": 51, "christoph": 51, "nemeth": 51, "29": [51, 61], "599": 51, "615": 51, "2019": 51, "michael": 51, "betancourt": 51, "riemannian": 51, "scienc": 51, "confer": 51, "gsi": 51, "2013": 51, "pari": 51, "franc": 51, "august": 51, "28": [51, 57, 59, 60, 61], "proceed": 51, "327": 51, "334": 51, "springer": 51, "introduct": [51, 55], "preprint": 51, "1701": 51, "02434": 51, "2017": 51, "simon": 51, "byrn": 51, "sam": 51, "livingston": 51, "mark": 51, "girolami": 51, "foundat": 51, "1410": 51, "5110": 51, "sergio": 51, "blane": 51, "fernando": 51, "casa": 51, "je": 51, "\u00fa": 51, "mar": 51, "\u0131": 51, "sanz": 51, "serna": 51, "hybrid": 51, "siam": 51, "journal": 51, "scientif": 51, "36": 51, "a1556": 51, "a1580": 51, "2014": 51, "tianqi": 51, "chen": 51, "guestrin": 51, "machin": [51, 54, 55], "1683": 51, "1691": 51, "pmlr": 51, "loop": [51, 52, 54], "www": 51, "jeremiecoullon": 51, "com": 51, "2021": 51, "jax_progress_bar": 51, "sgmcmcjax": 51, "lightweight": 51, "open": 51, "softwar": 51, "7": [51, 56, 57, 59, 60, 61], "72": [51, 61], "4113": 51, "2022": 51, "hai": 51, "dang": 51, "dau": 51, "nicola": 51, "chopin": 51, "2011": 51, "02328": 51, "wei": 51, "deng": 51, "siqi": 51, "liang": 51, "botao": 51, "hao": 51, "guang": 51, "lin": 51, "fame": 51, "2202": 51, "09867": 51, "multi": [51, 58], "modal": 51, "15725": 51, "15736": 51, "nan": 51, "ding": 51, "youhan": 51, "fang": 51, "ryan": 51, "babbush": 51, "changyou": 51, "robert": 51, "skeel": 51, "hartmut": 51, "neven": 51, "bayesian": [51, 58, 61], "andrew": 51, "john": 51, "carlin": 51, "hal": 51, "stern": 51, "donald": 51, "analysi": 51, "chapman": 51, "hall": 51, "crc": 51, "1995": 51, "page": 51, "457": 51, "472": 51, "charl": 51, "j": [51, 56, 57, 59], "473": 51, "483": 51, "handbook": 51, "20116022": 51, "45": 51, "matthew": 51, "hoffman": 51, "alexei": 51, "radul": 51, "pavel": 51, "sountsov": 51, "artifici": 51, "intellig": 51, "3907": 51, "3915": 51, "mach": 51, "15": [51, 53, 56, 57, 59, 60, 61], "1593": 51, "1623": 51, "7799": 51, "7813": 51, "junpeng": 51, "lao": 51, "suter": 51, "ian": 51, "langmor": 51, "cyril": 51, "chimisov": 51, "ashish": 51, "saxena": 51, "dave": 51, "moor": 51, "rif": 51, "saurou": 51, "joshua": 51, "dillon": 51, "tfp": [51, 58, 59], "modern": [51, 55], "tool": 51, "hardwar": [51, 62], "2002": 51, "01184": 51, "xiaoyu": 51, "valerio": 51, "perron": 51, "leonard": 51, "hasenclev": 51, "yee": 51, "whye": 51, "teh": 51, "sebastian": 51, "vollmer": 51, "1236": 51, "1245": 51, "yi": 51, "ma": 51, "complet": [51, 55], "recip": 51, "2015": 51, "On": 51, "ordinari": 51, "differenti": [51, 52, 54, 56], "composit": 51, "16": [51, 53, 55, 56, 57, 59, 60], "151": 51, "168": 51, "iain": 51, "murrai": 51, "david": 51, "mackai": 51, "thirteenth": 51, "541": 51, "548": 51, "jmlr": 51, "workshop": 51, "2010": 51, "radford": 51, "neal": 51, "reversibli": 51, "2001": 51, "11950": 51, "kiril": 51, "neklyudov": 51, "5790": 51, "5814": 51, "yurii": 51, "convex": [51, 52], "problem": [51, 53, 55, 59, 60], "program": [51, 55, 58], "120": 51, "221": 51, "259": 51, "2009": 51, "du": [51, 56], "phan": 51, "neeraj": 51, "pradhan": 51, "martin": 51, "jankowiak": 51, "flexibl": 51, "acceler": [51, 62], "probabilist": [51, 55, 58], "1912": 51, "11554": 51, "herbert": 51, "robbin": 51, "sutton": 51, "monro": 51, "annal": 51, "400": 51, "407": 51, "1951": 51, "geoffrei": 51, "roeder": 51, "yuhuai": 51, "wu": 51, "duvenaud": 51, "michali": 51, "titsia": 51, "omiro": 51, "papaspiliopoulo": 51, "royal": 51, "societi": 51, "methodologi": 51, "80": 51, "749": 51, "767": 51, "2018": 51, "guanyang": 51, "wang": 51, "bernoulli": 51, "2012": 51, "2033": 51, "bob": 51, "carpent": 51, "aki": 51, "vehtari": 51, "research": [51, 55, 62], "23": [51, 55, 56, 57], "306": 51, "49": [51, 56, 61], "primit": [52, 53, 55, 57, 59, 60], "whenev": 52, "applic": 52, "sometim": 52, "your": [52, 54, 55, 62], "express": 52, "forward": [52, 55, 58], "mode": [52, 54, 56], "backward": 52, "despair": 52, "cover": [52, 55], "academ": 52, "understand": [52, 53, 55], "min_": 52, "tediou": 52, "minimis": 52, "happen": [52, 53, 55], "show": [52, 54, 56, 57, 58, 59, 60], "modifi": [52, 53], "overal": 52, "our": [52, 53, 56, 58], "theori": 52, "conjug": [52, 56], "transport": 52, "begin": [52, 53, 56, 58], "align": [52, 56], "langl": 52, "rangl": 52, "qquad": 52, "And": [52, 54, 55, 56, 59, 60], "min_i": 52, "datetim": [52, 53, 54, 55, 56, 57, 58, 59, 60, 61], "date": [52, 53, 54, 55, 56, 57, 58, 59, 60, 61], "todai": [52, 53, 54, 55, 56, 57, 58, 59, 60, 61], "strftime": [52, 53, 54, 55, 56, 57, 58, 59, 60, 61], "scipi": [52, 53, 55, 61, 62], "out": [52, 53, 54, 55], "ab": [52, 53], "later": 52, "undefin": 52, "rais": [52, 54], "has_aux": 52, "print": [52, 54, 55, 56, 57, 58, 60], "work": [52, 53, 55, 56, 62], "while_loop": 52, "fori_loop": 52, "leverag": 52, "asum": 52, "uniqu": 52, "df": [52, 56], "dx": [52, 56], "bigg": 52, "dh": [52, 56], "dy": [52, 56], "argmin": 52, "custom_vjp": [52, 54], "functool": [52, 58], "partial": [52, 58], "custom_jvp": 52, "nondiff_argnum": 52, "f_with_gradi": 52, "defjvp": 52, "f_jac_vec_prod": 52, "tangent": 52, "x_dot": 52, "fact": [52, 55], "f_out": 52, "31415": 52, "560483": 52, "float32": [52, 53, 54, 60, 61], "weak_typ": [52, 54, 61], "previous": 52, "obvious": [52, 54], "true_f": 52, "56049085": 52, "compar": [52, 54, 55], "5604908466339111": 52, "5604829788208008": 52, "word": 52, "suffic": 52, "toi": 52, "stat": [52, 53, 55, 61, 62], "logpdf": [52, 53, 55, 61, 62], "step_kei": 52, "5600991": 52, "6666666": 52, "0519627": 52, "6498852": 52, "7897134": 52, "techniqu": 53, "joint": [53, 56, 58], "newcommand": 53, "xx": 53, "boldsymbol": 53, "yy": 53, "altern": 53, "sim": [53, 56], "mid": 53, "ideal": 53, "analyt": 53, "appropri": 53, "hand": [53, 54], "idea": [53, 55], "block": [53, 62], "xx_1": 53, "ldot": 53, "xx_n": 53, "simplic": 53, "notebook": [53, 55, 57, 58, 59, 60, 61], "focu": 53, "jsp": 53, "suppos": 53, "multivari": 53, "operatornam": [53, 56], "mvnormal": 53, "pdf": 53, "concaten": 53, "multivariate_norm": 53, "zeros_lik": 53, "cov": 53, "illustr": [53, 55], "ll": [53, 55, 61], "write": [53, 54], "mwg_kernel": 53, "hold": 53, "correctli": [53, 54, 57], "issu": [53, 55], "algorithmst": 53, "class": [53, 58], "log_prob": 53, "xx_": 53, "yy_": 53, "via": [53, 54, 55, 62], "xx_t": 53, "yy_t": 53, "cdot": 53, "approach": [53, 58], "readi": [53, 54, 55, 61], "paramt": 53, "mwg_init_x": 53, "mwg_init_i": 53, "mwg_step_fn_x": 53, "mwg_step_fn_i": 53, "rmcstate": 53, "step_fn": 53, "rng_key_x": 53, "rng_key_i": 53, "num": 53, "copi": 53, "logdensity_x": 53, "logdensity_i": 53, "transition_gener": 53, "ey": 53, "initial_st": [53, 55, 56, 57, 58, 59, 60, 61], "sampling_loop": 53, "one_step": [53, 55, 56, 57, 58, 59, 60, 61], "sample_kei": [53, 55, 56, 57, 58, 59, 60, 61], "10_000": [53, 61], "cpu": [53, 54, 58, 61, 62], "sy": [53, 54, 55, 58, 61], "wall": [53, 54, 55, 58, 61], "matplotlib": [53, 57, 59, 60, 61], "pyplot": [53, 57, 59, 60, 61], "plt": [53, 57, 59, 60, 61], "arviz": [53, 56, 57, 59, 60], "az": [53, 56, 57, 59, 60], "idata": [53, 56, 57, 59, 60], "from_dict": [53, 56, 57, 59, 60], "plot_pair": 53, "kind": [53, 54, 56], "hexbin": 53, "tight_layout": [53, 57, 59, 60], "attempt": 53, "mwg_kernel_gener": 53, "elemtn": 53, "chi": 53, "len": [53, 55, 56, 58], "zip": [53, 58, 59], "logdensity_k": 53, "kwarg": 53, "_k": 53, "sampling_loop_gener": 53, "positions_gener": 53, "reus": 53, "805": 53, "819": 53, "810": 53, "tree_map": 53, "But": 53, "ve": [53, 55], "could": [53, 55], "ourselv": 53, "cost": 53, "expens": [53, 55], "deeper": 53, "low": 53, "generaliz": 53, "prototyp": 53, "adequ": 53, "get": [53, 54, 58, 59, 61], "unrol": 53, "caus": 53, "long": [53, 55, 56, 57, 59, 60, 62], "mitig": 53, "replac": 53, "sake": 53, "fulli": 53, "blackjax": [54, 55, 56, 57, 58, 59, 60], "oryx": 54, "pymc": 54, "tensorflow": [54, 60], "nevertheless": 54, "incompat": 54, "compatibl": 54, "judg": [54, 55], "hint": 54, "solut": [54, 55], "tensor": [54, 56, 59], "srng": [54, 56], "randomstream": [54, 56], "loc": [54, 55, 61, 62], "n_rv": 54, "i_rv": 54, "categor": [54, 58], "y_rv": [54, 56], "opt": [54, 59], "hostedtoolcach": [54, 59], "9": [54, 55, 56, 57, 58, 59, 60, 61], "18": [54, 55, 56, 57, 58, 59, 60], "x64": [54, 59], "lib": [54, 59], "python3": [54, 59], "site": [54, 59], "packag": [54, 59], "distutil": 54, "system_info": 54, "py": [54, 59], "2159": 54, "userwarn": 54, "vendor": 54, "bla": [54, 59], "netlib": 54, "wors": 54, "easili": [54, 55, 56, 57], "gain": 54, "switch": [54, 56], "_calc_info": 54, "predict": [54, 56, 58], "sampling_fn": 54, "516455713134264": 54, "1609480326942554": 54, "care": 54, "graph": [54, 56], "aeppl": [54, 56], "joint_logprob": [54, 56], "y_vv": [54, 56], "clone": 54, "i_vv": 54, "as_tensor": 54, "int64": [54, 56], "component_logdens": 54, "realiz": 54, "total_logdens": 54, "logsumexp": 54, "15039347": 54, "written": [54, 58, 62], "grad": [54, 56], "try": [54, 55], "luckili": 54, "total_logdensity_grad": 54, "44711512": 54, "abl": [54, 55, 56], "wrapper": 54, "hcb": 54, "numba_logpdf": 54, "arg": 54, "result_shap": 54, "call_grad": 54, "vjp_fwd": 54, "vjp_bwd": 54, "y_bar": 54, "defvjp": 54, "complain": [54, 56], "1503935": 54, "init_kei": [54, 56, 57, 58], "nuts_kei": [54, 62], "5826156": 54, "9966314": 54, "32264766": 54, "notic": 54, "quit": 54, "slowli": 54, "lot": 54, "forth": 54, "execut": [54, 55], "100_000": 54, "73": 54, "1_000": [54, 55, 61, 62], "104": 54, "overhead": 54, "workload": 54, "spent": 54, "becom": [54, 55], "irrelev": 54, "callback": 54, "bit": 54, "scope": [54, 58], "tutori": [54, 55, 62], "inspir": 54, "triton": 54, "regist": 54, "few": 55, "ubiquit": 55, "trend": 55, "emerg": 55, "thousand": 55, "whatev": 55, "thank": 55, "gpu": 55, "tpu": [55, 62], "best": [55, 56], "simd": 55, "rememb": 55, "instruct": [55, 56, 57, 59, 60], "intruct": 55, "unexpect": 55, "consequ": 55, "higher": 55, "abstract": 55, "core": [55, 58], "walkthrough": 55, "invit": 55, "read": 55, "regress": 55, "inher": 55, "limit": 55, "observ": [55, 56, 59, 60, 61, 62], "log_scal": [55, 61], "inference_loop": [55, 56, 57, 58, 59, 60, 61], "dramat": 55, "poorli": 55, "inv_mass_matrix": [55, 61], "put": 55, "foot": 55, "multiprocess": 55, "cpu_count": 55, "newcom": 55, "immedi": 55, "benefit": 55, "awesom": 55, "insid": 55, "inference_loop_multiple_chain": 55, "prepar": 55, "again": 55, "in_ax": 55, "2_000": 55, "block_until_readi": [55, 57, 61], "24": [55, 56], "obviou": 55, "said": 55, "1024": 55, "wait": 55, "slowest": 55, "think": 55, "unfortun": 55, "although": 55, "grant": 55, "led": 55, "synchron": 55, "truli": 55, "help": 55, "prove": 55, "treat": 55, "regardless": 55, "avail": 55, "topic": 55, "xla_flag": 55, "environment": 55, "o": [55, 56], "environ": 55, "xla_force_host_platform_device_count": 55, "format": 55, "advis": 55, "confirm": 55, "successfuli": 55, "command": 55, "ask": 55, "snippet": 55, "fn": 55, "parallel_fn": 55, "logic": 55, "xla": [55, 62], "num_replica": 55, "question": 55, "workaround": 55, "involv": 55, "nest": [55, 58], "against": 55, "count": 55, "side": 55, "static_broadcasted_argnum": 55, "wouldn": 55, "prefer": 55, "highlight": 55, "pmap_stat": 55, "96": 55, "wow": 55, "intuit": 55, "transpos": 55, "2000": [55, 57], "minut": 55, "saw": 55, "mention": 55, "someth": 55, "design": [55, 61, 62], "altogeth": 55, "exhibit": 55, "behind": 55, "activ": [55, 58], "area": 55, "languag": [56, 57, 58, 59, 60], "pleas": [56, 57, 59, 60], "instal": [56, 57, 59, 60], "repositori": [56, 57, 59, 60], "binomi": 56, "respons": 56, "rat": 56, "tumor": 56, "peopl": [56, 62], "test": [56, 61, 62], "group_siz": 56, "19": 56, "46": 56, "27": 56, "47": 56, "13": 56, "48": 56, "positv": 56, "n_of_posit": 56, "12": [56, 57, 59, 60], "n_rat_tumor": 56, "improp": 56, "a_vv": 56, "b_vv": 56, "logprior": 56, "theta_rv": 56, "prior_predictive_fn": 56, "35": 56, "41": 56, "44": 56, "a_rv": 56, "b_rv": 56, "loglikelihood": 56, "theta_vv": 56, "logprob": 56, "unbound": 56, "odd": 56, "transformvaluesrewrit": 56, "logoddstransform": 56, "transforms_op": 56, "extra_rewrit": 56, "aemcmc": 56, "dprint": 56, "elemwis": 56, "no_inplac": 56, "id": 56, "mul": 56, "tensorconst": 56, "acc_dtyp": 56, "float64": 56, "makevector": 56, "and_": 56, "le": 56, "inplacedimshuffl": 56, "tensortyp": 56, "71": 56, "gammaln": 56, "ba": 56, "bb": 56, "bc": 56, "bd": 56, "BE": 56, "bf": 56, "bg": 56, "bh": 56, "bi": 56, "bj": 56, "bk": 56, "eq": 56, "bl": 56, "sigmoid": 56, "bm": 56, "val_clon": 56, "tran": 56, "bn": 56, "bo": 56, "bp": 56, "bq": 56, "br": 56, "bt": 56, "bu": 56, "bv": 56, "bw": 56, "inf": 56, "bx": 56, "BY": 56, "bz": 56, "ca": 56, "cb": 56, "cc": 56, "cd": 56, "ce": 56, "cf": 56, "cg": 56, "ch": 56, "ci": 56, "cj": 56, "ck": 56, "cl": 56, "cm": 56, "cn": 56, "co": 56, "cp": 56, "cq": 56, "cr": 56, "ct": 56, "cu": 56, "cv": 56, "cw": 56, "cx": 56, "cy": 56, "cz": 56, "da": 56, "db": 56, "dc": 56, "dd": 56, "de": 56, "dg": 56, "ge": 56, "di": 56, "dj": 56, "dk": 56, "dl": 56, "dm": 56, "dn": 56, "dp": 56, "dq": 56, "dr": 56, "dt": 56, "dv": 56, "dw": 56, "dz": 56, "ea": 56, "eb": 56, "ec": 56, "ed": 56, "ee": 56, "ef": 56, "eg": 56, "eh": 56, "ei": 56, "ej": 56, "log1p": 56, "ek": 56, "el": 56, "em": 56, "en": [56, 59], "eo": 56, "er": 56, "eu": 56, "ev": 56, "gt": 56, "ew": 56, "ex": 56, "ez": 56, "fa": 56, "fb": 56, "fc": 56, "fd": 56, "fe": 56, "ff": 56, "ipykernel": 56, "iostream": 56, "outstream": 56, "0x7f83d5f592e0": 56, "backend": 56, "logprob_jax": 56, "vm": 56, "jit_fn": 56, "wrap": 56, "life": 56, "flat_posit": 56, "go": [56, 62], "init_param_fn": 56, "seed": 56, "key1": 56, "key2": 56, "key3": 56, "minval": 56, "maxval": 56, "init_posit": [56, 59], "n_adapt": 56, "3000": 56, "n_sampl": 56, "warmup_kei": [56, 57, 58, 59, 60, 61], "plot_trac": [56, 57, 59, 60], "togeth": [57, 58], "reproduc": [57, 58, 59, 60], "eight": [57, 59, 60], "school": [57, 59, 60], "credit": 57, "team": 57, "hierarch": [57, 59, 60], "dist": 57, "reparam": 57, "transformreparam": 57, "eight_schools_noncent": 57, "halfcauchi": [57, 59], "plate": 57, "handler": 57, "config": 57, "transformeddistribut": 57, "affinetransform": 57, "ob": [57, 59], "initialize_model": 57, "init_param": 57, "potential_fn_gen": 57, "model_arg": 57, "dynamic_arg": 57, "num_warmup": 57, "num_diverg": 57, "2f": [57, 60], "87": 57, "00": 57, "plot": [57, 60, 61], "treatment": [57, 60], "plot_posterior": 57, "var_nam": [57, 60], "theta_bas": 57, "compact": [57, 60], "06": 57, "26": 57, "nativ": 58, "network": 58, "bnn": 58, "iri": 58, "sklearn": 58, "load_iri": 58, "featur": 58, "num_featur": 58, "num_class": 58, "target_nam": 58, "150": 58, "aesara": 58, "layer": 58, "random_vari": 58, "ppl": [58, 62], "tensorflow_prob": [58, 60], "substrat": [58, 60], "tfd": [58, 60], "dim_out": 58, "nn": 58, "relu": 58, "dim_in": 58, "w_kei": 58, "b_kei": 58, "sample_shap": 58, "perceptron": 58, "tag": 58, "collis": 58, "mlp": 58, "hidden_s": 58, "num_hidden": 58, "subkei": 58, "enumer": 58, "layer_": 58, "logit": 58, "mlp_kei": 58, "label_kei": 58, "joint_sampl": 58, "initial_weight": 58, "dict_kei": 58, "layer_1": 58, "layer_2": 58, "layer_3": 58, "num_paramet": 58, "tree_util": 58, "tree_flatten": 58, "2953": 58, "joint_log_prob": 58, "198": 58, "76": 58, "accuraci": 58, "interven": 58, "inject": 58, "posterior_weight": 58, "pred_kei": 58, "output_logit": 58, "output_prob": 58, "softmax": 58, "argmax": 58, "bma": 58, "9794667": 58, "9866667": 58, "descript": [59, 60], "pm": 59, "theta_1": 59, "warn": 59, "pytensor": 59, "get_jaxified_logp": 59, "sampling_jax": 59, "rv": 59, "value_var": 59, "tqdm": 59, "auto": 59, "21": 59, "tqdmwarn": 59, "iprogress": 59, "jupyt": 59, "ipywidget": 59, "readthedoc": 59, "io": 59, "user_instal": 59, "autonotebook": 59, "notebook_tqdm": 59, "init_position_dict": 59, "initial_point": 59, "50_000": [59, 60], "num_school": 60, "treatment_effect": 60, "treatment_stddev": 60, "se": 60, "jdc": 60, "jointdistributioncoroutineautobatch": 60, "avg_effect": 60, "log_tau": 60, "avg_stddev": 60, "theta_prim": 60, "school_effects_standard": 60, "yhat": 60, "pinned_model": 60, "experimental_pin": 60, "unnormalized_log_prob": 60, "92": 60, "dictionnari": 60, "school_effects_sampl": 60, "newaxi": 60, "correpond": 60, "showcas": 61, "recov": 61, "unconstrain": 61, "decent": 61, "000": 61, "31933": 61, "953": 61, "1265": 61, "6089": 61, "59030": 61, "027": 61, "hmc_kernel": 61, "mcmc_sampl": 61, "449": 61, "\u00b5": 61, "53": 61, "fig": 61, "ax1": 61, "subplot": 61, "ncol": 61, "figsiz": 61, "set_xlabel": 61, "set_ylabel": 61, "4_000": 61, "43": 61, "main": 62, "branch": 62, "latest": 62, "releas": 62, "art": 62, "robust": 62, "realli": 62, "menu": 62, "pip": 62, "nightli": 62, "conda": 62, "forg": 62, "By": 62, "relev": 62, "support": 62}, "objects": {"": [[11, 0, 0, "-", "blackjax"]], "blackjax": [[0, 0, 0, "-", "_version"], [3, 0, 0, "-", "adaptation"], [9, 0, 0, "-", "base"], [10, 0, 0, "-", "diagnostics"], [16, 0, 0, "-", "mcmc"], [28, 0, 0, "-", "optimizers"], [30, 0, 0, "-", "progress_bar"], [34, 0, 0, "-", "sgmcmc"], [41, 0, 0, "-", "smc"], [45, 0, 0, "-", "types"], [46, 0, 0, "-", "util"], [47, 0, 0, "-", "vi"]], "blackjax._version": [[0, 1, 1, "", "TYPE_CHECKING"], [0, 1, 1, "", "VERSION_TUPLE"], [0, 1, 1, "", "version"], [0, 1, 1, "", "version_tuple"]], "blackjax.adaptation": [[1, 0, 0, "-", "base"], [2, 0, 0, "-", "chees_adaptation"], [4, 0, 0, "-", "mass_matrix"], [5, 0, 0, "-", "meads_adaptation"], [6, 0, 0, "-", "pathfinder_adaptation"], [7, 0, 0, "-", "step_size"], [8, 0, 0, "-", "window_adaptation"]], "blackjax.adaptation.base": [[1, 2, 1, "", "AdaptationInfo"], [1, 2, 1, "", "AdaptationResults"]], "blackjax.adaptation.base.AdaptationInfo": [[1, 3, 1, "", "adaptation_state"], [1, 3, 1, "", "info"], [1, 3, 1, "", "state"]], "blackjax.adaptation.base.AdaptationResults": [[1, 3, 1, "", "parameters"], [1, 3, 1, "", "state"]], "blackjax.adaptation.chees_adaptation": [[2, 2, 1, "", "ChEESAdaptationState"], [2, 1, 1, "", "OPTIMAL_TARGET_ACCEPTANCE_RATE"], [2, 4, 1, "", "base"], [2, 4, 1, "", "chees_adaptation"]], "blackjax.adaptation.chees_adaptation.ChEESAdaptationState": [[2, 3, 1, "", "da_state"], [2, 3, 1, "", "log_step_size_moving_average"], [2, 3, 1, "", "log_trajectory_length_moving_average"], [2, 3, 1, "", "optim_state"], [2, 3, 1, "", "random_generator_arg"], [2, 3, 1, "", "step"], [2, 3, 1, "", "step_size"], [2, 3, 1, "", "trajectory_length"]], "blackjax.adaptation.mass_matrix": [[4, 2, 1, "", "MassMatrixAdaptationState"], [4, 2, 1, "", "WelfordAlgorithmState"], [4, 4, 1, "", "mass_matrix_adaptation"], [4, 4, 1, "", "welford_algorithm"]], "blackjax.adaptation.mass_matrix.MassMatrixAdaptationState": [[4, 3, 1, "", "inverse_mass_matrix"], [4, 3, 1, "", "wc_state"]], "blackjax.adaptation.mass_matrix.WelfordAlgorithmState": [[4, 3, 1, "", "m2"], [4, 3, 1, "", "mean"], [4, 3, 1, "", "sample_size"]], "blackjax.adaptation.meads_adaptation": [[5, 2, 1, "", "MEADSAdaptationState"], [5, 4, 1, "", "base"], [5, 4, 1, "", "maximum_eigenvalue"], [5, 4, 1, "", "meads_adaptation"]], "blackjax.adaptation.meads_adaptation.MEADSAdaptationState": [[5, 3, 1, "", "alpha"], [5, 3, 1, "", "current_iteration"], [5, 3, 1, "", "delta"], [5, 3, 1, "", "position_sigma"], [5, 3, 1, "", "step_size"]], "blackjax.adaptation.pathfinder_adaptation": [[6, 2, 1, "", "PathfinderAdaptationState"], [6, 4, 1, "", "base"], [6, 4, 1, "", "pathfinder_adaptation"]], "blackjax.adaptation.pathfinder_adaptation.PathfinderAdaptationState": [[6, 3, 1, "", "inverse_mass_matrix"], [6, 3, 1, "", "ss_state"], [6, 3, 1, "", "step_size"]], "blackjax.adaptation.step_size": [[7, 2, 1, "", "DualAveragingAdaptationState"], [7, 4, 1, "", "dual_averaging_adaptation"], [7, 4, 1, "", "find_reasonable_step_size"]], "blackjax.adaptation.step_size.DualAveragingAdaptationState": [[7, 3, 1, "", "avg_error"], [7, 3, 1, "", "log_step_size"], [7, 3, 1, "", "log_step_size_avg"], [7, 3, 1, "", "mu"], [7, 3, 1, "", "step"]], "blackjax.adaptation.window_adaptation": [[8, 2, 1, "", "WindowAdaptationState"], [8, 4, 1, "", "base"], [8, 4, 1, "", "build_schedule"], [8, 4, 1, "", "window_adaptation"]], "blackjax.adaptation.window_adaptation.WindowAdaptationState": [[8, 3, 1, "", "imm_state"], [8, 3, 1, "", "inverse_mass_matrix"], [8, 3, 1, "", "ss_state"], [8, 3, 1, "", "step_size"]], "blackjax.base": [[9, 2, 1, "", "AdaptationAlgorithm"], [9, 1, 1, "", "Info"], [9, 2, 1, "", "InitFn"], [9, 1, 1, "", "Position"], [9, 2, 1, "", "RunFn"], [9, 2, 1, "", "SamplingAlgorithm"], [9, 1, 1, "", "State"], [9, 2, 1, "", "UpdateFn"], [9, 2, 1, "", "VIAlgorithm"]], "blackjax.base.AdaptationAlgorithm": [[9, 3, 1, "", "run"]], "blackjax.base.SamplingAlgorithm": [[9, 3, 1, "", "init"], [9, 3, 1, "", "step"]], "blackjax.base.VIAlgorithm": [[9, 3, 1, "", "init"], [9, 3, 1, "", "sample"], [9, 3, 1, "", "step"]], "blackjax.diagnostics": [[10, 4, 1, "", "effective_sample_size"], [10, 4, 1, "", "potential_scale_reduction"]], "blackjax.mcmc": [[12, 0, 0, "-", "diffusions"], [13, 0, 0, "-", "elliptical_slice"], [14, 0, 0, "-", "ghmc"], [15, 0, 0, "-", "hmc"], [17, 0, 0, "-", "integrators"], [18, 0, 0, "-", "mala"], [19, 0, 0, "-", "marginal_latent_gaussian"], [20, 0, 0, "-", "metrics"], [21, 0, 0, "-", "nuts"], [22, 0, 0, "-", "periodic_orbital"], [23, 0, 0, "-", "proposal"], [24, 0, 0, "-", "random_walk"], [25, 0, 0, "-", "termination"], [26, 0, 0, "-", "trajectory"]], "blackjax.mcmc.diffusions": [[12, 4, 1, "", "overdamped_langevin"]], "blackjax.mcmc.elliptical_slice": [[13, 2, 1, "", "EllipSliceInfo"], [13, 2, 1, "", "EllipSliceState"], [13, 4, 1, "", "build_kernel"], [13, 2, 1, "", "elliptical_slice"], [13, 4, 1, "", "init"]], "blackjax.mcmc.elliptical_slice.EllipSliceInfo": [[13, 3, 1, "", "momentum"], [13, 3, 1, "", "subiter"], [13, 3, 1, "", "theta"]], "blackjax.mcmc.elliptical_slice.EllipSliceState": [[13, 3, 1, "", "logdensity"], [13, 3, 1, "", "position"]], "blackjax.mcmc.elliptical_slice.elliptical_slice": [[13, 3, 1, "", "build_kernel"], [13, 3, 1, "", "init"]], "blackjax.mcmc.ghmc": [[14, 2, 1, "", "GHMCState"], [14, 4, 1, "", "build_kernel"], [14, 2, 1, "", "ghmc"], [14, 4, 1, "", "init"]], "blackjax.mcmc.ghmc.GHMCState": [[14, 3, 1, "", "logdensity"], [14, 3, 1, "", "logdensity_grad"], [14, 3, 1, "", "momentum"], [14, 3, 1, "", "position"], [14, 3, 1, "", "slice"]], "blackjax.mcmc.ghmc.ghmc": [[14, 3, 1, "", "build_kernel"], [14, 3, 1, "", "init"]], "blackjax.mcmc.hmc": [[15, 2, 1, "", "DynamicHMCState"], [15, 2, 1, "", "HMCInfo"], [15, 2, 1, "", "HMCState"], [15, 4, 1, "", "build_dynamic_kernel"], [15, 4, 1, "", "build_kernel"], [15, 2, 1, "", "dynamic_hmc"], [15, 2, 1, "", "hmc"], [15, 4, 1, "", "init"], [15, 4, 1, "", "init_dynamic"]], "blackjax.mcmc.hmc.DynamicHMCState": [[15, 3, 1, "", "logdensity"], [15, 3, 1, "", "logdensity_grad"], [15, 3, 1, "", "position"], [15, 3, 1, "", "random_generator_arg"]], "blackjax.mcmc.hmc.HMCInfo": [[15, 3, 1, "", "acceptance_rate"], [15, 3, 1, "", "energy"], [15, 3, 1, "", "is_accepted"], [15, 3, 1, "", "is_divergent"], [15, 3, 1, "", "momentum"], [15, 3, 1, "", "num_integration_steps"], [15, 3, 1, "", "proposal"]], "blackjax.mcmc.hmc.HMCState": [[15, 3, 1, "", "logdensity"], [15, 3, 1, "", "logdensity_grad"], [15, 3, 1, "", "position"]], "blackjax.mcmc.hmc.dynamic_hmc": [[15, 3, 1, "", "build_kernel"], [15, 3, 1, "", "init"]], "blackjax.mcmc.hmc.hmc": [[15, 3, 1, "", "build_kernel"], [15, 3, 1, "", "init"]], "blackjax.mcmc.integrators": [[17, 1, 1, "", "mclachlan"], [17, 1, 1, "", "noneuclidean_leapfrog"], [17, 1, 1, "", "noneuclidean_mclachlan"], [17, 1, 1, "", "noneuclidean_yoshida"], [17, 1, 1, "", "velocity_verlet"], [17, 1, 1, "", "yoshida"]], "blackjax.mcmc.mala": [[18, 2, 1, "", "MALAInfo"], [18, 2, 1, "", "MALAState"], [18, 4, 1, "", "build_kernel"], [18, 4, 1, "", "init"], [18, 2, 1, "", "mala"]], "blackjax.mcmc.mala.MALAInfo": [[18, 3, 1, "", "acceptance_rate"], [18, 3, 1, "", "is_accepted"]], "blackjax.mcmc.mala.MALAState": [[18, 3, 1, "", "logdensity"], [18, 3, 1, "", "logdensity_grad"], [18, 3, 1, "", "position"]], "blackjax.mcmc.mala.mala": [[18, 3, 1, "", "build_kernel"], [18, 3, 1, "", "init"]], "blackjax.mcmc.marginal_latent_gaussian": [[19, 2, 1, "", "MarginalInfo"], [19, 2, 1, "", "MarginalState"], [19, 4, 1, "", "init_and_kernel"], [19, 2, 1, "", "mgrad_gaussian"]], "blackjax.mcmc.marginal_latent_gaussian.MarginalInfo": [[19, 3, 1, "", "acceptance_rate"], [19, 3, 1, "", "is_accepted"], [19, 3, 1, "", "proposal"]], "blackjax.mcmc.marginal_latent_gaussian.MarginalState": [[19, 3, 1, "", "U_grad_x"], [19, 3, 1, "", "U_x"], [19, 3, 1, "", "logdensity"], [19, 3, 1, "", "logdensity_grad"], [19, 3, 1, "", "position"]], "blackjax.mcmc.metrics": [[20, 4, 1, "", "gaussian_euclidean"]], "blackjax.mcmc.nuts": [[21, 2, 1, "", "NUTSInfo"], [21, 4, 1, "", "build_kernel"], [21, 1, 1, "", "init"], [21, 2, 1, "", "nuts"]], "blackjax.mcmc.nuts.NUTSInfo": [[21, 3, 1, "", "acceptance_rate"], [21, 3, 1, "", "energy"], [21, 3, 1, "", "is_divergent"], [21, 3, 1, "", "is_turning"], [21, 3, 1, "", "momentum"], [21, 3, 1, "", "num_integration_steps"], [21, 3, 1, "", "num_trajectory_expansions"], [21, 3, 1, "", "trajectory_leftmost_state"], [21, 3, 1, "", "trajectory_rightmost_state"]], "blackjax.mcmc.nuts.nuts": [[21, 3, 1, "", "build_kernel"], [21, 3, 1, "", "init"]], "blackjax.mcmc.periodic_orbital": [[22, 2, 1, "", "PeriodicOrbitalState"], [22, 4, 1, "", "build_kernel"], [22, 4, 1, "", "init"], [22, 2, 1, "", "orbital_hmc"]], "blackjax.mcmc.periodic_orbital.PeriodicOrbitalState": [[22, 3, 1, "", "directions"], [22, 3, 1, "", "logdensities"], [22, 3, 1, "", "logdensities_grad"], [22, 3, 1, "", "positions"], [22, 3, 1, "", "weights"]], "blackjax.mcmc.periodic_orbital.orbital_hmc": [[22, 3, 1, "", "build_kernel"], [22, 3, 1, "", "init"]], "blackjax.mcmc.proposal": [[23, 2, 1, "", "Proposal"], [23, 2, 1, "", "SampleProposal"], [23, 1, 1, "", "TrajectoryState"], [23, 4, 1, "", "asymmetric_proposal_generator"], [23, 4, 1, "", "nonreversible_slice_sampling"], [23, 4, 1, "", "progressive_biased_sampling"], [23, 4, 1, "", "progressive_uniform_sampling"], [23, 4, 1, "", "proposal_from_energy_diff"], [23, 4, 1, "", "proposal_generator"], [23, 4, 1, "", "static_binomial_sampling"]], "blackjax.mcmc.proposal.Proposal": [[23, 3, 1, "", "energy"], [23, 3, 1, "", "state"], [23, 3, 1, "", "sum_log_p_accept"], [23, 3, 1, "", "weight"]], "blackjax.mcmc.random_walk": [[24, 2, 1, "", "RWInfo"], [24, 2, 1, "", "RWState"], [24, 2, 1, "", "additive_step_random_walk"], [24, 4, 1, "", "build_additive_step"], [24, 4, 1, "", "build_irmh"], [24, 4, 1, "", "build_rmh"], [24, 4, 1, "", "build_rmh_transition_energy"], [24, 2, 1, "", "irmh"], [24, 4, 1, "", "normal"], [24, 2, 1, "", "rmh"], [24, 4, 1, "", "rmh_proposal"]], "blackjax.mcmc.random_walk.RWInfo": [[24, 3, 1, "", "acceptance_rate"], [24, 3, 1, "", "is_accepted"], [24, 3, 1, "", "proposal"]], "blackjax.mcmc.random_walk.RWState": [[24, 3, 1, "", "logdensity"], [24, 3, 1, "", "position"]], "blackjax.mcmc.random_walk.additive_step_random_walk": [[24, 3, 1, "", "build_kernel"], [24, 3, 1, "", "init"], [24, 5, 1, "", "normal_random_walk"]], "blackjax.mcmc.random_walk.irmh": [[24, 3, 1, "", "build_kernel"], [24, 3, 1, "", "init"]], "blackjax.mcmc.random_walk.rmh": [[24, 3, 1, "", "build_kernel"], [24, 3, 1, "", "init"]], "blackjax.mcmc.termination": [[25, 2, 1, "", "IterativeUTurnState"], [25, 4, 1, "", "iterative_uturn_numpyro"]], "blackjax.mcmc.termination.IterativeUTurnState": [[25, 3, 1, "", "idx_max"], [25, 3, 1, "", "idx_min"], [25, 3, 1, "", "momentum"], [25, 3, 1, "", "momentum_sum"]], "blackjax.mcmc.trajectory": [[26, 2, 1, "", "DynamicExpansionState"], [26, 2, 1, "", "DynamicIntegrationState"], [26, 2, 1, "", "Trajectory"], [26, 4, 1, "", "append_to_trajectory"], [26, 4, 1, "", "dynamic_multiplicative_expansion"], [26, 4, 1, "", "dynamic_progressive_integration"], [26, 4, 1, "", "dynamic_recursive_integration"], [26, 4, 1, "", "hmc_energy"], [26, 4, 1, "", "merge_trajectories"], [26, 4, 1, "", "reorder_trajectories"], [26, 4, 1, "", "static_integration"]], "blackjax.mcmc.trajectory.DynamicExpansionState": [[26, 3, 1, "", "proposal"], [26, 3, 1, "", "step"], [26, 3, 1, "", "termination_state"], [26, 3, 1, "", "trajectory"]], "blackjax.mcmc.trajectory.DynamicIntegrationState": [[26, 3, 1, "", "proposal"], [26, 3, 1, "", "step"], [26, 3, 1, "", "termination_state"], [26, 3, 1, "", "trajectory"]], "blackjax.mcmc.trajectory.Trajectory": [[26, 3, 1, "", "leftmost_state"], [26, 3, 1, "", "momentum_sum"], [26, 3, 1, "", "num_states"], [26, 3, 1, "", "rightmost_state"]], "blackjax.optimizers": [[27, 0, 0, "-", "dual_averaging"], [29, 0, 0, "-", "lbfgs"]], "blackjax.optimizers.dual_averaging": [[27, 2, 1, "", "DualAveragingState"], [27, 4, 1, "", "dual_averaging"]], "blackjax.optimizers.dual_averaging.DualAveragingState": [[27, 3, 1, "", "avg_error"], [27, 3, 1, "", "log_x"], [27, 3, 1, "", "log_x_avg"], [27, 3, 1, "", "mu"], [27, 3, 1, "", "step"]], "blackjax.optimizers.lbfgs": [[29, 2, 1, "", "LBFGSHistory"], [29, 4, 1, "", "bfgs_sample"], [29, 4, 1, "", "lbfgs_inverse_hessian_factors"], [29, 4, 1, "", "lbfgs_inverse_hessian_formula_1"], [29, 4, 1, "", "lbfgs_inverse_hessian_formula_2"], [29, 4, 1, "", "minimize_lbfgs"]], "blackjax.optimizers.lbfgs.LBFGSHistory": [[29, 3, 1, "", "alpha"], [29, 3, 1, "", "f"], [29, 3, 1, "", "g"], [29, 3, 1, "", "update_mask"], [29, 3, 1, "", "x"]], "blackjax.progress_bar": [[30, 4, 1, "", "progress_bar_scan"]], "blackjax.sgmcmc": [[31, 0, 0, "-", "csgld"], [32, 0, 0, "-", "diffusions"], [33, 0, 0, "-", "gradients"], [35, 0, 0, "-", "sghmc"], [36, 0, 0, "-", "sgld"], [37, 0, 0, "-", "sgnht"]], "blackjax.sgmcmc.csgld": [[31, 2, 1, "", "ContourSGLDState"], [31, 4, 1, "", "build_kernel"], [31, 2, 1, "", "csgld"], [31, 4, 1, "", "init"]], "blackjax.sgmcmc.csgld.ContourSGLDState": [[31, 3, 1, "", "energy_idx"], [31, 3, 1, "", "energy_pdf"], [31, 3, 1, "", "position"]], "blackjax.sgmcmc.csgld.csgld": [[31, 3, 1, "", "build_kernel"], [31, 3, 1, "", "init"]], "blackjax.sgmcmc.diffusions": [[32, 4, 1, "", "overdamped_langevin"], [32, 4, 1, "", "sghmc"], [32, 4, 1, "", "sgnht"]], "blackjax.sgmcmc.gradients": [[33, 4, 1, "", "control_variates"], [33, 4, 1, "", "grad_estimator"], [33, 4, 1, "", "logdensity_estimator"]], "blackjax.sgmcmc.sghmc": [[35, 4, 1, "", "build_kernel"], [35, 4, 1, "", "init"], [35, 2, 1, "", "sghmc"]], "blackjax.sgmcmc.sghmc.sghmc": [[35, 3, 1, "", "build_kernel"], [35, 3, 1, "", "init"]], "blackjax.sgmcmc.sgld": [[36, 4, 1, "", "build_kernel"], [36, 4, 1, "", "init"], [36, 2, 1, "", "sgld"]], "blackjax.sgmcmc.sgld.sgld": [[36, 3, 1, "", "build_kernel"], [36, 3, 1, "", "init"]], "blackjax.sgmcmc.sgnht": [[37, 2, 1, "", "SGNHTState"], [37, 4, 1, "", "build_kernel"], [37, 4, 1, "", "init"], [37, 2, 1, "", "sgnht"]], "blackjax.sgmcmc.sgnht.SGNHTState": [[37, 3, 1, "", "momentum"], [37, 3, 1, "", "position"], [37, 3, 1, "", "xi"]], "blackjax.sgmcmc.sgnht.sgnht": [[37, 3, 1, "", "build_kernel"], [37, 3, 1, "", "init"]], "blackjax.smc": [[38, 0, 0, "-", "adaptive_tempered"], [39, 0, 0, "-", "base"], [40, 0, 0, "-", "ess"], [42, 0, 0, "-", "resampling"], [43, 0, 0, "-", "solver"], [44, 0, 0, "-", "tempered"]], "blackjax.smc.adaptive_tempered": [[38, 2, 1, "", "adaptive_tempered_smc"], [38, 4, 1, "", "build_kernel"]], "blackjax.smc.adaptive_tempered.adaptive_tempered_smc": [[38, 3, 1, "", "build_kernel"], [38, 3, 1, "", "init"]], "blackjax.smc.base": [[39, 2, 1, "", "SMCInfo"], [39, 2, 1, "", "SMCState"], [39, 4, 1, "", "init"], [39, 4, 1, "", "step"]], "blackjax.smc.base.SMCInfo": [[39, 3, 1, "", "ancestors"], [39, 3, 1, "", "log_likelihood_increment"], [39, 3, 1, "", "update_info"]], "blackjax.smc.base.SMCState": [[39, 3, 1, "", "particles"], [39, 3, 1, "", "weights"]], "blackjax.smc.ess": [[40, 4, 1, "", "ess"], [40, 4, 1, "", "ess_solver"], [40, 4, 1, "", "log_ess"]], "blackjax.smc.resampling": [[42, 4, 1, "", "multinomial"], [42, 4, 1, "", "residual"], [42, 4, 1, "", "stratified"], [42, 4, 1, "", "systematic"]], "blackjax.smc.solver": [[43, 4, 1, "", "dichotomy"]], "blackjax.smc.tempered": [[44, 2, 1, "", "TemperedSMCState"], [44, 4, 1, "", "build_kernel"], [44, 4, 1, "", "init"]], "blackjax.smc.tempered.TemperedSMCState": [[44, 3, 1, "", "lmbda"], [44, 3, 1, "", "particles"], [44, 3, 1, "", "weights"]], "blackjax.types": [[45, 1, 1, "", "Array"], [45, 1, 1, "", "ArrayLikeTree"], [45, 1, 1, "", "ArrayTree"], [45, 1, 1, "", "PRNGKey"]], "blackjax.util": [[46, 4, 1, "", "generate_gaussian_noise"], [46, 4, 1, "", "generate_unit_vector"], [46, 4, 1, "", "index_pytree"], [46, 4, 1, "", "linear_map"], [46, 4, 1, "", "pytree_size"]], "blackjax.vi": [[48, 0, 0, "-", "meanfield_vi"], [49, 0, 0, "-", "pathfinder"], [50, 0, 0, "-", "svgd"]], "blackjax.vi.meanfield_vi": [[48, 2, 1, "", "MFVIInfo"], [48, 2, 1, "", "MFVIState"], [48, 4, 1, "", "generate_meanfield_logdensity"], [48, 2, 1, "", "meanfield_vi"], [48, 4, 1, "", "sample"], [48, 4, 1, "", "step"]], "blackjax.vi.meanfield_vi.MFVIInfo": [[48, 3, 1, "", "elbo"]], "blackjax.vi.meanfield_vi.MFVIState": [[48, 3, 1, "", "mu"], [48, 3, 1, "", "opt_state"], [48, 3, 1, "", "rho"]], "blackjax.vi.meanfield_vi.meanfield_vi": [[48, 3, 1, "", "init"], [48, 3, 1, "", "sample"], [48, 3, 1, "", "step"]], "blackjax.vi.pathfinder": [[49, 2, 1, "", "PathfinderState"], [49, 4, 1, "", "approximate"], [49, 2, 1, "", "pathfinder"], [49, 4, 1, "", "sample"]], "blackjax.vi.pathfinder.PathfinderState": [[49, 3, 1, "", "alpha"], [49, 3, 1, "", "beta"], [49, 3, 1, "", "elbo"], [49, 3, 1, "", "gamma"], [49, 3, 1, "", "grad_position"], [49, 3, 1, "", "position"]], "blackjax.vi.pathfinder.pathfinder": [[49, 3, 1, "", "approximate"], [49, 3, 1, "", "sample"]], "blackjax.vi.svgd": [[50, 4, 1, "", "rbf_kernel"], [50, 2, 1, "", "svgd"], [50, 4, 1, "", "update_median_heuristic"]], "blackjax.vi.svgd.svgd": [[50, 3, 1, "", "build_kernel"], [50, 3, 1, "", "init"]]}, "objtypes": {"0": "py:module", "1": "py:data", "2": "py:class", "3": "py:attribute", "4": "py:function", "5": "py:method"}, "objnames": {"0": ["py", "module", "Python module"], "1": ["py", "data", "Python data"], "2": ["py", "class", "Python class"], "3": ["py", "attribute", "Python attribute"], "4": ["py", "function", "Python function"], "5": ["py", "method", "Python method"]}, "titleterms": {"blackjax": [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 52, 53, 61, 62], "_version": 0, "modul": [0, 1, 2, 4, 5, 6, 7, 8, 9, 10, 12, 13, 14, 15, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 29, 30, 31, 32, 33, 35, 36, 37, 38, 39, 40, 42, 43, 44, 45, 46, 48, 49, 50], "content": [0, 1, 2, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 42, 43, 44, 45, 46, 48, 49, 50], "adapt": [1, 2, 3, 4, 5, 6, 7, 8, 61], "base": [1, 9, 39], "class": [1, 2, 4, 5, 6, 7, 8, 9, 11, 13, 14, 15, 18, 19, 21, 22, 23, 24, 25, 26, 27, 29, 31, 35, 36, 37, 38, 39, 44, 48, 49, 50], "chees_adapt": 2, "function": [2, 4, 5, 6, 7, 8, 10, 11, 12, 13, 14, 15, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 42, 43, 44, 46, 48, 49, 50, 52, 54], "attribut": [2, 9, 11, 21, 23], "submodul": [3, 11, 16, 28, 34, 41, 47], "mass_matrix": 4, "meads_adapt": 5, "pathfinder_adapt": 6, "step_siz": 7, "window_adapt": 8, "diagnost": 10, "subpackag": 11, "packag": [11, 34], "mcmc": [12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26], "diffus": [12, 32], "elliptical_slic": 13, "ghmc": 14, "hmc": [15, 61], "integr": 17, "mala": 18, "marginal_latent_gaussian": 19, "metric": 20, "nut": [21, 55, 61], "periodic_orbit": 22, "propos": 23, "random_walk": 24, "paramet": [24, 53, 61], "termin": 25, "trajectori": 26, "optim": [27, 28, 29], "dual_averag": 27, "lbfg": 29, "progress_bar": 30, "sgmcmc": [31, 32, 33, 34, 35, 36, 37], "csgld": 31, "gradient": [33, 52], "sghmc": 35, "sgld": 36, "sgnht": 37, "smc": [38, 39, 40, 41, 42, 43, 44], "adaptive_temp": 38, "ess": 40, "resampl": 42, "solver": 43, "temper": 44, "type": 45, "util": 46, "vi": [47, 48, 49, 50], "meanfield_vi": 48, "pathfind": 49, "svgd": 50, "bibliographi": 51, "us": [52, 54, 55, 56, 57, 58, 59, 60, 61], "custom": [52, 54], "defin": 52, "minimum": 52, "anoth": 52, "try": 52, "different": 52, "jax": [52, 54, 55], "grad": 52, "deriv": 52, "mathemat": 52, "tell": 52, "make": 52, "sure": 52, "result": [52, 53], "i": [52, 54], "correct": 52, "how": 53, "build": [53, 61], "metropoli": 53, "within": 53, "gibb": 53, "sampler": [53, 61], "The": [53, 61], "model": [53, 54, 56, 57, 58, 59, 60], "mwg": 53, "sampl": [53, 55], "construct": 53, "kernel": [53, 61], "set": [53, 61], "initi": [53, 61], "state": [53, 61], "each": 53, "algorithm": 53, "loop": [53, 61], "gener": 53, "check": 53, "develop": 53, "note": [53, 55], "logdens": 54, "compat": 54, "": [54, 61], "primit": 54, "aesara": [54, 56], "compil": 54, "numba": 54, "experiment": 54, "host_callback": 54, "call": 54, "xla": 54, "faster": 54, "multipl": 55, "chain": 55, "parallel": 55, "vector": 55, "v": 55, "vmap": 55, "pmap": 55, "A": [55, 61], "cpu": 55, "choos": 55, "number": 55, "devic": 55, "back": 55, "our": 55, "exampl": 55, "conclus": 55, "befor": [56, 57, 59, 60], "you": [56, 57, 59, 60], "start": [56, 57, 59, 60], "numpyro": 57, "oryx": 58, "pymc": 59, "tfp": 60, "quick": 61, "introduct": 61, "problem": 61, "infer": 61, "stan": 61, "window": 61, "welcom": 62, "hello": 62, "world": 62, "instal": 62, "gpu": 62, "instruct": 62}, "envversion": {"sphinx.domains.c": 2, "sphinx.domains.changeset": 1, "sphinx.domains.citation": 1, "sphinx.domains.cpp": 8, "sphinx.domains.index": 1, "sphinx.domains.javascript": 2, "sphinx.domains.math": 2, "sphinx.domains.python": 3, "sphinx.domains.rst": 2, "sphinx.domains.std": 2, "sphinx.ext.intersphinx": 1, "sphinx.ext.viewcode": 1, "sphinxcontrib.bibtex": 9, "sphinx": 57}, "alltitles": {"blackjax._version": [[0, "module-blackjax._version"]], "Module Contents": [[0, "module-contents"], [1, "module-contents"], [2, "module-contents"], [4, "module-contents"], [5, "module-contents"], [6, "module-contents"], [7, "module-contents"], [8, "module-contents"], [9, "module-contents"], [10, "module-contents"], [12, "module-contents"], [13, "module-contents"], [14, "module-contents"], [15, "module-contents"], [17, "module-contents"], [18, "module-contents"], [19, "module-contents"], [20, "module-contents"], [21, "module-contents"], [22, "module-contents"], [23, "module-contents"], [24, "module-contents"], [25, "module-contents"], [26, "module-contents"], [27, "module-contents"], [29, "module-contents"], [30, "module-contents"], [31, "module-contents"], [32, "module-contents"], [33, "module-contents"], [35, "module-contents"], [36, "module-contents"], [37, "module-contents"], [38, "module-contents"], [39, "module-contents"], [40, "module-contents"], [42, "module-contents"], [43, "module-contents"], [44, "module-contents"], [45, "module-contents"], [46, "module-contents"], [48, "module-contents"], [49, "module-contents"], [50, "module-contents"]], "blackjax.adaptation.base": [[1, "module-blackjax.adaptation.base"]], "Classes": [[1, "classes"], [2, "classes"], [4, "classes"], [5, "classes"], [6, "classes"], [7, "classes"], [8, "classes"], [9, "classes"], [11, "classes"], [13, "classes"], [14, "classes"], [15, "classes"], [18, "classes"], [19, "classes"], [21, "classes"], [22, "classes"], [23, "classes"], [24, "classes"], [25, "classes"], [26, "classes"], [27, "classes"], [29, "classes"], [31, "classes"], [35, "classes"], [36, "classes"], [37, "classes"], [38, "classes"], [39, "classes"], [44, "classes"], [48, "classes"], [49, "classes"], [50, "classes"]], "blackjax.adaptation.chees_adaptation": [[2, "module-blackjax.adaptation.chees_adaptation"]], "Functions": [[2, "functions"], [4, "functions"], [5, "functions"], [6, "functions"], [7, "functions"], [8, "functions"], [10, "functions"], [11, "functions"], [12, "functions"], [13, "functions"], [14, "functions"], [15, "functions"], [18, "functions"], [19, "functions"], [20, "functions"], [21, "functions"], [22, "functions"], [23, "functions"], [24, "functions"], [25, "functions"], [26, "functions"], [27, "functions"], [29, "functions"], [30, "functions"], [31, "functions"], [32, "functions"], [33, "functions"], [34, "functions"], [35, "functions"], [36, "functions"], [37, "functions"], [38, "functions"], [39, "functions"], [40, "functions"], [42, "functions"], [43, "functions"], [44, "functions"], [46, "functions"], [48, "functions"], [49, "functions"], [50, "functions"]], "Attributes": [[2, "attributes"], [9, "attributes"], [11, "attributes"], [21, "attributes"], [23, "attributes"]], "blackjax.adaptation": [[3, "module-blackjax.adaptation"]], "Submodules": [[3, "submodules"], [11, "submodules"], [16, "submodules"], [28, "submodules"], [34, "submodules"], [41, "submodules"], [47, "submodules"]], "blackjax.adaptation.mass_matrix": [[4, "module-blackjax.adaptation.mass_matrix"]], "blackjax.adaptation.meads_adaptation": [[5, "module-blackjax.adaptation.meads_adaptation"]], "blackjax.adaptation.pathfinder_adaptation": [[6, "module-blackjax.adaptation.pathfinder_adaptation"]], "blackjax.adaptation.step_size": [[7, "module-blackjax.adaptation.step_size"]], "blackjax.adaptation.window_adaptation": [[8, "module-blackjax.adaptation.window_adaptation"]], "blackjax.base": [[9, "module-blackjax.base"]], "blackjax.diagnostics": [[10, "module-blackjax.diagnostics"]], "blackjax": [[11, "module-blackjax"]], "Subpackages": [[11, "subpackages"]], "Package Contents": [[11, "package-contents"], [34, "package-contents"]], "blackjax.mcmc.diffusions": [[12, "module-blackjax.mcmc.diffusions"]], "blackjax.mcmc.elliptical_slice": [[13, "module-blackjax.mcmc.elliptical_slice"]], "blackjax.mcmc.ghmc": [[14, "module-blackjax.mcmc.ghmc"]], "blackjax.mcmc.hmc": [[15, "module-blackjax.mcmc.hmc"]], "blackjax.mcmc": [[16, "module-blackjax.mcmc"]], "blackjax.mcmc.integrators": [[17, "module-blackjax.mcmc.integrators"]], "blackjax.mcmc.mala": [[18, "module-blackjax.mcmc.mala"]], "blackjax.mcmc.marginal_latent_gaussian": [[19, "module-blackjax.mcmc.marginal_latent_gaussian"]], "blackjax.mcmc.metrics": [[20, "module-blackjax.mcmc.metrics"]], "blackjax.mcmc.nuts": [[21, "module-blackjax.mcmc.nuts"]], "blackjax.mcmc.periodic_orbital": [[22, "module-blackjax.mcmc.periodic_orbital"]], "blackjax.mcmc.proposal": [[23, "module-blackjax.mcmc.proposal"]], "blackjax.mcmc.random_walk": [[24, "module-blackjax.mcmc.random_walk"]], "Parameter": [[24, "parameter"]], "blackjax.mcmc.termination": [[25, "module-blackjax.mcmc.termination"]], "blackjax.mcmc.trajectory": [[26, "module-blackjax.mcmc.trajectory"]], "blackjax.optimizers.dual_averaging": [[27, "module-blackjax.optimizers.dual_averaging"]], "blackjax.optimizers": [[28, "module-blackjax.optimizers"]], "blackjax.optimizers.lbfgs": [[29, "module-blackjax.optimizers.lbfgs"]], "blackjax.progress_bar": [[30, "module-blackjax.progress_bar"]], "blackjax.sgmcmc.csgld": [[31, "module-blackjax.sgmcmc.csgld"]], "blackjax.sgmcmc.diffusions": [[32, "module-blackjax.sgmcmc.diffusions"]], "blackjax.sgmcmc.gradients": [[33, "module-blackjax.sgmcmc.gradients"]], "blackjax.sgmcmc": [[34, "module-blackjax.sgmcmc"]], "blackjax.sgmcmc.sghmc": [[35, "module-blackjax.sgmcmc.sghmc"]], "blackjax.sgmcmc.sgld": [[36, "module-blackjax.sgmcmc.sgld"]], "blackjax.sgmcmc.sgnht": [[37, "module-blackjax.sgmcmc.sgnht"]], "blackjax.smc.adaptive_tempered": [[38, "module-blackjax.smc.adaptive_tempered"]], "blackjax.smc.base": [[39, "module-blackjax.smc.base"]], "blackjax.smc.ess": [[40, "module-blackjax.smc.ess"]], "blackjax.smc": [[41, "module-blackjax.smc"]], "blackjax.smc.resampling": [[42, "module-blackjax.smc.resampling"]], "blackjax.smc.solver": [[43, "module-blackjax.smc.solver"]], "blackjax.smc.tempered": [[44, "module-blackjax.smc.tempered"]], "blackjax.types": [[45, "module-blackjax.types"]], "blackjax.util": [[46, "module-blackjax.util"]], "blackjax.vi": [[47, "module-blackjax.vi"]], "blackjax.vi.meanfield_vi": [[48, "module-blackjax.vi.meanfield_vi"]], "blackjax.vi.pathfinder": [[49, "module-blackjax.vi.pathfinder"]], "blackjax.vi.svgd": [[50, "module-blackjax.vi.svgd"]], "Bibliography": [[51, "bibliography"]], "Use custom gradients": [[52, "use-custom-gradients"]], "Functions defined as the minimum of another function": [[52, "functions-defined-as-the-minimum-of-another-function"]], "Trying to differentate the function with jax.grad": [[52, "trying-to-differentate-the-function-with-jax-grad"]], "Deriving the gradient mathematically": [[52, "deriving-the-gradient-mathematically"]], "Telling JAX to use a custom gradient": [[52, "telling-jax-to-use-a-custom-gradient"]], "Making sure the result is correct": [[52, "making-sure-the-result-is-correct"]], "Using the function with Blackjax": [[52, "using-the-function-with-blackjax"]], "How to build a Metropolis-Within-Gibbs sampler?": [[53, "how-to-build-a-metropolis-within-gibbs-sampler"]], "The Model": [[53, "the-model"]], "MWG Sampling in BlackJAX": [[53, "mwg-sampling-in-blackjax"]], "Construct the MWG Kernel": [[53, "construct-the-mwg-kernel"]], "Sampler Parameters": [[53, "sampler-parameters"], [61, "sampler-parameters"]], "Set the Initial State of Each Algorithm": [[53, "set-the-initial-state-of-each-algorithm"]], "Build the Sampling Loop": [[53, "build-the-sampling-loop"], [53, "id1"]], "Sampling": [[53, "sampling"], [53, "id2"]], "General MWG Kernel": [[53, "general-mwg-kernel"]], "Check Result": [[53, "check-result"]], "Developer Notes": [[53, "developer-notes"]], "Use a logdensity function that is not compatible with JAX\u2019s primitives": [[54, "use-a-logdensity-function-that-is-not-compatible-with-jax-s-primitives"]], "Aesara model compiled to Numba": [[54, "aesara-model-compiled-to-numba"]], "Use jax.experimental.host_callback to call Numba functions": [[54, "use-jax-experimental-host-callback-to-call-numba-functions"]], "Use custom XLA calls to call Numba functions faster": [[54, "use-custom-xla-calls-to-call-numba-functions-faster"]], "Sample with multiple chains in parallel": [[55, "sample-with-multiple-chains-in-parallel"]], "Vectorization vs parallelization": [[55, "vectorization-vs-parallelization"]], "NUTS in parallel": [[55, "nuts-in-parallel"]], "Using jax.vmap": [[55, "using-jax-vmap"]], "Using jax.pmap": [[55, "using-jax-pmap"]], "A note on using jax.pmap on CPU": [[55, "a-note-on-using-jax-pmap-on-cpu"]], "Choosing the number of devices": [[55, "choosing-the-number-of-devices"]], "Back to our example": [[55, "back-to-our-example"]], "Conclusions": [[55, "conclusions"]], "Use with Aesara models": [[56, "use-with-aesara-models"]], "Before you start": [[56, null], [57, null], [59, null], [60, null]], "Use with Numpyro models": [[57, "use-with-numpyro-models"]], "Use with Oryx models": [[58, "use-with-oryx-models"]], "Use with PyMC models": [[59, "use-with-pymc-models"]], "Use with TFP models": [[60, "use-with-tfp-models"]], "A Quick Introduction to Blackjax": [[61, "a-quick-introduction-to-blackjax"]], "The Problem": [[61, "the-problem"]], "HMC": [[61, "hmc"]], "Set the Initial State": [[61, "set-the-initial-state"]], "Build the Kernel and Inference Loop": [[61, "build-the-kernel-and-inference-loop"]], "Inference": [[61, "inference"]], "NUTS": [[61, "nuts"]], "Use Stan\u2019s Window Adaptation": [[61, "use-stan-s-window-adaptation"]], "Welcome to Blackjax!": [[62, "welcome-to-blackjax"]], "Hello World": [[62, "hello-world"]], "Installation": [[62, "installation"]], "GPU instructions": [[62, null]]}, "indexentries": {"type_checking (in module blackjax._version)": [[0, "blackjax._version.TYPE_CHECKING"]], "version_tuple (in module blackjax._version)": [[0, "blackjax._version.VERSION_TUPLE"], [0, "blackjax._version.version_tuple"]], "blackjax._version": [[0, "module-blackjax._version"]], "module": [[0, "module-blackjax._version"], [1, "module-blackjax.adaptation.base"], [2, "module-blackjax.adaptation.chees_adaptation"], [3, "module-blackjax.adaptation"], [4, "module-blackjax.adaptation.mass_matrix"], [5, "module-blackjax.adaptation.meads_adaptation"], [6, "module-blackjax.adaptation.pathfinder_adaptation"], [7, "module-blackjax.adaptation.step_size"], [8, "module-blackjax.adaptation.window_adaptation"], [9, "module-blackjax.base"], [10, "module-blackjax.diagnostics"], [11, "module-blackjax"], [12, "module-blackjax.mcmc.diffusions"], [13, "module-blackjax.mcmc.elliptical_slice"], [14, "module-blackjax.mcmc.ghmc"], [15, "module-blackjax.mcmc.hmc"], [16, "module-blackjax.mcmc"], [17, "module-blackjax.mcmc.integrators"], [18, "module-blackjax.mcmc.mala"], [19, "module-blackjax.mcmc.marginal_latent_gaussian"], [20, "module-blackjax.mcmc.metrics"], [21, "module-blackjax.mcmc.nuts"], [22, "module-blackjax.mcmc.periodic_orbital"], [23, "module-blackjax.mcmc.proposal"], [24, "module-blackjax.mcmc.random_walk"], [25, "module-blackjax.mcmc.termination"], [26, "module-blackjax.mcmc.trajectory"], [27, "module-blackjax.optimizers.dual_averaging"], [28, "module-blackjax.optimizers"], [29, "module-blackjax.optimizers.lbfgs"], [30, "module-blackjax.progress_bar"], [31, "module-blackjax.sgmcmc.csgld"], [32, "module-blackjax.sgmcmc.diffusions"], [33, "module-blackjax.sgmcmc.gradients"], [34, "module-blackjax.sgmcmc"], [35, "module-blackjax.sgmcmc.sghmc"], [36, "module-blackjax.sgmcmc.sgld"], [37, "module-blackjax.sgmcmc.sgnht"], [38, "module-blackjax.smc.adaptive_tempered"], [39, "module-blackjax.smc.base"], [40, "module-blackjax.smc.ess"], [41, "module-blackjax.smc"], [42, "module-blackjax.smc.resampling"], [43, "module-blackjax.smc.solver"], [44, "module-blackjax.smc.tempered"], [45, "module-blackjax.types"], [46, "module-blackjax.util"], [47, "module-blackjax.vi"], [48, "module-blackjax.vi.meanfield_vi"], [49, "module-blackjax.vi.pathfinder"], [50, "module-blackjax.vi.svgd"]], "version (in module blackjax._version)": [[0, "blackjax._version.version"]], "adaptationinfo (class in blackjax.adaptation.base)": [[1, "blackjax.adaptation.base.AdaptationInfo"]], "adaptationresults (class in blackjax.adaptation.base)": [[1, "blackjax.adaptation.base.AdaptationResults"]], "adaptation_state (adaptationinfo attribute)": [[1, "blackjax.adaptation.base.AdaptationInfo.adaptation_state"]], "blackjax.adaptation.base": [[1, "module-blackjax.adaptation.base"]], "info (adaptationinfo attribute)": [[1, "blackjax.adaptation.base.AdaptationInfo.info"]], "parameters (adaptationresults attribute)": [[1, "blackjax.adaptation.base.AdaptationResults.parameters"]], "state (adaptationinfo attribute)": [[1, "blackjax.adaptation.base.AdaptationInfo.state"]], "state (adaptationresults attribute)": [[1, "blackjax.adaptation.base.AdaptationResults.state"]], "cheesadaptationstate (class in blackjax.adaptation.chees_adaptation)": [[2, "blackjax.adaptation.chees_adaptation.ChEESAdaptationState"]], "optimal_target_acceptance_rate (in module blackjax.adaptation.chees_adaptation)": [[2, "blackjax.adaptation.chees_adaptation.OPTIMAL_TARGET_ACCEPTANCE_RATE"]], "base() (in module blackjax.adaptation.chees_adaptation)": [[2, "blackjax.adaptation.chees_adaptation.base"]], "blackjax.adaptation.chees_adaptation": [[2, "module-blackjax.adaptation.chees_adaptation"]], "chees_adaptation() (in module blackjax.adaptation.chees_adaptation)": [[2, "blackjax.adaptation.chees_adaptation.chees_adaptation"]], "da_state (cheesadaptationstate attribute)": [[2, "blackjax.adaptation.chees_adaptation.ChEESAdaptationState.da_state"]], "log_step_size_moving_average (cheesadaptationstate attribute)": [[2, "blackjax.adaptation.chees_adaptation.ChEESAdaptationState.log_step_size_moving_average"]], "log_trajectory_length_moving_average (cheesadaptationstate attribute)": [[2, "blackjax.adaptation.chees_adaptation.ChEESAdaptationState.log_trajectory_length_moving_average"]], "optim_state (cheesadaptationstate attribute)": [[2, "blackjax.adaptation.chees_adaptation.ChEESAdaptationState.optim_state"]], "random_generator_arg (cheesadaptationstate attribute)": [[2, "blackjax.adaptation.chees_adaptation.ChEESAdaptationState.random_generator_arg"]], "step (cheesadaptationstate attribute)": [[2, "blackjax.adaptation.chees_adaptation.ChEESAdaptationState.step"]], "step_size (cheesadaptationstate attribute)": [[2, "blackjax.adaptation.chees_adaptation.ChEESAdaptationState.step_size"]], "trajectory_length (cheesadaptationstate attribute)": [[2, "blackjax.adaptation.chees_adaptation.ChEESAdaptationState.trajectory_length"]], "blackjax.adaptation": [[3, "module-blackjax.adaptation"]], "massmatrixadaptationstate (class in blackjax.adaptation.mass_matrix)": [[4, "blackjax.adaptation.mass_matrix.MassMatrixAdaptationState"]], "welfordalgorithmstate (class in blackjax.adaptation.mass_matrix)": [[4, "blackjax.adaptation.mass_matrix.WelfordAlgorithmState"]], "blackjax.adaptation.mass_matrix": [[4, "module-blackjax.adaptation.mass_matrix"]], "inverse_mass_matrix (massmatrixadaptationstate attribute)": [[4, "blackjax.adaptation.mass_matrix.MassMatrixAdaptationState.inverse_mass_matrix"]], "m2 (welfordalgorithmstate attribute)": [[4, "blackjax.adaptation.mass_matrix.WelfordAlgorithmState.m2"]], "mass_matrix_adaptation() (in module blackjax.adaptation.mass_matrix)": [[4, "blackjax.adaptation.mass_matrix.mass_matrix_adaptation"]], "mean (welfordalgorithmstate attribute)": [[4, "blackjax.adaptation.mass_matrix.WelfordAlgorithmState.mean"]], "sample_size (welfordalgorithmstate attribute)": [[4, "blackjax.adaptation.mass_matrix.WelfordAlgorithmState.sample_size"]], "wc_state (massmatrixadaptationstate attribute)": [[4, "blackjax.adaptation.mass_matrix.MassMatrixAdaptationState.wc_state"]], "welford_algorithm() (in module blackjax.adaptation.mass_matrix)": [[4, "blackjax.adaptation.mass_matrix.welford_algorithm"]], "meadsadaptationstate (class in blackjax.adaptation.meads_adaptation)": [[5, "blackjax.adaptation.meads_adaptation.MEADSAdaptationState"]], "alpha (meadsadaptationstate attribute)": [[5, "blackjax.adaptation.meads_adaptation.MEADSAdaptationState.alpha"]], "base() (in module blackjax.adaptation.meads_adaptation)": [[5, "blackjax.adaptation.meads_adaptation.base"]], "blackjax.adaptation.meads_adaptation": [[5, "module-blackjax.adaptation.meads_adaptation"]], "current_iteration (meadsadaptationstate attribute)": [[5, "blackjax.adaptation.meads_adaptation.MEADSAdaptationState.current_iteration"]], "delta (meadsadaptationstate attribute)": [[5, "blackjax.adaptation.meads_adaptation.MEADSAdaptationState.delta"]], "maximum_eigenvalue() (in module blackjax.adaptation.meads_adaptation)": [[5, "blackjax.adaptation.meads_adaptation.maximum_eigenvalue"]], "meads_adaptation() (in module blackjax.adaptation.meads_adaptation)": [[5, "blackjax.adaptation.meads_adaptation.meads_adaptation"]], "position_sigma (meadsadaptationstate attribute)": [[5, "blackjax.adaptation.meads_adaptation.MEADSAdaptationState.position_sigma"]], "step_size (meadsadaptationstate attribute)": [[5, "blackjax.adaptation.meads_adaptation.MEADSAdaptationState.step_size"]], "pathfinderadaptationstate (class in blackjax.adaptation.pathfinder_adaptation)": [[6, "blackjax.adaptation.pathfinder_adaptation.PathfinderAdaptationState"]], "base() (in module blackjax.adaptation.pathfinder_adaptation)": [[6, "blackjax.adaptation.pathfinder_adaptation.base"]], "blackjax.adaptation.pathfinder_adaptation": [[6, "module-blackjax.adaptation.pathfinder_adaptation"]], "inverse_mass_matrix (pathfinderadaptationstate attribute)": [[6, "blackjax.adaptation.pathfinder_adaptation.PathfinderAdaptationState.inverse_mass_matrix"]], "pathfinder_adaptation() (in module blackjax.adaptation.pathfinder_adaptation)": [[6, "blackjax.adaptation.pathfinder_adaptation.pathfinder_adaptation"]], "ss_state (pathfinderadaptationstate attribute)": [[6, "blackjax.adaptation.pathfinder_adaptation.PathfinderAdaptationState.ss_state"]], "step_size (pathfinderadaptationstate attribute)": [[6, "blackjax.adaptation.pathfinder_adaptation.PathfinderAdaptationState.step_size"]], "dualaveragingadaptationstate (class in blackjax.adaptation.step_size)": [[7, "blackjax.adaptation.step_size.DualAveragingAdaptationState"]], "avg_error (dualaveragingadaptationstate attribute)": [[7, "blackjax.adaptation.step_size.DualAveragingAdaptationState.avg_error"]], "blackjax.adaptation.step_size": [[7, "module-blackjax.adaptation.step_size"]], "dual_averaging_adaptation() (in module blackjax.adaptation.step_size)": [[7, "blackjax.adaptation.step_size.dual_averaging_adaptation"]], "find_reasonable_step_size() (in module blackjax.adaptation.step_size)": [[7, "blackjax.adaptation.step_size.find_reasonable_step_size"]], "log_step_size (dualaveragingadaptationstate attribute)": [[7, "blackjax.adaptation.step_size.DualAveragingAdaptationState.log_step_size"]], "log_step_size_avg (dualaveragingadaptationstate attribute)": [[7, "blackjax.adaptation.step_size.DualAveragingAdaptationState.log_step_size_avg"]], "mu (dualaveragingadaptationstate attribute)": [[7, "blackjax.adaptation.step_size.DualAveragingAdaptationState.mu"]], "step (dualaveragingadaptationstate attribute)": [[7, "blackjax.adaptation.step_size.DualAveragingAdaptationState.step"]], "windowadaptationstate (class in blackjax.adaptation.window_adaptation)": [[8, "blackjax.adaptation.window_adaptation.WindowAdaptationState"]], "base() (in module blackjax.adaptation.window_adaptation)": [[8, "blackjax.adaptation.window_adaptation.base"]], "blackjax.adaptation.window_adaptation": [[8, "module-blackjax.adaptation.window_adaptation"]], "build_schedule() (in module blackjax.adaptation.window_adaptation)": [[8, "blackjax.adaptation.window_adaptation.build_schedule"]], "imm_state (windowadaptationstate attribute)": [[8, "blackjax.adaptation.window_adaptation.WindowAdaptationState.imm_state"]], "inverse_mass_matrix (windowadaptationstate attribute)": [[8, "blackjax.adaptation.window_adaptation.WindowAdaptationState.inverse_mass_matrix"]], "ss_state (windowadaptationstate attribute)": [[8, "blackjax.adaptation.window_adaptation.WindowAdaptationState.ss_state"]], "step_size (windowadaptationstate attribute)": [[8, "blackjax.adaptation.window_adaptation.WindowAdaptationState.step_size"]], "window_adaptation() (in module blackjax.adaptation.window_adaptation)": [[8, "blackjax.adaptation.window_adaptation.window_adaptation"]], "adaptationalgorithm (class in blackjax.base)": [[9, "blackjax.base.AdaptationAlgorithm"]], "info (in module blackjax.base)": [[9, "blackjax.base.Info"]], "initfn (class in blackjax.base)": [[9, "blackjax.base.InitFn"]], "position (in module blackjax.base)": [[9, "blackjax.base.Position"]], "runfn (class in blackjax.base)": [[9, "blackjax.base.RunFn"]], "samplingalgorithm (class in blackjax.base)": [[9, "blackjax.base.SamplingAlgorithm"]], "state (in module blackjax.base)": [[9, "blackjax.base.State"]], "updatefn (class in blackjax.base)": [[9, "blackjax.base.UpdateFn"]], "vialgorithm (class in blackjax.base)": [[9, "blackjax.base.VIAlgorithm"]], "blackjax.base": [[9, "module-blackjax.base"]], "init (samplingalgorithm attribute)": [[9, "blackjax.base.SamplingAlgorithm.init"]], "init (vialgorithm attribute)": [[9, "blackjax.base.VIAlgorithm.init"]], "run (adaptationalgorithm attribute)": [[9, "blackjax.base.AdaptationAlgorithm.run"]], "sample (vialgorithm attribute)": [[9, "blackjax.base.VIAlgorithm.sample"]], "step (samplingalgorithm attribute)": [[9, "blackjax.base.SamplingAlgorithm.step"]], "step (vialgorithm attribute)": [[9, "blackjax.base.VIAlgorithm.step"]], "blackjax.diagnostics": [[10, "module-blackjax.diagnostics"]], "effective_sample_size() (in module blackjax.diagnostics)": [[10, "blackjax.diagnostics.effective_sample_size"]], "potential_scale_reduction() (in module blackjax.diagnostics)": [[10, "blackjax.diagnostics.potential_scale_reduction"]], "blackjax": [[11, "module-blackjax"]], "blackjax.mcmc.diffusions": [[12, "module-blackjax.mcmc.diffusions"]], "overdamped_langevin() (in module blackjax.mcmc.diffusions)": [[12, "blackjax.mcmc.diffusions.overdamped_langevin"]], "ellipsliceinfo (class in blackjax.mcmc.elliptical_slice)": [[13, "blackjax.mcmc.elliptical_slice.EllipSliceInfo"]], "ellipslicestate (class in blackjax.mcmc.elliptical_slice)": [[13, "blackjax.mcmc.elliptical_slice.EllipSliceState"]], "blackjax.mcmc.elliptical_slice": [[13, "module-blackjax.mcmc.elliptical_slice"]], "build_kernel (elliptical_slice attribute)": [[13, "blackjax.mcmc.elliptical_slice.elliptical_slice.build_kernel"]], "build_kernel() (in module blackjax.mcmc.elliptical_slice)": [[13, "blackjax.mcmc.elliptical_slice.build_kernel"]], "elliptical_slice (class in blackjax.mcmc.elliptical_slice)": [[13, "blackjax.mcmc.elliptical_slice.elliptical_slice"]], "init (elliptical_slice attribute)": [[13, "blackjax.mcmc.elliptical_slice.elliptical_slice.init"]], "init() (in module blackjax.mcmc.elliptical_slice)": [[13, "blackjax.mcmc.elliptical_slice.init"]], "logdensity (ellipslicestate attribute)": [[13, "blackjax.mcmc.elliptical_slice.EllipSliceState.logdensity"]], "momentum (ellipsliceinfo attribute)": [[13, "blackjax.mcmc.elliptical_slice.EllipSliceInfo.momentum"]], "position (ellipslicestate attribute)": [[13, "blackjax.mcmc.elliptical_slice.EllipSliceState.position"]], "subiter (ellipsliceinfo attribute)": [[13, "blackjax.mcmc.elliptical_slice.EllipSliceInfo.subiter"]], "theta (ellipsliceinfo attribute)": [[13, "blackjax.mcmc.elliptical_slice.EllipSliceInfo.theta"]], "ghmcstate (class in blackjax.mcmc.ghmc)": [[14, "blackjax.mcmc.ghmc.GHMCState"]], "blackjax.mcmc.ghmc": [[14, "module-blackjax.mcmc.ghmc"]], "build_kernel (ghmc attribute)": [[14, "blackjax.mcmc.ghmc.ghmc.build_kernel"]], "build_kernel() (in module blackjax.mcmc.ghmc)": [[14, "blackjax.mcmc.ghmc.build_kernel"]], "ghmc (class in blackjax.mcmc.ghmc)": [[14, "blackjax.mcmc.ghmc.ghmc"]], "init (ghmc attribute)": [[14, "blackjax.mcmc.ghmc.ghmc.init"]], "init() (in module blackjax.mcmc.ghmc)": [[14, "blackjax.mcmc.ghmc.init"]], "logdensity (ghmcstate attribute)": [[14, "blackjax.mcmc.ghmc.GHMCState.logdensity"]], "logdensity_grad (ghmcstate attribute)": [[14, "blackjax.mcmc.ghmc.GHMCState.logdensity_grad"]], "momentum (ghmcstate attribute)": [[14, "blackjax.mcmc.ghmc.GHMCState.momentum"]], "position (ghmcstate attribute)": [[14, "blackjax.mcmc.ghmc.GHMCState.position"]], "slice (ghmcstate attribute)": [[14, "blackjax.mcmc.ghmc.GHMCState.slice"]], "dynamichmcstate (class in blackjax.mcmc.hmc)": [[15, "blackjax.mcmc.hmc.DynamicHMCState"]], "hmcinfo (class in blackjax.mcmc.hmc)": [[15, "blackjax.mcmc.hmc.HMCInfo"]], "hmcstate (class in blackjax.mcmc.hmc)": [[15, "blackjax.mcmc.hmc.HMCState"]], "acceptance_rate (hmcinfo attribute)": [[15, "blackjax.mcmc.hmc.HMCInfo.acceptance_rate"]], "blackjax.mcmc.hmc": [[15, "module-blackjax.mcmc.hmc"]], "build_dynamic_kernel() (in module blackjax.mcmc.hmc)": [[15, "blackjax.mcmc.hmc.build_dynamic_kernel"]], "build_kernel (dynamic_hmc attribute)": [[15, "blackjax.mcmc.hmc.dynamic_hmc.build_kernel"]], "build_kernel (hmc attribute)": [[15, "blackjax.mcmc.hmc.hmc.build_kernel"]], "build_kernel() (in module blackjax.mcmc.hmc)": [[15, "blackjax.mcmc.hmc.build_kernel"]], "dynamic_hmc (class in blackjax.mcmc.hmc)": [[15, "blackjax.mcmc.hmc.dynamic_hmc"]], "energy (hmcinfo attribute)": [[15, "blackjax.mcmc.hmc.HMCInfo.energy"]], "hmc (class in blackjax.mcmc.hmc)": [[15, "blackjax.mcmc.hmc.hmc"]], "init (dynamic_hmc attribute)": [[15, "blackjax.mcmc.hmc.dynamic_hmc.init"]], "init (hmc attribute)": [[15, "blackjax.mcmc.hmc.hmc.init"]], "init() (in module blackjax.mcmc.hmc)": [[15, "blackjax.mcmc.hmc.init"]], "init_dynamic() (in module blackjax.mcmc.hmc)": [[15, "blackjax.mcmc.hmc.init_dynamic"]], "is_accepted (hmcinfo attribute)": [[15, "blackjax.mcmc.hmc.HMCInfo.is_accepted"]], "is_divergent (hmcinfo attribute)": [[15, "blackjax.mcmc.hmc.HMCInfo.is_divergent"]], "logdensity (dynamichmcstate attribute)": [[15, "blackjax.mcmc.hmc.DynamicHMCState.logdensity"]], "logdensity (hmcstate attribute)": [[15, "blackjax.mcmc.hmc.HMCState.logdensity"]], "logdensity_grad (dynamichmcstate attribute)": [[15, "blackjax.mcmc.hmc.DynamicHMCState.logdensity_grad"]], "logdensity_grad (hmcstate attribute)": [[15, "blackjax.mcmc.hmc.HMCState.logdensity_grad"]], "momentum (hmcinfo attribute)": [[15, "blackjax.mcmc.hmc.HMCInfo.momentum"]], "num_integration_steps (hmcinfo attribute)": [[15, "blackjax.mcmc.hmc.HMCInfo.num_integration_steps"]], "position (dynamichmcstate attribute)": [[15, "blackjax.mcmc.hmc.DynamicHMCState.position"]], "position (hmcstate attribute)": [[15, "blackjax.mcmc.hmc.HMCState.position"]], "proposal (hmcinfo attribute)": [[15, "blackjax.mcmc.hmc.HMCInfo.proposal"]], "random_generator_arg (dynamichmcstate attribute)": [[15, "blackjax.mcmc.hmc.DynamicHMCState.random_generator_arg"]], "blackjax.mcmc": [[16, "module-blackjax.mcmc"]], "blackjax.mcmc.integrators": [[17, "module-blackjax.mcmc.integrators"]], "mclachlan (in module blackjax.mcmc.integrators)": [[17, "blackjax.mcmc.integrators.mclachlan"]], "noneuclidean_leapfrog (in module blackjax.mcmc.integrators)": [[17, "blackjax.mcmc.integrators.noneuclidean_leapfrog"]], "noneuclidean_mclachlan (in module blackjax.mcmc.integrators)": [[17, "blackjax.mcmc.integrators.noneuclidean_mclachlan"]], "noneuclidean_yoshida (in module blackjax.mcmc.integrators)": [[17, "blackjax.mcmc.integrators.noneuclidean_yoshida"]], "velocity_verlet (in module blackjax.mcmc.integrators)": [[17, "blackjax.mcmc.integrators.velocity_verlet"]], "yoshida (in module blackjax.mcmc.integrators)": [[17, "blackjax.mcmc.integrators.yoshida"]], "malainfo (class in blackjax.mcmc.mala)": [[18, "blackjax.mcmc.mala.MALAInfo"]], "malastate (class in blackjax.mcmc.mala)": [[18, "blackjax.mcmc.mala.MALAState"]], "acceptance_rate (malainfo attribute)": [[18, "blackjax.mcmc.mala.MALAInfo.acceptance_rate"]], "blackjax.mcmc.mala": [[18, "module-blackjax.mcmc.mala"]], "build_kernel (mala attribute)": [[18, "blackjax.mcmc.mala.mala.build_kernel"]], "build_kernel() (in module blackjax.mcmc.mala)": [[18, "blackjax.mcmc.mala.build_kernel"]], "init (mala attribute)": [[18, "blackjax.mcmc.mala.mala.init"]], "init() (in module blackjax.mcmc.mala)": [[18, "blackjax.mcmc.mala.init"]], "is_accepted (malainfo attribute)": [[18, "blackjax.mcmc.mala.MALAInfo.is_accepted"]], "logdensity (malastate attribute)": [[18, "blackjax.mcmc.mala.MALAState.logdensity"]], "logdensity_grad (malastate attribute)": [[18, "blackjax.mcmc.mala.MALAState.logdensity_grad"]], "mala (class in blackjax.mcmc.mala)": [[18, "blackjax.mcmc.mala.mala"]], "position (malastate attribute)": [[18, "blackjax.mcmc.mala.MALAState.position"]], "marginalinfo (class in blackjax.mcmc.marginal_latent_gaussian)": [[19, "blackjax.mcmc.marginal_latent_gaussian.MarginalInfo"]], "marginalstate (class in blackjax.mcmc.marginal_latent_gaussian)": [[19, "blackjax.mcmc.marginal_latent_gaussian.MarginalState"]], "u_grad_x (marginalstate attribute)": [[19, "blackjax.mcmc.marginal_latent_gaussian.MarginalState.U_grad_x"]], "u_x (marginalstate attribute)": [[19, "blackjax.mcmc.marginal_latent_gaussian.MarginalState.U_x"]], "acceptance_rate (marginalinfo attribute)": [[19, "blackjax.mcmc.marginal_latent_gaussian.MarginalInfo.acceptance_rate"]], "blackjax.mcmc.marginal_latent_gaussian": [[19, "module-blackjax.mcmc.marginal_latent_gaussian"]], "init_and_kernel() (in module blackjax.mcmc.marginal_latent_gaussian)": [[19, "blackjax.mcmc.marginal_latent_gaussian.init_and_kernel"]], "is_accepted (marginalinfo attribute)": [[19, "blackjax.mcmc.marginal_latent_gaussian.MarginalInfo.is_accepted"]], "logdensity (marginalstate attribute)": [[19, "blackjax.mcmc.marginal_latent_gaussian.MarginalState.logdensity"]], "logdensity_grad (marginalstate attribute)": [[19, "blackjax.mcmc.marginal_latent_gaussian.MarginalState.logdensity_grad"]], "mgrad_gaussian (class in blackjax.mcmc.marginal_latent_gaussian)": [[19, "blackjax.mcmc.marginal_latent_gaussian.mgrad_gaussian"]], "position (marginalstate attribute)": [[19, "blackjax.mcmc.marginal_latent_gaussian.MarginalState.position"]], "proposal (marginalinfo attribute)": [[19, "blackjax.mcmc.marginal_latent_gaussian.MarginalInfo.proposal"]], "blackjax.mcmc.metrics": [[20, "module-blackjax.mcmc.metrics"]], "gaussian_euclidean() (in module blackjax.mcmc.metrics)": [[20, "blackjax.mcmc.metrics.gaussian_euclidean"]], "nutsinfo (class in blackjax.mcmc.nuts)": [[21, "blackjax.mcmc.nuts.NUTSInfo"]], "acceptance_rate (nutsinfo attribute)": [[21, "blackjax.mcmc.nuts.NUTSInfo.acceptance_rate"]], "blackjax.mcmc.nuts": [[21, "module-blackjax.mcmc.nuts"]], "build_kernel (nuts attribute)": [[21, "blackjax.mcmc.nuts.nuts.build_kernel"]], "build_kernel() (in module blackjax.mcmc.nuts)": [[21, "blackjax.mcmc.nuts.build_kernel"]], "energy (nutsinfo attribute)": [[21, "blackjax.mcmc.nuts.NUTSInfo.energy"]], "init (in module blackjax.mcmc.nuts)": [[21, "blackjax.mcmc.nuts.init"]], "init (nuts attribute)": [[21, "blackjax.mcmc.nuts.nuts.init"]], "is_divergent (nutsinfo attribute)": [[21, "blackjax.mcmc.nuts.NUTSInfo.is_divergent"]], "is_turning (nutsinfo attribute)": [[21, "blackjax.mcmc.nuts.NUTSInfo.is_turning"]], "momentum (nutsinfo attribute)": [[21, "blackjax.mcmc.nuts.NUTSInfo.momentum"]], "num_integration_steps (nutsinfo attribute)": [[21, "blackjax.mcmc.nuts.NUTSInfo.num_integration_steps"]], "num_trajectory_expansions (nutsinfo attribute)": [[21, "blackjax.mcmc.nuts.NUTSInfo.num_trajectory_expansions"]], "nuts (class in blackjax.mcmc.nuts)": [[21, "blackjax.mcmc.nuts.nuts"]], "trajectory_leftmost_state (nutsinfo attribute)": [[21, "blackjax.mcmc.nuts.NUTSInfo.trajectory_leftmost_state"]], "trajectory_rightmost_state (nutsinfo attribute)": [[21, "blackjax.mcmc.nuts.NUTSInfo.trajectory_rightmost_state"]], "periodicorbitalstate (class in blackjax.mcmc.periodic_orbital)": [[22, "blackjax.mcmc.periodic_orbital.PeriodicOrbitalState"]], "blackjax.mcmc.periodic_orbital": [[22, "module-blackjax.mcmc.periodic_orbital"]], "build_kernel (orbital_hmc attribute)": [[22, "blackjax.mcmc.periodic_orbital.orbital_hmc.build_kernel"]], "build_kernel() (in module blackjax.mcmc.periodic_orbital)": [[22, "blackjax.mcmc.periodic_orbital.build_kernel"]], "directions (periodicorbitalstate attribute)": [[22, "blackjax.mcmc.periodic_orbital.PeriodicOrbitalState.directions"]], "init (orbital_hmc attribute)": [[22, "blackjax.mcmc.periodic_orbital.orbital_hmc.init"]], "init() (in module blackjax.mcmc.periodic_orbital)": [[22, "blackjax.mcmc.periodic_orbital.init"]], "logdensities (periodicorbitalstate attribute)": [[22, "blackjax.mcmc.periodic_orbital.PeriodicOrbitalState.logdensities"]], "logdensities_grad (periodicorbitalstate attribute)": [[22, "blackjax.mcmc.periodic_orbital.PeriodicOrbitalState.logdensities_grad"]], "orbital_hmc (class in blackjax.mcmc.periodic_orbital)": [[22, "blackjax.mcmc.periodic_orbital.orbital_hmc"]], "positions (periodicorbitalstate attribute)": [[22, "blackjax.mcmc.periodic_orbital.PeriodicOrbitalState.positions"]], "weights (periodicorbitalstate attribute)": [[22, "blackjax.mcmc.periodic_orbital.PeriodicOrbitalState.weights"]], "proposal (class in blackjax.mcmc.proposal)": [[23, "blackjax.mcmc.proposal.Proposal"]], "sampleproposal (class in blackjax.mcmc.proposal)": [[23, "blackjax.mcmc.proposal.SampleProposal"]], "trajectorystate (in module blackjax.mcmc.proposal)": [[23, "blackjax.mcmc.proposal.TrajectoryState"]], "asymmetric_proposal_generator() (in module blackjax.mcmc.proposal)": [[23, "blackjax.mcmc.proposal.asymmetric_proposal_generator"]], "blackjax.mcmc.proposal": [[23, "module-blackjax.mcmc.proposal"]], "energy (proposal attribute)": [[23, "blackjax.mcmc.proposal.Proposal.energy"]], "nonreversible_slice_sampling() (in module blackjax.mcmc.proposal)": [[23, "blackjax.mcmc.proposal.nonreversible_slice_sampling"]], "progressive_biased_sampling() (in module blackjax.mcmc.proposal)": [[23, "blackjax.mcmc.proposal.progressive_biased_sampling"]], "progressive_uniform_sampling() (in module blackjax.mcmc.proposal)": [[23, "blackjax.mcmc.proposal.progressive_uniform_sampling"]], "proposal_from_energy_diff() (in module blackjax.mcmc.proposal)": [[23, "blackjax.mcmc.proposal.proposal_from_energy_diff"]], "proposal_generator() (in module blackjax.mcmc.proposal)": [[23, "blackjax.mcmc.proposal.proposal_generator"]], "state (proposal attribute)": [[23, "blackjax.mcmc.proposal.Proposal.state"]], "static_binomial_sampling() (in module blackjax.mcmc.proposal)": [[23, "blackjax.mcmc.proposal.static_binomial_sampling"]], "sum_log_p_accept (proposal attribute)": [[23, "blackjax.mcmc.proposal.Proposal.sum_log_p_accept"]], "weight (proposal attribute)": [[23, "blackjax.mcmc.proposal.Proposal.weight"]], "rwinfo (class in blackjax.mcmc.random_walk)": [[24, "blackjax.mcmc.random_walk.RWInfo"]], "rwstate (class in blackjax.mcmc.random_walk)": [[24, "blackjax.mcmc.random_walk.RWState"]], "acceptance_rate (rwinfo attribute)": [[24, "blackjax.mcmc.random_walk.RWInfo.acceptance_rate"]], "additive_step_random_walk (class in blackjax.mcmc.random_walk)": [[24, "blackjax.mcmc.random_walk.additive_step_random_walk"]], "blackjax.mcmc.random_walk": [[24, "module-blackjax.mcmc.random_walk"]], "build_additive_step() (in module blackjax.mcmc.random_walk)": [[24, "blackjax.mcmc.random_walk.build_additive_step"]], "build_irmh() (in module blackjax.mcmc.random_walk)": [[24, "blackjax.mcmc.random_walk.build_irmh"]], "build_kernel (additive_step_random_walk attribute)": [[24, "blackjax.mcmc.random_walk.additive_step_random_walk.build_kernel"]], "build_kernel (irmh attribute)": [[24, "blackjax.mcmc.random_walk.irmh.build_kernel"]], "build_kernel (rmh attribute)": [[24, "blackjax.mcmc.random_walk.rmh.build_kernel"]], "build_rmh() (in module blackjax.mcmc.random_walk)": [[24, "blackjax.mcmc.random_walk.build_rmh"]], "build_rmh_transition_energy() (in module blackjax.mcmc.random_walk)": [[24, "blackjax.mcmc.random_walk.build_rmh_transition_energy"]], "init (additive_step_random_walk attribute)": [[24, "blackjax.mcmc.random_walk.additive_step_random_walk.init"]], "init (irmh attribute)": [[24, "blackjax.mcmc.random_walk.irmh.init"]], "init (rmh attribute)": [[24, "blackjax.mcmc.random_walk.rmh.init"]], "irmh (class in blackjax.mcmc.random_walk)": [[24, "blackjax.mcmc.random_walk.irmh"]], "is_accepted (rwinfo attribute)": [[24, "blackjax.mcmc.random_walk.RWInfo.is_accepted"]], "logdensity (rwstate attribute)": [[24, "blackjax.mcmc.random_walk.RWState.logdensity"]], "normal() (in module blackjax.mcmc.random_walk)": [[24, "blackjax.mcmc.random_walk.normal"]], "normal_random_walk() (additive_step_random_walk class method)": [[24, "blackjax.mcmc.random_walk.additive_step_random_walk.normal_random_walk"]], "position (rwstate attribute)": [[24, "blackjax.mcmc.random_walk.RWState.position"]], "proposal (rwinfo attribute)": [[24, "blackjax.mcmc.random_walk.RWInfo.proposal"]], "rmh (class in blackjax.mcmc.random_walk)": [[24, "blackjax.mcmc.random_walk.rmh"]], "rmh_proposal() (in module blackjax.mcmc.random_walk)": [[24, "blackjax.mcmc.random_walk.rmh_proposal"]], "iterativeuturnstate (class in blackjax.mcmc.termination)": [[25, "blackjax.mcmc.termination.IterativeUTurnState"]], "blackjax.mcmc.termination": [[25, "module-blackjax.mcmc.termination"]], "idx_max (iterativeuturnstate attribute)": [[25, "blackjax.mcmc.termination.IterativeUTurnState.idx_max"]], "idx_min (iterativeuturnstate attribute)": [[25, "blackjax.mcmc.termination.IterativeUTurnState.idx_min"]], "iterative_uturn_numpyro() (in module blackjax.mcmc.termination)": [[25, "blackjax.mcmc.termination.iterative_uturn_numpyro"]], "momentum (iterativeuturnstate attribute)": [[25, "blackjax.mcmc.termination.IterativeUTurnState.momentum"]], "momentum_sum (iterativeuturnstate attribute)": [[25, "blackjax.mcmc.termination.IterativeUTurnState.momentum_sum"]], "dynamicexpansionstate (class in blackjax.mcmc.trajectory)": [[26, "blackjax.mcmc.trajectory.DynamicExpansionState"]], "dynamicintegrationstate (class in blackjax.mcmc.trajectory)": [[26, "blackjax.mcmc.trajectory.DynamicIntegrationState"]], "trajectory (class in blackjax.mcmc.trajectory)": [[26, "blackjax.mcmc.trajectory.Trajectory"]], "append_to_trajectory() (in module blackjax.mcmc.trajectory)": [[26, "blackjax.mcmc.trajectory.append_to_trajectory"]], "blackjax.mcmc.trajectory": [[26, "module-blackjax.mcmc.trajectory"]], "dynamic_multiplicative_expansion() (in module blackjax.mcmc.trajectory)": [[26, "blackjax.mcmc.trajectory.dynamic_multiplicative_expansion"]], "dynamic_progressive_integration() (in module blackjax.mcmc.trajectory)": [[26, "blackjax.mcmc.trajectory.dynamic_progressive_integration"]], "dynamic_recursive_integration() (in module blackjax.mcmc.trajectory)": [[26, "blackjax.mcmc.trajectory.dynamic_recursive_integration"]], "hmc_energy() (in module blackjax.mcmc.trajectory)": [[26, "blackjax.mcmc.trajectory.hmc_energy"]], "leftmost_state (trajectory attribute)": [[26, "blackjax.mcmc.trajectory.Trajectory.leftmost_state"]], "merge_trajectories() (in module blackjax.mcmc.trajectory)": [[26, "blackjax.mcmc.trajectory.merge_trajectories"]], "momentum_sum (trajectory attribute)": [[26, "blackjax.mcmc.trajectory.Trajectory.momentum_sum"]], "num_states (trajectory attribute)": [[26, "blackjax.mcmc.trajectory.Trajectory.num_states"]], "proposal (dynamicexpansionstate attribute)": [[26, "blackjax.mcmc.trajectory.DynamicExpansionState.proposal"]], "proposal (dynamicintegrationstate attribute)": [[26, "blackjax.mcmc.trajectory.DynamicIntegrationState.proposal"]], "reorder_trajectories() (in module blackjax.mcmc.trajectory)": [[26, "blackjax.mcmc.trajectory.reorder_trajectories"]], "rightmost_state (trajectory attribute)": [[26, "blackjax.mcmc.trajectory.Trajectory.rightmost_state"]], "static_integration() (in module blackjax.mcmc.trajectory)": [[26, "blackjax.mcmc.trajectory.static_integration"]], "step (dynamicexpansionstate attribute)": [[26, "blackjax.mcmc.trajectory.DynamicExpansionState.step"]], "step (dynamicintegrationstate attribute)": [[26, "blackjax.mcmc.trajectory.DynamicIntegrationState.step"]], "termination_state (dynamicexpansionstate attribute)": [[26, "blackjax.mcmc.trajectory.DynamicExpansionState.termination_state"]], "termination_state (dynamicintegrationstate attribute)": [[26, "blackjax.mcmc.trajectory.DynamicIntegrationState.termination_state"]], "trajectory (dynamicexpansionstate attribute)": [[26, "blackjax.mcmc.trajectory.DynamicExpansionState.trajectory"]], "trajectory (dynamicintegrationstate attribute)": [[26, "blackjax.mcmc.trajectory.DynamicIntegrationState.trajectory"]], "dualaveragingstate (class in blackjax.optimizers.dual_averaging)": [[27, "blackjax.optimizers.dual_averaging.DualAveragingState"]], "avg_error (dualaveragingstate attribute)": [[27, "blackjax.optimizers.dual_averaging.DualAveragingState.avg_error"]], "blackjax.optimizers.dual_averaging": [[27, "module-blackjax.optimizers.dual_averaging"]], "dual_averaging() (in module blackjax.optimizers.dual_averaging)": [[27, "blackjax.optimizers.dual_averaging.dual_averaging"]], "log_x (dualaveragingstate attribute)": [[27, "blackjax.optimizers.dual_averaging.DualAveragingState.log_x"]], "log_x_avg (dualaveragingstate attribute)": [[27, "blackjax.optimizers.dual_averaging.DualAveragingState.log_x_avg"]], "mu (dualaveragingstate attribute)": [[27, "blackjax.optimizers.dual_averaging.DualAveragingState.mu"]], "step (dualaveragingstate attribute)": [[27, "blackjax.optimizers.dual_averaging.DualAveragingState.step"]], "blackjax.optimizers": [[28, "module-blackjax.optimizers"]], "lbfgshistory (class in blackjax.optimizers.lbfgs)": [[29, "blackjax.optimizers.lbfgs.LBFGSHistory"]], "alpha (lbfgshistory attribute)": [[29, "blackjax.optimizers.lbfgs.LBFGSHistory.alpha"]], "bfgs_sample() (in module blackjax.optimizers.lbfgs)": [[29, "blackjax.optimizers.lbfgs.bfgs_sample"]], "blackjax.optimizers.lbfgs": [[29, "module-blackjax.optimizers.lbfgs"]], "f (lbfgshistory attribute)": [[29, "blackjax.optimizers.lbfgs.LBFGSHistory.f"]], "g (lbfgshistory attribute)": [[29, "blackjax.optimizers.lbfgs.LBFGSHistory.g"]], "lbfgs_inverse_hessian_factors() (in module blackjax.optimizers.lbfgs)": [[29, "blackjax.optimizers.lbfgs.lbfgs_inverse_hessian_factors"]], "lbfgs_inverse_hessian_formula_1() (in module blackjax.optimizers.lbfgs)": [[29, "blackjax.optimizers.lbfgs.lbfgs_inverse_hessian_formula_1"]], "lbfgs_inverse_hessian_formula_2() (in module blackjax.optimizers.lbfgs)": [[29, "blackjax.optimizers.lbfgs.lbfgs_inverse_hessian_formula_2"]], "minimize_lbfgs() (in module blackjax.optimizers.lbfgs)": [[29, "blackjax.optimizers.lbfgs.minimize_lbfgs"]], "update_mask (lbfgshistory attribute)": [[29, "blackjax.optimizers.lbfgs.LBFGSHistory.update_mask"]], "x (lbfgshistory attribute)": [[29, "blackjax.optimizers.lbfgs.LBFGSHistory.x"]], "blackjax.progress_bar": [[30, "module-blackjax.progress_bar"]], "progress_bar_scan() (in module blackjax.progress_bar)": [[30, "blackjax.progress_bar.progress_bar_scan"]], "contoursgldstate (class in blackjax.sgmcmc.csgld)": [[31, "blackjax.sgmcmc.csgld.ContourSGLDState"]], "blackjax.sgmcmc.csgld": [[31, "module-blackjax.sgmcmc.csgld"]], "build_kernel (csgld attribute)": [[31, "blackjax.sgmcmc.csgld.csgld.build_kernel"]], "build_kernel() (in module blackjax.sgmcmc.csgld)": [[31, "blackjax.sgmcmc.csgld.build_kernel"]], "csgld (class in blackjax.sgmcmc.csgld)": [[31, "blackjax.sgmcmc.csgld.csgld"]], "energy_idx (contoursgldstate attribute)": [[31, "blackjax.sgmcmc.csgld.ContourSGLDState.energy_idx"]], "energy_pdf (contoursgldstate attribute)": [[31, "blackjax.sgmcmc.csgld.ContourSGLDState.energy_pdf"]], "init (csgld attribute)": [[31, "blackjax.sgmcmc.csgld.csgld.init"]], "init() (in module blackjax.sgmcmc.csgld)": [[31, "blackjax.sgmcmc.csgld.init"]], "position (contoursgldstate attribute)": [[31, "blackjax.sgmcmc.csgld.ContourSGLDState.position"]], "blackjax.sgmcmc.diffusions": [[32, "module-blackjax.sgmcmc.diffusions"]], "overdamped_langevin() (in module blackjax.sgmcmc.diffusions)": [[32, "blackjax.sgmcmc.diffusions.overdamped_langevin"]], "sghmc() (in module blackjax.sgmcmc.diffusions)": [[32, "blackjax.sgmcmc.diffusions.sghmc"]], "sgnht() (in module blackjax.sgmcmc.diffusions)": [[32, "blackjax.sgmcmc.diffusions.sgnht"]], "blackjax.sgmcmc.gradients": [[33, "module-blackjax.sgmcmc.gradients"]], "control_variates() (in module blackjax.sgmcmc.gradients)": [[33, "blackjax.sgmcmc.gradients.control_variates"]], "grad_estimator() (in module blackjax.sgmcmc.gradients)": [[33, "blackjax.sgmcmc.gradients.grad_estimator"]], "logdensity_estimator() (in module blackjax.sgmcmc.gradients)": [[33, "blackjax.sgmcmc.gradients.logdensity_estimator"]], "blackjax.sgmcmc": [[34, "module-blackjax.sgmcmc"]], "blackjax.sgmcmc.sghmc": [[35, "module-blackjax.sgmcmc.sghmc"]], "build_kernel (sghmc attribute)": [[35, "blackjax.sgmcmc.sghmc.sghmc.build_kernel"]], "build_kernel() (in module blackjax.sgmcmc.sghmc)": [[35, "blackjax.sgmcmc.sghmc.build_kernel"]], "init (sghmc attribute)": [[35, "blackjax.sgmcmc.sghmc.sghmc.init"]], "init() (in module blackjax.sgmcmc.sghmc)": [[35, "blackjax.sgmcmc.sghmc.init"]], "sghmc (class in blackjax.sgmcmc.sghmc)": [[35, "blackjax.sgmcmc.sghmc.sghmc"]], "blackjax.sgmcmc.sgld": [[36, "module-blackjax.sgmcmc.sgld"]], "build_kernel (sgld attribute)": [[36, "blackjax.sgmcmc.sgld.sgld.build_kernel"]], "build_kernel() (in module blackjax.sgmcmc.sgld)": [[36, "blackjax.sgmcmc.sgld.build_kernel"]], "init (sgld attribute)": [[36, "blackjax.sgmcmc.sgld.sgld.init"]], "init() (in module blackjax.sgmcmc.sgld)": [[36, "blackjax.sgmcmc.sgld.init"]], "sgld (class in blackjax.sgmcmc.sgld)": [[36, "blackjax.sgmcmc.sgld.sgld"]], "sgnhtstate (class in blackjax.sgmcmc.sgnht)": [[37, "blackjax.sgmcmc.sgnht.SGNHTState"]], "blackjax.sgmcmc.sgnht": [[37, "module-blackjax.sgmcmc.sgnht"]], "build_kernel (sgnht attribute)": [[37, "blackjax.sgmcmc.sgnht.sgnht.build_kernel"]], "build_kernel() (in module blackjax.sgmcmc.sgnht)": [[37, "blackjax.sgmcmc.sgnht.build_kernel"]], "init (sgnht attribute)": [[37, "blackjax.sgmcmc.sgnht.sgnht.init"]], "init() (in module blackjax.sgmcmc.sgnht)": [[37, "blackjax.sgmcmc.sgnht.init"]], "momentum (sgnhtstate attribute)": [[37, "blackjax.sgmcmc.sgnht.SGNHTState.momentum"]], "position (sgnhtstate attribute)": [[37, "blackjax.sgmcmc.sgnht.SGNHTState.position"]], "sgnht (class in blackjax.sgmcmc.sgnht)": [[37, "blackjax.sgmcmc.sgnht.sgnht"]], "xi (sgnhtstate attribute)": [[37, "blackjax.sgmcmc.sgnht.SGNHTState.xi"]], "adaptive_tempered_smc (class in blackjax.smc.adaptive_tempered)": [[38, "blackjax.smc.adaptive_tempered.adaptive_tempered_smc"]], "blackjax.smc.adaptive_tempered": [[38, "module-blackjax.smc.adaptive_tempered"]], "build_kernel (adaptive_tempered_smc attribute)": [[38, "blackjax.smc.adaptive_tempered.adaptive_tempered_smc.build_kernel"]], "build_kernel() (in module blackjax.smc.adaptive_tempered)": [[38, "blackjax.smc.adaptive_tempered.build_kernel"]], "init (adaptive_tempered_smc attribute)": [[38, "blackjax.smc.adaptive_tempered.adaptive_tempered_smc.init"]], "smcinfo (class in blackjax.smc.base)": [[39, "blackjax.smc.base.SMCInfo"]], "smcstate (class in blackjax.smc.base)": [[39, "blackjax.smc.base.SMCState"]], "ancestors (smcinfo attribute)": [[39, "blackjax.smc.base.SMCInfo.ancestors"]], "blackjax.smc.base": [[39, "module-blackjax.smc.base"]], "init() (in module blackjax.smc.base)": [[39, "blackjax.smc.base.init"]], "log_likelihood_increment (smcinfo attribute)": [[39, "blackjax.smc.base.SMCInfo.log_likelihood_increment"]], "particles (smcstate attribute)": [[39, "blackjax.smc.base.SMCState.particles"]], "step() (in module blackjax.smc.base)": [[39, "blackjax.smc.base.step"]], "update_info (smcinfo attribute)": [[39, "blackjax.smc.base.SMCInfo.update_info"]], "weights (smcstate attribute)": [[39, "blackjax.smc.base.SMCState.weights"]], "blackjax.smc.ess": [[40, "module-blackjax.smc.ess"]], "ess() (in module blackjax.smc.ess)": [[40, "blackjax.smc.ess.ess"]], "ess_solver() (in module blackjax.smc.ess)": [[40, "blackjax.smc.ess.ess_solver"]], "log_ess() (in module blackjax.smc.ess)": [[40, "blackjax.smc.ess.log_ess"]], "blackjax.smc": [[41, "module-blackjax.smc"]], "blackjax.smc.resampling": [[42, "module-blackjax.smc.resampling"]], "multinomial() (in module blackjax.smc.resampling)": [[42, "blackjax.smc.resampling.multinomial"]], "residual() (in module blackjax.smc.resampling)": [[42, "blackjax.smc.resampling.residual"]], "stratified() (in module blackjax.smc.resampling)": [[42, "blackjax.smc.resampling.stratified"]], "systematic() (in module blackjax.smc.resampling)": [[42, "blackjax.smc.resampling.systematic"]], "blackjax.smc.solver": [[43, "module-blackjax.smc.solver"]], "dichotomy() (in module blackjax.smc.solver)": [[43, "blackjax.smc.solver.dichotomy"]], "temperedsmcstate (class in blackjax.smc.tempered)": [[44, "blackjax.smc.tempered.TemperedSMCState"]], "blackjax.smc.tempered": [[44, "module-blackjax.smc.tempered"]], "build_kernel() (in module blackjax.smc.tempered)": [[44, "blackjax.smc.tempered.build_kernel"]], "init() (in module blackjax.smc.tempered)": [[44, "blackjax.smc.tempered.init"]], "lmbda (temperedsmcstate attribute)": [[44, "blackjax.smc.tempered.TemperedSMCState.lmbda"]], "particles (temperedsmcstate attribute)": [[44, "blackjax.smc.tempered.TemperedSMCState.particles"]], "weights (temperedsmcstate attribute)": [[44, "blackjax.smc.tempered.TemperedSMCState.weights"]], "array (in module blackjax.types)": [[45, "blackjax.types.Array"]], "arrayliketree (in module blackjax.types)": [[45, "blackjax.types.ArrayLikeTree"]], "arraytree (in module blackjax.types)": [[45, "blackjax.types.ArrayTree"]], "prngkey (in module blackjax.types)": [[45, "blackjax.types.PRNGKey"]], "blackjax.types": [[45, "module-blackjax.types"]], "blackjax.util": [[46, "module-blackjax.util"]], "generate_gaussian_noise() (in module blackjax.util)": [[46, "blackjax.util.generate_gaussian_noise"]], "generate_unit_vector() (in module blackjax.util)": [[46, "blackjax.util.generate_unit_vector"]], "index_pytree() (in module blackjax.util)": [[46, "blackjax.util.index_pytree"]], "linear_map() (in module blackjax.util)": [[46, "blackjax.util.linear_map"]], "pytree_size() (in module blackjax.util)": [[46, "blackjax.util.pytree_size"]], "blackjax.vi": [[47, "module-blackjax.vi"]], "mfviinfo (class in blackjax.vi.meanfield_vi)": [[48, "blackjax.vi.meanfield_vi.MFVIInfo"]], "mfvistate (class in blackjax.vi.meanfield_vi)": [[48, "blackjax.vi.meanfield_vi.MFVIState"]], "blackjax.vi.meanfield_vi": [[48, "module-blackjax.vi.meanfield_vi"]], "elbo (mfviinfo attribute)": [[48, "blackjax.vi.meanfield_vi.MFVIInfo.elbo"]], "generate_meanfield_logdensity() (in module blackjax.vi.meanfield_vi)": [[48, "blackjax.vi.meanfield_vi.generate_meanfield_logdensity"]], "init (meanfield_vi attribute)": [[48, "blackjax.vi.meanfield_vi.meanfield_vi.init"]], "meanfield_vi (class in blackjax.vi.meanfield_vi)": [[48, "blackjax.vi.meanfield_vi.meanfield_vi"]], "mu (mfvistate attribute)": [[48, "blackjax.vi.meanfield_vi.MFVIState.mu"]], "opt_state (mfvistate attribute)": [[48, "blackjax.vi.meanfield_vi.MFVIState.opt_state"]], "rho (mfvistate attribute)": [[48, "blackjax.vi.meanfield_vi.MFVIState.rho"]], "sample (meanfield_vi attribute)": [[48, "blackjax.vi.meanfield_vi.meanfield_vi.sample"]], "sample() (in module blackjax.vi.meanfield_vi)": [[48, "blackjax.vi.meanfield_vi.sample"]], "step (meanfield_vi attribute)": [[48, "blackjax.vi.meanfield_vi.meanfield_vi.step"]], "step() (in module blackjax.vi.meanfield_vi)": [[48, "blackjax.vi.meanfield_vi.step"]], "pathfinderstate (class in blackjax.vi.pathfinder)": [[49, "blackjax.vi.pathfinder.PathfinderState"]], "alpha (pathfinderstate attribute)": [[49, "blackjax.vi.pathfinder.PathfinderState.alpha"]], "approximate (pathfinder attribute)": [[49, "blackjax.vi.pathfinder.pathfinder.approximate"]], "approximate() (in module blackjax.vi.pathfinder)": [[49, "blackjax.vi.pathfinder.approximate"]], "beta (pathfinderstate attribute)": [[49, "blackjax.vi.pathfinder.PathfinderState.beta"]], "blackjax.vi.pathfinder": [[49, "module-blackjax.vi.pathfinder"]], "elbo (pathfinderstate attribute)": [[49, "blackjax.vi.pathfinder.PathfinderState.elbo"]], "gamma (pathfinderstate attribute)": [[49, "blackjax.vi.pathfinder.PathfinderState.gamma"]], "grad_position (pathfinderstate attribute)": [[49, "blackjax.vi.pathfinder.PathfinderState.grad_position"]], "pathfinder (class in blackjax.vi.pathfinder)": [[49, "blackjax.vi.pathfinder.pathfinder"]], "position (pathfinderstate attribute)": [[49, "blackjax.vi.pathfinder.PathfinderState.position"]], "sample (pathfinder attribute)": [[49, "blackjax.vi.pathfinder.pathfinder.sample"]], "sample() (in module blackjax.vi.pathfinder)": [[49, "blackjax.vi.pathfinder.sample"]], "blackjax.vi.svgd": [[50, "module-blackjax.vi.svgd"]], "build_kernel (svgd attribute)": [[50, "blackjax.vi.svgd.svgd.build_kernel"]], "init (svgd attribute)": [[50, "blackjax.vi.svgd.svgd.init"]], "rbf_kernel() (in module blackjax.vi.svgd)": [[50, "blackjax.vi.svgd.rbf_kernel"]], "svgd (class in blackjax.vi.svgd)": [[50, "blackjax.vi.svgd.svgd"]], "update_median_heuristic() (in module blackjax.vi.svgd)": [[50, "blackjax.vi.svgd.update_median_heuristic"]]}}) \ No newline at end of file