Deploying to gh-pages from @ 5764a2b 🚀

_sources/index.md

@@ -57,13 +57,7 @@ If you want to use Blackjax with a model implemented with a PPL, go to the relat
 ```{code-block} bash
 pip install blackjax
 ```
-:::
 
-:::{tab-item} Nightly
-```{code-block} bash
-pip install blackjax-nightly
-```
-:::
 
 :::{tab-item} Conda
 ```{code-block} bash

examples/howto_custom_gradients.html

@@ -600,7 +600,7 @@ <h3>Using the function with Blackjax<a class="headerlink" href="#using-the-funct
 </div>
 <div class="cell_output docutils container">
 <div class="output text_plain highlight-myst-ansi notranslate"><div class="highlight"><pre><span></span>(HMCState(position=1.0, logdensity=Array(-2.5600991, dtype=float32), logdensity_grad=Array(-1.6666666, dtype=float32, weak_type=True)),
- HMCState(position=Array(1.3687104, dtype=float32, weak_type=True), logdensity=Array(-3.3443608, dtype=float32), logdensity_grad=Array(-2.6176217, dtype=float32, weak_type=True)))
+ HMCState(position=Array(0.46280023, dtype=float32, weak_type=True), logdensity=Array(-1.9669967, dtype=float32), logdensity_grad=Array(-0.60558957, dtype=float32, weak_type=True)))
 </pre></div>
 </div>
 </div>

examples/howto_metropolis_within_gibbs.html

@@ -605,8 +605,8 @@ <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 929 ms, sys: 48.1 ms, total: 977 ms
-Wall time: 966 ms
+<div class="output stream highlight-myst-ansi notranslate"><div class="highlight"><pre><span></span>CPU times: user 916 ms, sys: 48.7 ms, total: 964 ms
+Wall time: 959 ms
 </pre></div>
 </div>
 </div>
@@ -623,7 +623,7 @@ <h3>Sampling<a class="headerlink" href="#sampling" title="Link to this heading">
 </div>
 </div>
 <div class="cell_output docutils container">
-<img alt="../_images/caf16ac55678b25884f4e2c131cc728a3ded1ea0a1fb3d34a79db37206a8542e.png" src="../_images/caf16ac55678b25884f4e2c131cc728a3ded1ea0a1fb3d34a79db37206a8542e.png" />
+<img alt="../_images/4218380d23f2ec06f6777ab06b43a4fd904cb1ae7a3024820639f938deadf660.png" src="../_images/4218380d23f2ec06f6777ab06b43a4fd904cb1ae7a3024820639f938deadf660.png" />
 </div>
 </div>
 </section>
@@ -744,8 +744,8 @@ <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 704 ms, sys: 28.8 ms, total: 733 ms
-Wall time: 727 ms
+<div class="output stream highlight-myst-ansi notranslate"><div class="highlight"><pre><span></span>CPU times: user 693 ms, sys: 30.5 ms, total: 724 ms
+Wall time: 712 ms
 </pre></div>
 </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 15.1 s, sys: 324 ms, total: 15.4 s
-Wall time: 15.4 s
+<div class="output stream highlight-myst-ansi notranslate"><div class="highlight"><pre><span></span>CPU times: user 14.7 s, sys: 254 ms, total: 15 s
+Wall time: 15 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.9 s, sys: 103 ms, total: 17 s
-Wall time: 5.46 s
+<div class="output stream highlight-myst-ansi notranslate"><div class="highlight"><pre><span></span>CPU times: user 18 s, sys: 124 ms, total: 18.1 s
+Wall time: 5.75 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 0x7f6b42cfe710&gt;
+<div class="output text_plain highlight-myst-ansi notranslate"><div class="highlight"><pre><span></span>&lt;ipykernel.iostream.OutStream at 0x7f3532452710&gt;
 </pre></div>
 </div>
 </div>
@@ -758,7 +758,7 @@ <h1>Use with Aesara models<a class="headerlink" href="#use-with-aesara-models" t
 </div>
 </div>
 <div class="cell_output docutils container">
-<img alt="../_images/f484a3bd8834489a1b61a1153ba3f7a955d438d3f22372bc034bc6b3708ad45b.png" src="../_images/f484a3bd8834489a1b61a1153ba3f7a955d438d3f22372bc034bc6b3708ad45b.png" />
+<img alt="../_images/7f795b5df8c2c4e60b016eb8516dd1445de3cbf42b5ac1b8ae986609b5015c72.png" src="../_images/7f795b5df8c2c4e60b016eb8516dd1445de3cbf42b5ac1b8ae986609b5015c72.png" />
 </div>
 </div>
 </section>

examples/howto_use_numpyro.html

@@ -527,7 +527,7 @@ <h1>Use with Numpyro models<a class="headerlink" href="#use-with-numpyro-models"
 </div>
 </div>
 <div class="cell_output docutils container">
-<div class="output stream highlight-myst-ansi notranslate"><div class="highlight"><pre><span></span>\Average acceptance rate: 0.90
+<div class="output stream highlight-myst-ansi notranslate"><div class="highlight"><pre><span></span>\Average acceptance rate: 0.92
 There were 0.00% divergent transitions
 </pre></div>
 </div>
@@ -552,7 +552,7 @@ <h1>Use with Numpyro models<a class="headerlink" href="#use-with-numpyro-models"
 </div>
 </div>
 <div class="cell_output docutils container">
-<img alt="../_images/d24aae7587d092c5b7c9eb175fa7be533743d5e5b8e01b61569e69c16399b991.png" src="../_images/d24aae7587d092c5b7c9eb175fa7be533743d5e5b8e01b61569e69c16399b991.png" />
+<img alt="../_images/ef1a3b862675229de274a7d2e6667b0b0b05c7b16484f91d938130efd74ed4e2.png" src="../_images/ef1a3b862675229de274a7d2e6667b0b0b05c7b16484f91d938130efd74ed4e2.png" />
 </div>
 </div>
 <div class="cell docutils container">
@@ -563,7 +563,7 @@ <h1>Use with Numpyro models<a class="headerlink" href="#use-with-numpyro-models"
 </div>
 </div>
 <div class="cell_output docutils container">
-<img alt="../_images/ebe8aea40840ef6e653bf78b9e53dbc571f84b9aeac52421382800ed090afdf4.png" src="../_images/ebe8aea40840ef6e653bf78b9e53dbc571f84b9aeac52421382800ed090afdf4.png" />
+<img alt="../_images/8e58a69780a759a5bed0801222d3bb94f494005e5c91db2ec27dbe86d94b9a37.png" src="../_images/8e58a69780a759a5bed0801222d3bb94f494005e5c91db2ec27dbe86d94b9a37.png" />
 </div>
 </div>
 <div class="cell tag_hide-input docutils container">
@@ -582,14 +582,14 @@ <h1>Use with Numpyro models<a class="headerlink" href="#use-with-numpyro-models"
 </div>
 </details>
 <div class="cell_output docutils container">
-<div class="output stream highlight-myst-ansi notranslate"><div class="highlight"><pre><span></span>Relative treatment effect for school 0: -0.11
-Relative treatment effect for school 1: -0.60
-Relative treatment effect for school 2: 0.55
-Relative treatment effect for school 3: 0.05
-Relative treatment effect for school 4: 0.04
-Relative treatment effect for school 5: -0.27
-Relative treatment effect for school 6: 0.32
-Relative treatment effect for school 7: 0.46
+<div class="output stream highlight-myst-ansi notranslate"><div class="highlight"><pre><span></span>Relative treatment effect for school 0: -0.40
+Relative treatment effect for school 1: 0.11
+Relative treatment effect for school 2: 0.41
+Relative treatment effect for school 3: -0.09
+Relative treatment effect for school 4: -0.31
+Relative treatment effect for school 5: 0.49
+Relative treatment effect for school 6: -0.06
+Relative treatment effect for school 7: 0.73
 </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 18 s, sys: 369 ms, total: 18.4 s
-Wall time: 18.3 s
+<div class="output stream highlight-myst-ansi notranslate"><div class="highlight"><pre><span></span>CPU times: user 18.5 s, sys: 403 ms, total: 18.9 s
+Wall time: 18.8 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.6 s, sys: 163 ms, total: 16.7 s
-Wall time: 16.6 s
+<div class="output stream highlight-myst-ansi notranslate"><div class="highlight"><pre><span></span>CPU times: user 16.8 s, sys: 236 ms, total: 17 s
+Wall time: 16.9 s
 </pre></div>
 </div>
 </div>
@@ -627,7 +627,7 @@ <h1>Use with Oryx models<a class="headerlink" href="#use-with-oryx-models" title
 </div>
 </details>
 <div class="cell_output docutils container">
-<div class="output stream highlight-myst-ansi notranslate"><div class="highlight"><pre><span></span>Average sample accuracy: 0.9792
+<div class="output stream highlight-myst-ansi notranslate"><div class="highlight"><pre><span></span>Average sample accuracy: 0.9795333
 BMA accuracy: 0.9866667
 </pre></div>
 </div>

examples/howto_use_pymc.html

@@ -512,7 +512,7 @@ <h1>Use with PyMC models<a class="headerlink" href="#use-with-pymc-models" title
 </div>
 </details>
 <div class="cell_output docutils container">
-<img alt="../_images/467333a0ba65666e388513a8cac9fd822fbca5592a7c69dfa8b9ea7239b7fb21.png" src="../_images/467333a0ba65666e388513a8cac9fd822fbca5592a7c69dfa8b9ea7239b7fb21.png" />
+<img alt="../_images/74172ed8800acc80df28ab360885898b8e2d1ece13dc86fff35f736d56e9ff1f.png" src="../_images/74172ed8800acc80df28ab360885898b8e2d1ece13dc86fff35f736d56e9ff1f.png" />
 </div>
 </div>
 </section>

examples/howto_use_tfp.html

@@ -509,7 +509,7 @@ <h1>Use with TFP models<a class="headerlink" href="#use-with-tfp-models" title="
 </div>
 </details>
 <div class="cell_output docutils container">
-<div class="output stream highlight-myst-ansi notranslate"><div class="highlight"><pre><span></span>Average acceptance rate: 0.90
+<div class="output stream highlight-myst-ansi notranslate"><div class="highlight"><pre><span></span>Average acceptance rate: 0.82
 </pre></div>
 </div>
 </div>
@@ -545,7 +545,7 @@ <h1>Use with TFP models<a class="headerlink" href="#use-with-tfp-models" title="
 </div>
 </details>
 <div class="cell_output docutils container">
-<img alt="../_images/22ef7ca4336ca6eae639ac6cedbe0ddfc95b16829096046960813aee8ee5caff.png" src="../_images/22ef7ca4336ca6eae639ac6cedbe0ddfc95b16829096046960813aee8ee5caff.png" />
+<img alt="../_images/44bd94bd1fb3f2dbaab41db2ed403074a91a70f7f60878d70f3dcf1d9d8be3cc.png" src="../_images/44bd94bd1fb3f2dbaab41db2ed403074a91a70f7f60878d70f3dcf1d9d8be3cc.png" />
 </div>
 </div>
 </section>

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(-35374.445, dtype=float32), logdensity_grad={&#39;loc&#39;: Array(1338.1416, dtype=float32, weak_type=True), &#39;log_scale&#39;: Array(65911.02, 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(-33807.004, dtype=float32), logdensity_grad={&#39;loc&#39;: Array(1187.644, dtype=float32, weak_type=True), &#39;log_scale&#39;: Array(62776.14, dtype=float32, weak_type=True)})
 </pre></div>
 </div>
 </div>
@@ -516,7 +516,7 @@ <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.39 s, sys: 40.8 ms, total: 1.43 s
+<div class="output stream highlight-myst-ansi notranslate"><div class="highlight"><pre><span></span>CPU times: user 1.4 s, sys: 36.4 ms, total: 1.43 s
 Wall time: 1.43 s
 </pre></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/05dd1394dc3be1f888557ae027c4fc82fa7c3bd5156bc89786b5f0ca85734626.png" src="../_images/05dd1394dc3be1f888557ae027c4fc82fa7c3bd5156bc89786b5f0ca85734626.png" />
+<img alt="../_images/a429c8eab2bade4b66e3e64025d569286bba8ec735d72036c8187679ff98ae76.png" src="../_images/a429c8eab2bade4b66e3e64025d569286bba8ec735d72036c8187679ff98ae76.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(-35374.445, dtype=float32), logdensity_grad={&#39;loc&#39;: Array(1338.1416, dtype=float32, weak_type=True), &#39;log_scale&#39;: Array(65911.02, 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(-33807.004, dtype=float32), logdensity_grad={&#39;loc&#39;: Array(1187.644, dtype=float32, weak_type=True), &#39;log_scale&#39;: Array(62776.14, 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.42 s, sys: 54.5 ms, total: 6.47 s
-Wall time: 6.46 s
+<div class="output stream highlight-myst-ansi notranslate"><div class="highlight"><pre><span></span>CPU times: user 6.38 s, sys: 52.6 ms, total: 6.43 s
+Wall time: 6.41 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/9a5e5639e27b98e1de29366a79413384923baeaad00754eeeb587866256a7fe3.png" src="../_images/9a5e5639e27b98e1de29366a79413384923baeaad00754eeeb587866256a7fe3.png" />
+<img alt="../_images/8532f58688e583c9b5ea554a6c9144a8dfff918fc3e2fae48a2a25747906a89e.png" src="../_images/8532f58688e583c9b5ea554a6c9144a8dfff918fc3e2fae48a2a25747906a89e.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.43 s, sys: 80.2 ms, total: 1.51 s
-Wall time: 1.49 s
+<div class="output stream highlight-myst-ansi notranslate"><div class="highlight"><pre><span></span>CPU times: user 1.57 s, sys: 52.1 ms, total: 1.62 s
+Wall time: 1.6 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: 28.2 ms, total: 1.08 s
-Wall time: 1.07 s
+<div class="output stream highlight-myst-ansi notranslate"><div class="highlight"><pre><span></span>CPU times: user 1.12 s, sys: 40.8 ms, total: 1.16 s
+Wall time: 1.14 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/eb70d58243cc038d81071ae5a4635ca174e4c7b84f3af5aacf2b75b7ce206ae2.png" src="../_images/eb70d58243cc038d81071ae5a4635ca174e4c7b84f3af5aacf2b75b7ce206ae2.png" />
+<img alt="../_images/ed85f53da5f2b09740c68bcc002d4fb4798e3abbab4df5d0a1e06ad80845dc80.png" src="../_images/ed85f53da5f2b09740c68bcc002d4fb4798e3abbab4df5d0a1e06ad80845dc80.png" />
 </div>
 </div>
 </section>

index.html

@@ -442,19 +442,13 @@ <h1>Installation<a class="headerlink" href="#installation" title="Link to this h
 <div class="highlight-bash notranslate"><div class="highlight"><pre><span></span>pip<span class="w"> </span>install<span class="w"> </span>blackjax
 </pre></div>
 </div>
-</div>
-<input id="sd-tab-item-1" name="sd-tab-set-0" type="radio">
-<label class="sd-tab-label" for="sd-tab-item-1">
-Nightly</label><div class="sd-tab-content docutils">
-<div class="highlight-bash notranslate"><div class="highlight"><pre><span></span>pip<span class="w"> </span>install<span class="w"> </span>blackjax-nightly
+<div class="sd-tab-item docutils">
+<p class="sd-tab-label rubric">Conda</p>
+<div class="sd-tab-content docutils">
+<div class="highlight-bash notranslate"><div class="highlight"><pre><span></span>conda<span class="w"> </span>install<span class="w"> </span>blackjax<span class="w"> </span>-c<span class="w"> </span>conda-forge
 </pre></div>
 </div>
 </div>
-<input id="sd-tab-item-2" name="sd-tab-set-0" type="radio">
-<label class="sd-tab-label" for="sd-tab-item-2">
-Conda</label><div class="sd-tab-content docutils">
-<div class="highlight-bash notranslate"><div class="highlight"><pre><span></span>conda<span class="w"> </span>install<span class="w"> </span>blackjax<span class="w"> </span>-c<span class="w"> </span>conda-forge
-</pre></div>
 </div>
 </div>
 </div>