Deploying to gh-pages from @ a3e235f 🚀

_modules/blackjax/mcmc/mclmc.html

@@ -466,7 +466,9 @@ <h1>Source code for blackjax.mcmc.mclmc</h1><div class="highlight"><pre>
 
 <span class="sd">    &quot;&quot;&quot;</span>
 
-    <span class="n">step</span> <span class="o">=</span> <span class="n">with_isokinetic_maruyama</span><span class="p">(</span><span class="n">integrator</span><span class="p">(</span><span class="n">logdensity_fn</span><span class="p">,</span> <span class="n">sqrt_diag_cov</span><span class="p">))</span>
+    <span class="n">step</span> <span class="o">=</span> <span class="n">with_isokinetic_maruyama</span><span class="p">(</span>
+        <span class="n">integrator</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">sqrt_diag_cov</span><span class="o">=</span><span class="n">sqrt_diag_cov</span><span class="p">)</span>
+    <span class="p">)</span>
 
     <span class="k">def</span> <span class="nf">kernel</span><span class="p">(</span>
         <span class="n">rng_key</span><span class="p">:</span> <span class="n">PRNGKey</span><span class="p">,</span> <span class="n">state</span><span class="p">:</span> <span class="n">IntegratorState</span><span class="p">,</span> <span class="n">L</span><span class="p">:</span> <span class="nb">float</span><span class="p">,</span> <span class="n">step_size</span><span class="p">:</span> <span class="nb">float</span>

examples/howto_metropolis_within_gibbs.html

@@ -605,7 +605,7 @@ <h3>Sampling<a class="headerlink" href="#sampling" title="Link to this heading">
 </div>
 </div>
 <div class="cell_output docutils container">
-<div class="output stream highlight-myst-ansi notranslate"><div class="highlight"><pre><span></span>CPU times: user 926 ms, sys: 44.9 ms, total: 971 ms
+<div class="output stream highlight-myst-ansi notranslate"><div class="highlight"><pre><span></span>CPU times: user 932 ms, sys: 36.4 ms, total: 969 ms
 Wall time: 962 ms
 </pre></div>
 </div>
@@ -744,7 +744,7 @@ <h3>Sampling<a class="headerlink" href="#id2" title="Link to this heading">#</a>
 </div>
 </div>
 <div class="cell_output docutils container">
-<div class="output stream highlight-myst-ansi notranslate"><div class="highlight"><pre><span></span>CPU times: user 701 ms, sys: 25 ms, total: 726 ms
+<div class="output stream highlight-myst-ansi notranslate"><div class="highlight"><pre><span></span>CPU times: user 685 ms, sys: 45.6 ms, total: 730 ms
 Wall time: 721 ms
 </pre></div>
 </div>

examples/howto_sample_multiple_chains.html

@@ -525,8 +525,8 @@ <h3>Using <code class="docutils literal notranslate"><span class="pre">jax.vmap<
 </div>
 </div>
 <div class="cell_output docutils container">
-<div class="output stream highlight-myst-ansi notranslate"><div class="highlight"><pre><span></span>CPU times: user 14.5 s, sys: 262 ms, total: 14.8 s
-Wall time: 14.8 s
+<div class="output stream highlight-myst-ansi notranslate"><div class="highlight"><pre><span></span>CPU times: user 14.2 s, sys: 270 ms, total: 14.5 s
+Wall time: 14.5 s
 </pre></div>
 </div>
 </div>
@@ -630,8 +630,8 @@ <h3>Back to our example<a class="headerlink" href="#back-to-our-example" title="
 </div>
 </div>
 <div class="cell_output docutils container">
-<div class="output stream highlight-myst-ansi notranslate"><div class="highlight"><pre><span></span>CPU times: user 16.4 s, sys: 59.4 ms, total: 16.5 s
-Wall time: 5.32 s
+<div class="output stream highlight-myst-ansi notranslate"><div class="highlight"><pre><span></span>CPU times: user 16.5 s, sys: 51.2 ms, total: 16.6 s
+Wall time: 5.34 s
 </pre></div>
 </div>
 </div>

examples/howto_use_aesara.html

@@ -656,7 +656,7 @@ <h1>Use with Aesara models<a class="headerlink" href="#use-with-aesara-models" t
                |Elemwise{sigmoid,no_inplace} [id FD]
 </pre></div>
 </div>
-<div class="output text_plain highlight-myst-ansi notranslate"><div class="highlight"><pre><span></span>&lt;ipykernel.iostream.OutStream at 0x7f8d5632e4d0&gt;
+<div class="output text_plain highlight-myst-ansi notranslate"><div class="highlight"><pre><span></span>&lt;ipykernel.iostream.OutStream at 0x7efeb0ea6530&gt;
 </pre></div>
 </div>
 </div>

examples/howto_use_oryx.html

@@ -547,8 +547,8 @@ <h1>Use with Oryx models<a class="headerlink" href="#use-with-oryx-models" title
 </div>
 </div>
 <div class="cell_output docutils container">
-<div class="output stream highlight-myst-ansi notranslate"><div class="highlight"><pre><span></span>CPU times: user 17.3 s, sys: 523 ms, total: 17.8 s
-Wall time: 17.7 s
+<div class="output stream highlight-myst-ansi notranslate"><div class="highlight"><pre><span></span>CPU times: user 17.6 s, sys: 519 ms, total: 18.2 s
+Wall time: 18.1 s
 </pre></div>
 </div>
 </div>
@@ -585,8 +585,8 @@ <h1>Use with Oryx models<a class="headerlink" href="#use-with-oryx-models" title
 </div>
 </div>
 <div class="cell_output docutils container">
-<div class="output stream highlight-myst-ansi notranslate"><div class="highlight"><pre><span></span>CPU times: user 16.3 s, sys: 165 ms, total: 16.5 s
-Wall time: 16.4 s
+<div class="output stream highlight-myst-ansi notranslate"><div class="highlight"><pre><span></span>CPU times: user 16.5 s, sys: 110 ms, total: 16.6 s
+Wall time: 16.5 s
 </pre></div>
 </div>
 </div>

examples/quickstart.html

@@ -468,7 +468,7 @@ <h3>Set the Initial State<a class="headerlink" href="#set-the-initial-state" tit
 </div>
 </div>
 <div class="cell_output docutils container">
-<div class="output text_plain highlight-myst-ansi notranslate"><div class="highlight"><pre><span></span>HMCState(position={&#39;loc&#39;: 1.0, &#39;log_scale&#39;: 1.0}, logdensity=Array(-35110.457, dtype=float32), logdensity_grad={&#39;loc&#39;: Array(1309.9021, dtype=float32, weak_type=True), &#39;log_scale&#39;: Array(65383.04, dtype=float32, weak_type=True)})
+<div class="output text_plain highlight-myst-ansi notranslate"><div class="highlight"><pre><span></span>HMCState(position={&#39;loc&#39;: 1.0, &#39;log_scale&#39;: 1.0}, logdensity=Array(-34929.08, dtype=float32), logdensity_grad={&#39;loc&#39;: Array(1291.825, dtype=float32, weak_type=True), &#39;log_scale&#39;: Array(65020.29, dtype=float32, weak_type=True)})
 </pre></div>
 </div>
 </div>
@@ -516,8 +516,8 @@ <h3>Inference<a class="headerlink" href="#inference" title="Link to this heading
 </div>
 </div>
 <div class="cell_output docutils container">
-<div class="output stream highlight-myst-ansi notranslate"><div class="highlight"><pre><span></span>CPU times: user 1.4 s, sys: 44.1 ms, total: 1.44 s
-Wall time: 1.44 s
+<div class="output stream highlight-myst-ansi notranslate"><div class="highlight"><pre><span></span>CPU times: user 1.38 s, sys: 52.2 ms, total: 1.43 s
+Wall time: 1.42 s
 </pre></div>
 </div>
 </div>
@@ -542,7 +542,7 @@ <h3>Inference<a class="headerlink" href="#inference" title="Link to this heading
 </div>
 </details>
 <div class="cell_output docutils container">
-<img alt="../_images/44e48b729d6294610c5bfab0b581f054a9e85291c8b57b54537c3bc3f410916f.png" src="../_images/44e48b729d6294610c5bfab0b581f054a9e85291c8b57b54537c3bc3f410916f.png" />
+<img alt="../_images/36c9027770a6e68566e9af5f83f83a080643bf5a976e93f233f5eede3bc7dd2e.png" src="../_images/36c9027770a6e68566e9af5f83f83a080643bf5a976e93f233f5eede3bc7dd2e.png" />
 </div>
 </div>
 </section>
@@ -569,7 +569,7 @@ <h2>NUTS<a class="headerlink" href="#nuts" title="Link to this heading">#</a></h
 </div>
 </div>
 <div class="cell_output docutils container">
-<div class="output text_plain highlight-myst-ansi notranslate"><div class="highlight"><pre><span></span>HMCState(position={&#39;loc&#39;: 1.0, &#39;log_scale&#39;: 1.0}, logdensity=Array(-35110.457, dtype=float32), logdensity_grad={&#39;loc&#39;: Array(1309.9021, dtype=float32, weak_type=True), &#39;log_scale&#39;: Array(65383.04, dtype=float32, weak_type=True)})
+<div class="output text_plain highlight-myst-ansi notranslate"><div class="highlight"><pre><span></span>HMCState(position={&#39;loc&#39;: 1.0, &#39;log_scale&#39;: 1.0}, logdensity=Array(-34929.08, dtype=float32), logdensity_grad={&#39;loc&#39;: Array(1291.825, dtype=float32, weak_type=True), &#39;log_scale&#39;: Array(65020.29, dtype=float32, weak_type=True)})
 </pre></div>
 </div>
 </div>
@@ -586,8 +586,8 @@ <h2>NUTS<a class="headerlink" href="#nuts" title="Link to this heading">#</a></h
 </div>
 </div>
 <div class="cell_output docutils container">
-<div class="output stream highlight-myst-ansi notranslate"><div class="highlight"><pre><span></span>CPU times: user 6.34 s, sys: 51.9 ms, total: 6.39 s
-Wall time: 6.38 s
+<div class="output stream highlight-myst-ansi notranslate"><div class="highlight"><pre><span></span>CPU times: user 6.4 s, sys: 30.6 ms, total: 6.43 s
+Wall time: 6.42 s
 </pre></div>
 </div>
 </div>
@@ -612,7 +612,7 @@ <h2>NUTS<a class="headerlink" href="#nuts" title="Link to this heading">#</a></h
 </div>
 </details>
 <div class="cell_output docutils container">
-<img alt="../_images/e3a1bfb2b0af9205137c80c766adf31c4d5d1da0c1ffcdea4a67448cb9e6e60d.png" src="../_images/e3a1bfb2b0af9205137c80c766adf31c4d5d1da0c1ffcdea4a67448cb9e6e60d.png" />
+<img alt="../_images/0815b9b65e02bd54ff301137d205df774a26a7aa6bf8d917cfb89895eda9d77e.png" src="../_images/0815b9b65e02bd54ff301137d205df774a26a7aa6bf8d917cfb89895eda9d77e.png" />
 </div>
 </div>
 <section id="use-stan-s-window-adaptation">
@@ -630,8 +630,8 @@ <h3>Use Stan’s Window Adaptation<a class="headerlink" href="#use-stan-s-window
 </div>
 </div>
 <div class="cell_output docutils container">
-<div class="output stream highlight-myst-ansi notranslate"><div class="highlight"><pre><span></span>CPU times: user 1.53 s, sys: 68.6 ms, total: 1.6 s
-Wall time: 1.59 s
+<div class="output stream highlight-myst-ansi notranslate"><div class="highlight"><pre><span></span>CPU times: user 1.55 s, sys: 80.2 ms, total: 1.63 s
+Wall time: 1.61 s
 </pre></div>
 </div>
 </div>
@@ -650,8 +650,8 @@ <h3>Use Stan’s Window Adaptation<a class="headerlink" href="#use-stan-s-window
 </div>
 </div>
 <div class="cell_output docutils container">
-<div class="output stream highlight-myst-ansi notranslate"><div class="highlight"><pre><span></span>CPU times: user 1.05 s, sys: 32.4 ms, total: 1.08 s
-Wall time: 1.06 s
+<div class="output stream highlight-myst-ansi notranslate"><div class="highlight"><pre><span></span>CPU times: user 1.06 s, sys: 43 ms, total: 1.1 s
+Wall time: 1.08 s
 </pre></div>
 </div>
 </div>
@@ -676,7 +676,7 @@ <h3>Use Stan’s Window Adaptation<a class="headerlink" href="#use-stan-s-window
 </div>
 </details>
 <div class="cell_output docutils container">
-<img alt="../_images/0de581c9b2517e088d8ff1934232338d8ca1dd3bef6e8610be89d76991288940.png" src="../_images/0de581c9b2517e088d8ff1934232338d8ca1dd3bef6e8610be89d76991288940.png" />
+<img alt="../_images/542b50046cd39b59dabc8c4587803adfa0a1a898f448ee544358f0fffd97eeaa.png" src="../_images/542b50046cd39b59dabc8c4587803adfa0a1a898f448ee544358f0fffd97eeaa.png" />
 </div>
 </div>
 </section>