Skip to content

Commit

Permalink
Deploying to gh-pages from @ 7a30818 🚀
Browse files Browse the repository at this point in the history
  • Loading branch information
@junpenglao
junpenglao committed Nov 29, 2023
1 parent 2b422b3 commit 232b389
Show file tree
Hide file tree
Showing 153 changed files with 1,770 additions and 1,307 deletions.
Binary file modified .doctrees/autoapi/blackjax/adaptation/chees_adaptation/index.doctree
View file
Binary file not shown.
Binary file modified .doctrees/autoapi/blackjax/optimizers/lbfgs/index.doctree
View file
Binary file not shown.
Binary file modified .doctrees/autoapi/blackjax/vi/pathfinder/index.doctree
View file
Binary file not shown.
Binary file modified .doctrees/environment.pickle
View file
Binary file not shown.
Binary file modified .doctrees/examples/howto_custom_gradients.doctree
View file
Binary file not shown.
Binary file modified .doctrees/examples/howto_metropolis_within_gibbs.doctree
View file
Binary file not shown.
Binary file modified .doctrees/examples/howto_other_frameworks.doctree
View file
Binary file not shown.
Binary file modified .doctrees/examples/howto_sample_multiple_chains.doctree
View file
Binary file not shown.
Binary file modified .doctrees/examples/howto_use_aesara.doctree
View file
Binary file not shown.
Binary file modified .doctrees/examples/howto_use_numpyro.doctree
View file
Binary file not shown.
Binary file modified .doctrees/examples/howto_use_oryx.doctree
View file
Binary file not shown.
Binary file modified .doctrees/examples/howto_use_pymc.doctree
View file
Binary file not shown.
Binary file modified .doctrees/examples/howto_use_tfp.doctree
View file
Binary file not shown.
Binary file modified .doctrees/examples/quickstart.doctree
View file
Binary file not shown.
Binary file removed _images/05ad9e2ee6955ea6c8c85cc58ff68cae4006e4c443645737ae98adf4d2573c4e.png
View file
Binary file not shown.
Binary file removed _images/0712ade51fae6d1f16e84d62d9c4bdf16b528f708f2ebb92f8021dc219074e72.png
View file
Binary file not shown.
Binary file removed _images/1f92fcef648ff9a68b4d6ecb0264cd452484f2f8342233ba62447e878db8b68e.png
View file
Binary file not shown.
Binary file added _images/288f01506d8acb126b0e5c979e57ab64e9b4329b5ce8d36fa8b28de4826a69bb.png
View file
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Binary file removed _images/296490cd5d519b1cd96971a2f54059c5cac326be0f2cc50b5c0f47b08294301e.png
View file
Binary file not shown.
Binary file added _images/31210180170a589ce373e4ff9f7c6daebaff8e47910ba9cf80eba6f178a12b55.png
View file
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Binary file added _images/45d0909134aed97a3c4ddb4b173bb7f76e4058ed35b861b2008dcad6f4d192d4.png
View file
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Binary file added _images/4c23d786aff2b394270108f7c904efcbf70fd098ed8a46827ebefae635782f7e.png
View file
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Binary file added _images/68046d740b504b1bb6c7194b55d290fbb64700c3e1b3613e153071b2f41bbf20.png
View file
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Binary file removed _images/76452134d079512c864d1806ee31807ab20d11948f9e96dc85aa1763e0b67eed.png
View file
Binary file not shown.
Binary file added _images/77f1f97a257424105cf362f3c7e61de0ee40b33c6721f9d6f978af74ccbe3092.png
View file
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Binary file removed _images/78e373daecfada121e1d9151b2175ed9460e91373336756890bd8f983268bf4b.png
View file
Binary file not shown.
Binary file removed _images/7e417ef15949cfd08dae29060e15abb64974ccfe709a3124e7360cc1d5192d24.png
View file
Binary file not shown.
Binary file removed _images/81e4c6546cb224894c6f42ed0c3d7396b17baff581cfdd1977dc7f21de54f925.png
View file
Binary file not shown.
Binary file added _images/a583028e4beb38c2946360f5a2f01eeb464e1dfd27cbb28148d0eee66a7c9220.png
View file
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Binary file removed _images/ab6da1f8ad04436da4bb616d1b82c1bcc894218d01f57ca6dce80f8a1c9ba15e.png
View file
Binary file not shown.
Binary file added _images/b03b256b23bd821c697f1e4f8a9dbcd5437bb2764f5dca1519c26ad65fee39e5.png
View file
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Binary file added _images/b8e8da212199431e71387dea0d01e237197089a1ae0989d35486a88ff69fb52e.png
View file
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
30 changes: 17 additions & 13 deletions _modules/blackjax/_version.html
View file
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@
<!DOCTYPE html>


<html lang="en" >
<html lang="en" data-content_root="" >

<head>
<meta charset="utf-8" />
Expand All @@ -18,12 +18,12 @@
</script>

<!-- Loaded before other Sphinx assets -->
<link href="../../_static/styles/theme.css?digest=365ca57ee442770a23c6" rel="stylesheet" />
<link href="../../_static/styles/bootstrap.css?digest=365ca57ee442770a23c6" rel="stylesheet" />
<link href="../../_static/styles/pydata-sphinx-theme.css?digest=365ca57ee442770a23c6" rel="stylesheet" />
<link href="../../_static/styles/theme.css?digest=5b4479735964841361fd" rel="stylesheet" />
<link href="../../_static/styles/bootstrap.css?digest=5b4479735964841361fd" rel="stylesheet" />
<link href="../../_static/styles/pydata-sphinx-theme.css?digest=5b4479735964841361fd" rel="stylesheet" />


<link href="../../_static/vendor/fontawesome/6.1.2/css/all.min.css?digest=365ca57ee442770a23c6" rel="stylesheet" />
<link href="../../_static/vendor/fontawesome/6.1.2/css/all.min.css?digest=5b4479735964841361fd" rel="stylesheet" />
<link rel="preload" as="font" type="font/woff2" crossorigin href="../../_static/vendor/fontawesome/6.1.2/webfonts/fa-solid-900.woff2" />
<link rel="preload" as="font" type="font/woff2" crossorigin href="../../_static/vendor/fontawesome/6.1.2/webfonts/fa-brands-400.woff2" />
<link rel="preload" as="font" type="font/woff2" crossorigin href="../../_static/vendor/fontawesome/6.1.2/webfonts/fa-regular-400.woff2" />
Expand All @@ -36,9 +36,9 @@
<link rel="stylesheet" type="text/css" href="../../_static/design-style.1e8bd061cd6da7fc9cf755528e8ffc24.min.css" />

<!-- Pre-loaded scripts that we'll load fully later -->
<link rel="preload" as="script" href="../../_static/scripts/bootstrap.js?digest=365ca57ee442770a23c6" />
<link rel="preload" as="script" href="../../_static/scripts/pydata-sphinx-theme.js?digest=365ca57ee442770a23c6" />
<script src="../../_static/vendor/fontawesome/6.1.2/js/all.min.js?digest=365ca57ee442770a23c6"></script>
<link rel="preload" as="script" href="../../_static/scripts/bootstrap.js?digest=5b4479735964841361fd" />
<link rel="preload" as="script" href="../../_static/scripts/pydata-sphinx-theme.js?digest=5b4479735964841361fd" />
<script src="../../_static/vendor/fontawesome/6.1.2/js/all.min.js?digest=5b4479735964841361fd"></script>

<script data-url_root="../../" id="documentation_options" src="../../_static/documentation_options.js"></script>
<script src="../../_static/doctools.js"></script>
Expand Down Expand Up @@ -120,6 +120,7 @@

<div class="sidebar-primary-items__start sidebar-primary__section">
<div class="sidebar-primary-item">



<a class="navbar-brand logo" href="../../index.html">
Expand Down Expand Up @@ -293,6 +294,7 @@
</button>



<script>
document.write(`
<button class="btn btn-sm navbar-btn theme-switch-button" title="light/dark" aria-label="light/dark" data-bs-placement="bottom" data-bs-toggle="tooltip">
Expand All @@ -303,6 +305,7 @@
`);
</script>


<script>
document.write(`
<button class="btn btn-sm navbar-btn search-button search-button__button" title="Search" aria-label="Search" data-bs-placement="bottom" data-bs-toggle="tooltip">
Expand Down Expand Up @@ -350,8 +353,8 @@ <h1>Source code for blackjax._version</h1><div class="highlight"><pre>
<span class="n">__version_tuple__</span><span class="p">:</span> <span class="n">VERSION_TUPLE</span>
<div class="viewcode-block" id="version_tuple"><a class="viewcode-back" href="../../autoapi/blackjax/_version/index.html#blackjax._version.version_tuple">[docs]</a><span class="n">version_tuple</span><span class="p">:</span> <span class="n">VERSION_TUPLE</span></div>

<span class="n">__version__</span> <span class="o">=</span> <span class="n">version</span> <span class="o">=</span> <span class="s1">&#39;0.1.dev1+ge0f107f&#39;</span>
<span class="n">__version_tuple__</span> <span class="o">=</span> <span class="n">version_tuple</span> <span class="o">=</span> <span class="p">(</span><span class="mi">0</span><span class="p">,</span> <span class="mi">1</span><span class="p">,</span> <span class="s1">&#39;dev1&#39;</span><span class="p">,</span> <span class="s1">&#39;ge0f107f&#39;</span><span class="p">)</span>
<span class="n">__version__</span> <span class="o">=</span> <span class="n">version</span> <span class="o">=</span> <span class="s1">&#39;0.1.dev1+g7a30818&#39;</span>
<span class="n">__version_tuple__</span> <span class="o">=</span> <span class="n">version_tuple</span> <span class="o">=</span> <span class="p">(</span><span class="mi">0</span><span class="p">,</span> <span class="mi">1</span><span class="p">,</span> <span class="s1">&#39;dev1&#39;</span><span class="p">,</span> <span class="s1">&#39;g7a30818&#39;</span><span class="p">)</span>
</pre></div>

</article>
Expand All @@ -362,7 +365,7 @@ <h1>Source code for blackjax._version</h1><div class="highlight"><pre>


<footer class="prev-next-footer">
<!-- Previous / next buttons -->

<div class="prev-next-area">
</div>
</footer>
Expand All @@ -387,6 +390,7 @@ <h1>Source code for blackjax._version</h1><div class="highlight"><pre>

<div class="footer-item">


<p class="copyright">

© Copyright 2023, The Blackjax developers.
Expand All @@ -413,8 +417,8 @@ <h1>Source code for blackjax._version</h1><div class="highlight"><pre>
</div>

<!-- Scripts loaded after <body> so the DOM is not blocked -->
<script src="../../_static/scripts/bootstrap.js?digest=365ca57ee442770a23c6"></script>
<script src="../../_static/scripts/pydata-sphinx-theme.js?digest=365ca57ee442770a23c6"></script>
<script src="../../_static/scripts/bootstrap.js?digest=5b4479735964841361fd"></script>
<script src="../../_static/scripts/pydata-sphinx-theme.js?digest=5b4479735964841361fd"></script>

<footer class="bd-footer">
</footer>
Expand Down
26 changes: 15 additions & 11 deletions _modules/blackjax/adaptation/base.html
View file
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@
<!DOCTYPE html>


<html lang="en" >
<html lang="en" data-content_root="" >

<head>
<meta charset="utf-8" />
Expand All @@ -18,12 +18,12 @@
</script>

<!-- Loaded before other Sphinx assets -->
<link href="../../../_static/styles/theme.css?digest=365ca57ee442770a23c6" rel="stylesheet" />
<link href="../../../_static/styles/bootstrap.css?digest=365ca57ee442770a23c6" rel="stylesheet" />
<link href="../../../_static/styles/pydata-sphinx-theme.css?digest=365ca57ee442770a23c6" rel="stylesheet" />
<link href="../../../_static/styles/theme.css?digest=5b4479735964841361fd" rel="stylesheet" />
<link href="../../../_static/styles/bootstrap.css?digest=5b4479735964841361fd" rel="stylesheet" />
<link href="../../../_static/styles/pydata-sphinx-theme.css?digest=5b4479735964841361fd" rel="stylesheet" />


<link href="../../../_static/vendor/fontawesome/6.1.2/css/all.min.css?digest=365ca57ee442770a23c6" rel="stylesheet" />
<link href="../../../_static/vendor/fontawesome/6.1.2/css/all.min.css?digest=5b4479735964841361fd" rel="stylesheet" />
<link rel="preload" as="font" type="font/woff2" crossorigin href="../../../_static/vendor/fontawesome/6.1.2/webfonts/fa-solid-900.woff2" />
<link rel="preload" as="font" type="font/woff2" crossorigin href="../../../_static/vendor/fontawesome/6.1.2/webfonts/fa-brands-400.woff2" />
<link rel="preload" as="font" type="font/woff2" crossorigin href="../../../_static/vendor/fontawesome/6.1.2/webfonts/fa-regular-400.woff2" />
Expand All @@ -36,9 +36,9 @@
<link rel="stylesheet" type="text/css" href="../../../_static/design-style.1e8bd061cd6da7fc9cf755528e8ffc24.min.css" />

<!-- Pre-loaded scripts that we'll load fully later -->
<link rel="preload" as="script" href="../../../_static/scripts/bootstrap.js?digest=365ca57ee442770a23c6" />
<link rel="preload" as="script" href="../../../_static/scripts/pydata-sphinx-theme.js?digest=365ca57ee442770a23c6" />
<script src="../../../_static/vendor/fontawesome/6.1.2/js/all.min.js?digest=365ca57ee442770a23c6"></script>
<link rel="preload" as="script" href="../../../_static/scripts/bootstrap.js?digest=5b4479735964841361fd" />
<link rel="preload" as="script" href="../../../_static/scripts/pydata-sphinx-theme.js?digest=5b4479735964841361fd" />
<script src="../../../_static/vendor/fontawesome/6.1.2/js/all.min.js?digest=5b4479735964841361fd"></script>

<script data-url_root="../../../" id="documentation_options" src="../../../_static/documentation_options.js"></script>
<script src="../../../_static/doctools.js"></script>
Expand Down Expand Up @@ -120,6 +120,7 @@

<div class="sidebar-primary-items__start sidebar-primary__section">
<div class="sidebar-primary-item">



<a class="navbar-brand logo" href="../../../index.html">
Expand Down Expand Up @@ -293,6 +294,7 @@
</button>



<script>
document.write(`
<button class="btn btn-sm navbar-btn theme-switch-button" title="light/dark" aria-label="light/dark" data-bs-placement="bottom" data-bs-toggle="tooltip">
Expand All @@ -303,6 +305,7 @@
`);
</script>


<script>
document.write(`
<button class="btn btn-sm navbar-btn search-button search-button__button" title="Search" aria-label="Search" data-bs-placement="bottom" data-bs-toggle="tooltip">
Expand Down Expand Up @@ -373,7 +376,7 @@ <h1>Source code for blackjax.adaptation.base</h1><div class="highlight"><pre>


<footer class="prev-next-footer">
<!-- Previous / next buttons -->

<div class="prev-next-area">
</div>
</footer>
Expand All @@ -398,6 +401,7 @@ <h1>Source code for blackjax.adaptation.base</h1><div class="highlight"><pre>

<div class="footer-item">


<p class="copyright">

© Copyright 2023, The Blackjax developers.
Expand All @@ -424,8 +428,8 @@ <h1>Source code for blackjax.adaptation.base</h1><div class="highlight"><pre>
</div>

<!-- Scripts loaded after <body> so the DOM is not blocked -->
<script src="../../../_static/scripts/bootstrap.js?digest=365ca57ee442770a23c6"></script>
<script src="../../../_static/scripts/pydata-sphinx-theme.js?digest=365ca57ee442770a23c6"></script>
<script src="../../../_static/scripts/bootstrap.js?digest=5b4479735964841361fd"></script>
<script src="../../../_static/scripts/pydata-sphinx-theme.js?digest=5b4479735964841361fd"></script>

<footer class="bd-footer">
</footer>
Expand Down
38 changes: 21 additions & 17 deletions _modules/blackjax/adaptation/chees_adaptation.html
View file
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@
<!DOCTYPE html>


<html lang="en" >
<html lang="en" data-content_root="" >

<head>
<meta charset="utf-8" />
Expand All @@ -18,12 +18,12 @@
</script>

<!-- Loaded before other Sphinx assets -->
<link href="../../../_static/styles/theme.css?digest=365ca57ee442770a23c6" rel="stylesheet" />
<link href="../../../_static/styles/bootstrap.css?digest=365ca57ee442770a23c6" rel="stylesheet" />
<link href="../../../_static/styles/pydata-sphinx-theme.css?digest=365ca57ee442770a23c6" rel="stylesheet" />
<link href="../../../_static/styles/theme.css?digest=5b4479735964841361fd" rel="stylesheet" />
<link href="../../../_static/styles/bootstrap.css?digest=5b4479735964841361fd" rel="stylesheet" />
<link href="../../../_static/styles/pydata-sphinx-theme.css?digest=5b4479735964841361fd" rel="stylesheet" />


<link href="../../../_static/vendor/fontawesome/6.1.2/css/all.min.css?digest=365ca57ee442770a23c6" rel="stylesheet" />
<link href="../../../_static/vendor/fontawesome/6.1.2/css/all.min.css?digest=5b4479735964841361fd" rel="stylesheet" />
<link rel="preload" as="font" type="font/woff2" crossorigin href="../../../_static/vendor/fontawesome/6.1.2/webfonts/fa-solid-900.woff2" />
<link rel="preload" as="font" type="font/woff2" crossorigin href="../../../_static/vendor/fontawesome/6.1.2/webfonts/fa-brands-400.woff2" />
<link rel="preload" as="font" type="font/woff2" crossorigin href="../../../_static/vendor/fontawesome/6.1.2/webfonts/fa-regular-400.woff2" />
Expand All @@ -36,9 +36,9 @@
<link rel="stylesheet" type="text/css" href="../../../_static/design-style.1e8bd061cd6da7fc9cf755528e8ffc24.min.css" />

<!-- Pre-loaded scripts that we'll load fully later -->
<link rel="preload" as="script" href="../../../_static/scripts/bootstrap.js?digest=365ca57ee442770a23c6" />
<link rel="preload" as="script" href="../../../_static/scripts/pydata-sphinx-theme.js?digest=365ca57ee442770a23c6" />
<script src="../../../_static/vendor/fontawesome/6.1.2/js/all.min.js?digest=365ca57ee442770a23c6"></script>
<link rel="preload" as="script" href="../../../_static/scripts/bootstrap.js?digest=5b4479735964841361fd" />
<link rel="preload" as="script" href="../../../_static/scripts/pydata-sphinx-theme.js?digest=5b4479735964841361fd" />
<script src="../../../_static/vendor/fontawesome/6.1.2/js/all.min.js?digest=5b4479735964841361fd"></script>

<script data-url_root="../../../" id="documentation_options" src="../../../_static/documentation_options.js"></script>
<script src="../../../_static/doctools.js"></script>
Expand Down Expand Up @@ -120,6 +120,7 @@

<div class="sidebar-primary-items__start sidebar-primary__section">
<div class="sidebar-primary-item">



<a class="navbar-brand logo" href="../../../index.html">
Expand Down Expand Up @@ -293,6 +294,7 @@
</button>



<script>
document.write(`
<button class="btn btn-sm navbar-btn theme-switch-button" title="light/dark" aria-label="light/dark" data-bs-placement="bottom" data-bs-toggle="tooltip">
Expand All @@ -303,6 +305,7 @@
`);
</script>


<script>
document.write(`
<button class="btn btn-sm navbar-btn search-button search-button__button" title="Search" aria-label="Search" data-bs-placement="bottom" data-bs-toggle="tooltip">
Expand Down Expand Up @@ -611,7 +614,7 @@ <h1>Source code for blackjax.adaptation.chees_adaptation</h1><div class="highlig


<div class="viewcode-block" id="chees_adaptation"><a class="viewcode-back" href="../../../autoapi/blackjax/adaptation/chees_adaptation/index.html#blackjax.adaptation.chees_adaptation.chees_adaptation">[docs]</a><span class="k">def</span> <span class="nf">chees_adaptation</span><span class="p">(</span>
<span class="n">logprob_fn</span><span class="p">:</span> <span class="n">Callable</span><span class="p">,</span>
<span class="n">logdensity_fn</span><span class="p">:</span> <span class="n">Callable</span><span class="p">,</span>
<span class="n">num_chains</span><span class="p">:</span> <span class="nb">int</span><span class="p">,</span>
<span class="o">*</span><span class="p">,</span>
<span class="n">jitter_generator</span><span class="p">:</span> <span class="n">Optional</span><span class="p">[</span><span class="n">Callable</span><span class="p">]</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
Expand Down Expand Up @@ -646,7 +649,7 @@ <h1>Source code for blackjax.adaptation.chees_adaptation</h1><div class="highlig

<span class="sd"> .. code::</span>

<span class="sd"> warmup = blackjax.chees_adaptation(logprob_fn, num_chains)</span>
<span class="sd"> warmup = blackjax.chees_adaptation(logdensity_fn, num_chains)</span>
<span class="sd"> key_warmup, key_sample = jax.random.split(rng_key)</span>
<span class="sd"> optim = optax.adam(learning_rate)</span>
<span class="sd"> (last_states, parameters), _ = warmup.run(</span>
Expand All @@ -656,12 +659,12 @@ <h1>Source code for blackjax.adaptation.chees_adaptation</h1><div class="highlig
<span class="sd"> optim,</span>
<span class="sd"> num_warmup_steps,</span>
<span class="sd"> )</span>
<span class="sd"> kernel = blackjax.dynamic_hmc(logprob_fn, **parameters).step</span>
<span class="sd"> kernel = blackjax.dynamic_hmc(logdensity_fn, **parameters).step</span>
<span class="sd"> new_states, info = jax.vmap(kernel)(key_sample, last_states)</span>

<span class="sd"> Parameters</span>
<span class="sd"> ----------</span>
<span class="sd"> logprob_fn</span>
<span class="sd"> logdensity_fn</span>
<span class="sd"> The log density probability density function from which we wish to sample.</span>
<span class="sd"> num_chains</span>
<span class="sd"> Number of chains used for cross-chain warm-up training.</span>
Expand Down Expand Up @@ -737,7 +740,7 @@ <h1>Source code for blackjax.adaptation.chees_adaptation</h1><div class="highlig
<span class="n">keys</span> <span class="o">=</span> <span class="n">jax</span><span class="o">.</span><span class="n">random</span><span class="o">.</span><span class="n">split</span><span class="p">(</span><span class="n">rng_key</span><span class="p">,</span> <span class="n">num_chains</span><span class="p">)</span>
<span class="n">_step_fn</span> <span class="o">=</span> <span class="n">partial</span><span class="p">(</span>
<span class="n">step_fn</span><span class="p">,</span>
<span class="n">logdensity_fn</span><span class="o">=</span><span class="n">logprob_fn</span><span class="p">,</span>
<span class="n">logdensity_fn</span><span class="o">=</span><span class="n">logdensity_fn</span><span class="p">,</span>
<span class="n">step_size</span><span class="o">=</span><span class="n">adaptation_state</span><span class="o">.</span><span class="n">step_size</span><span class="p">,</span>
<span class="n">inverse_mass_matrix</span><span class="o">=</span><span class="n">jnp</span><span class="o">.</span><span class="n">ones</span><span class="p">(</span><span class="n">num_dim</span><span class="p">),</span>
<span class="n">trajectory_length_adjusted</span><span class="o">=</span><span class="n">adaptation_state</span><span class="o">.</span><span class="n">trajectory_length</span>
Expand All @@ -760,7 +763,7 @@ <h1>Source code for blackjax.adaptation.chees_adaptation</h1><div class="highlig
<span class="p">)</span>

<span class="n">batch_init</span> <span class="o">=</span> <span class="n">jax</span><span class="o">.</span><span class="n">vmap</span><span class="p">(</span>
<span class="k">lambda</span> <span class="n">p</span><span class="p">:</span> <span class="n">hmc</span><span class="o">.</span><span class="n">init_dynamic</span><span class="p">(</span><span class="n">p</span><span class="p">,</span> <span class="n">logprob_fn</span><span class="p">,</span> <span class="n">init_random_arg</span><span class="p">)</span>
<span class="k">lambda</span> <span class="n">p</span><span class="p">:</span> <span class="n">hmc</span><span class="o">.</span><span class="n">init_dynamic</span><span class="p">(</span><span class="n">p</span><span class="p">,</span> <span class="n">logdensity_fn</span><span class="p">,</span> <span class="n">init_random_arg</span><span class="p">)</span>
<span class="p">)</span>
<span class="n">init_states</span> <span class="o">=</span> <span class="n">batch_init</span><span class="p">(</span><span class="n">positions</span><span class="p">)</span>
<span class="n">init_adaptation_state</span> <span class="o">=</span> <span class="n">init</span><span class="p">(</span><span class="n">init_random_arg</span><span class="p">,</span> <span class="n">step_size</span><span class="p">)</span>
Expand Down Expand Up @@ -801,7 +804,7 @@ <h1>Source code for blackjax.adaptation.chees_adaptation</h1><div class="highlig


<footer class="prev-next-footer">
<!-- Previous / next buttons -->

<div class="prev-next-area">
</div>
</footer>
Expand All @@ -826,6 +829,7 @@ <h1>Source code for blackjax.adaptation.chees_adaptation</h1><div class="highlig

<div class="footer-item">


<p class="copyright">

© Copyright 2023, The Blackjax developers.
Expand All @@ -852,8 +856,8 @@ <h1>Source code for blackjax.adaptation.chees_adaptation</h1><div class="highlig
</div>

<!-- Scripts loaded after <body> so the DOM is not blocked -->
<script src="../../../_static/scripts/bootstrap.js?digest=365ca57ee442770a23c6"></script>
<script src="../../../_static/scripts/pydata-sphinx-theme.js?digest=365ca57ee442770a23c6"></script>
<script src="../../../_static/scripts/bootstrap.js?digest=5b4479735964841361fd"></script>
<script src="../../../_static/scripts/pydata-sphinx-theme.js?digest=5b4479735964841361fd"></script>

<footer class="bd-footer">
</footer>
Expand Down
Loading

0 comments on commit 232b389

Please sign in to comment.