Deploying to gh-pages from @ af79fa4 🚀

_modules/blackjax/_version.html

@@ -392,8 +392,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>
 
 
-<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+g40efb6c&#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;g40efb6c&#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+gaf79fa4&#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;gaf79fa4&#39;</span><span class="p">)</span>
 </pre></div>
 
                 </article>

_static/pygments.css

@@ -3,77 +3,77 @@ html[data-theme="light"] .highlight td.linenos .normal { color: inherit; backgro
 html[data-theme="light"] .highlight span.linenos { color: inherit; background-color: transparent; padding-left: 5px; padding-right: 5px; }
 html[data-theme="light"] .highlight td.linenos .special { color: #000000; background-color: #ffffc0; padding-left: 5px; padding-right: 5px; }
 html[data-theme="light"] .highlight span.linenos.special { color: #000000; background-color: #ffffc0; padding-left: 5px; padding-right: 5px; }
-html[data-theme="light"] .highlight .hll { background-color: #7971292e }
-html[data-theme="light"] .highlight { background: #fefefe; color: #545454 }
-html[data-theme="light"] .highlight .c { color: #797129 } /* Comment */
-html[data-theme="light"] .highlight .err { color: #d91e18 } /* Error */
-html[data-theme="light"] .highlight .k { color: #7928a1 } /* Keyword */
-html[data-theme="light"] .highlight .l { color: #797129 } /* Literal */
-html[data-theme="light"] .highlight .n { color: #545454 } /* Name */
-html[data-theme="light"] .highlight .o { color: #008000 } /* Operator */
-html[data-theme="light"] .highlight .p { color: #545454 } /* Punctuation */
-html[data-theme="light"] .highlight .ch { color: #797129 } /* Comment.Hashbang */
-html[data-theme="light"] .highlight .cm { color: #797129 } /* Comment.Multiline */
-html[data-theme="light"] .highlight .cp { color: #797129 } /* Comment.Preproc */
-html[data-theme="light"] .highlight .cpf { color: #797129 } /* Comment.PreprocFile */
-html[data-theme="light"] .highlight .c1 { color: #797129 } /* Comment.Single */
-html[data-theme="light"] .highlight .cs { color: #797129 } /* Comment.Special */
-html[data-theme="light"] .highlight .gd { color: #007faa } /* Generic.Deleted */
+html[data-theme="light"] .highlight .hll { background-color: #fae4c2 }
+html[data-theme="light"] .highlight { background: #fefefe; color: #080808 }
+html[data-theme="light"] .highlight .c { color: #515151 } /* Comment */
+html[data-theme="light"] .highlight .err { color: #a12236 } /* Error */
+html[data-theme="light"] .highlight .k { color: #6730c5 } /* Keyword */
+html[data-theme="light"] .highlight .l { color: #7f4707 } /* Literal */
+html[data-theme="light"] .highlight .n { color: #080808 } /* Name */
+html[data-theme="light"] .highlight .o { color: #00622f } /* Operator */
+html[data-theme="light"] .highlight .p { color: #080808 } /* Punctuation */
+html[data-theme="light"] .highlight .ch { color: #515151 } /* Comment.Hashbang */
+html[data-theme="light"] .highlight .cm { color: #515151 } /* Comment.Multiline */
+html[data-theme="light"] .highlight .cp { color: #515151 } /* Comment.Preproc */
+html[data-theme="light"] .highlight .cpf { color: #515151 } /* Comment.PreprocFile */
+html[data-theme="light"] .highlight .c1 { color: #515151 } /* Comment.Single */
+html[data-theme="light"] .highlight .cs { color: #515151 } /* Comment.Special */
+html[data-theme="light"] .highlight .gd { color: #005b82 } /* Generic.Deleted */
 html[data-theme="light"] .highlight .ge { font-style: italic } /* Generic.Emph */
-html[data-theme="light"] .highlight .gh { color: #007faa } /* Generic.Heading */
+html[data-theme="light"] .highlight .gh { color: #005b82 } /* Generic.Heading */
 html[data-theme="light"] .highlight .gs { font-weight: bold } /* Generic.Strong */
-html[data-theme="light"] .highlight .gu { color: #007faa } /* Generic.Subheading */
-html[data-theme="light"] .highlight .kc { color: #7928a1 } /* Keyword.Constant */
-html[data-theme="light"] .highlight .kd { color: #7928a1 } /* Keyword.Declaration */
-html[data-theme="light"] .highlight .kn { color: #7928a1 } /* Keyword.Namespace */
-html[data-theme="light"] .highlight .kp { color: #7928a1 } /* Keyword.Pseudo */
-html[data-theme="light"] .highlight .kr { color: #7928a1 } /* Keyword.Reserved */
-html[data-theme="light"] .highlight .kt { color: #797129 } /* Keyword.Type */
-html[data-theme="light"] .highlight .ld { color: #797129 } /* Literal.Date */
-html[data-theme="light"] .highlight .m { color: #797129 } /* Literal.Number */
-html[data-theme="light"] .highlight .s { color: #008000 } /* Literal.String */
-html[data-theme="light"] .highlight .na { color: #797129 } /* Name.Attribute */
-html[data-theme="light"] .highlight .nb { color: #797129 } /* Name.Builtin */
-html[data-theme="light"] .highlight .nc { color: #007faa } /* Name.Class */
-html[data-theme="light"] .highlight .no { color: #007faa } /* Name.Constant */
-html[data-theme="light"] .highlight .nd { color: #797129 } /* Name.Decorator */
-html[data-theme="light"] .highlight .ni { color: #008000 } /* Name.Entity */
-html[data-theme="light"] .highlight .ne { color: #7928a1 } /* Name.Exception */
-html[data-theme="light"] .highlight .nf { color: #007faa } /* Name.Function */
-html[data-theme="light"] .highlight .nl { color: #797129 } /* Name.Label */
-html[data-theme="light"] .highlight .nn { color: #545454 } /* Name.Namespace */
-html[data-theme="light"] .highlight .nx { color: #545454 } /* Name.Other */
-html[data-theme="light"] .highlight .py { color: #007faa } /* Name.Property */
-html[data-theme="light"] .highlight .nt { color: #007faa } /* Name.Tag */
-html[data-theme="light"] .highlight .nv { color: #d91e18 } /* Name.Variable */
-html[data-theme="light"] .highlight .ow { color: #7928a1 } /* Operator.Word */
-html[data-theme="light"] .highlight .pm { color: #545454 } /* Punctuation.Marker */
-html[data-theme="light"] .highlight .w { color: #545454 } /* Text.Whitespace */
-html[data-theme="light"] .highlight .mb { color: #797129 } /* Literal.Number.Bin */
-html[data-theme="light"] .highlight .mf { color: #797129 } /* Literal.Number.Float */
-html[data-theme="light"] .highlight .mh { color: #797129 } /* Literal.Number.Hex */
-html[data-theme="light"] .highlight .mi { color: #797129 } /* Literal.Number.Integer */
-html[data-theme="light"] .highlight .mo { color: #797129 } /* Literal.Number.Oct */
-html[data-theme="light"] .highlight .sa { color: #008000 } /* Literal.String.Affix */
-html[data-theme="light"] .highlight .sb { color: #008000 } /* Literal.String.Backtick */
-html[data-theme="light"] .highlight .sc { color: #008000 } /* Literal.String.Char */
-html[data-theme="light"] .highlight .dl { color: #008000 } /* Literal.String.Delimiter */
-html[data-theme="light"] .highlight .sd { color: #008000 } /* Literal.String.Doc */
-html[data-theme="light"] .highlight .s2 { color: #008000 } /* Literal.String.Double */
-html[data-theme="light"] .highlight .se { color: #008000 } /* Literal.String.Escape */
-html[data-theme="light"] .highlight .sh { color: #008000 } /* Literal.String.Heredoc */
-html[data-theme="light"] .highlight .si { color: #008000 } /* Literal.String.Interpol */
-html[data-theme="light"] .highlight .sx { color: #008000 } /* Literal.String.Other */
-html[data-theme="light"] .highlight .sr { color: #d91e18 } /* Literal.String.Regex */
-html[data-theme="light"] .highlight .s1 { color: #008000 } /* Literal.String.Single */
-html[data-theme="light"] .highlight .ss { color: #007faa } /* Literal.String.Symbol */
-html[data-theme="light"] .highlight .bp { color: #797129 } /* Name.Builtin.Pseudo */
-html[data-theme="light"] .highlight .fm { color: #007faa } /* Name.Function.Magic */
-html[data-theme="light"] .highlight .vc { color: #d91e18 } /* Name.Variable.Class */
-html[data-theme="light"] .highlight .vg { color: #d91e18 } /* Name.Variable.Global */
-html[data-theme="light"] .highlight .vi { color: #d91e18 } /* Name.Variable.Instance */
-html[data-theme="light"] .highlight .vm { color: #797129 } /* Name.Variable.Magic */
-html[data-theme="light"] .highlight .il { color: #797129 } /* Literal.Number.Integer.Long */
+html[data-theme="light"] .highlight .gu { color: #005b82 } /* Generic.Subheading */
+html[data-theme="light"] .highlight .kc { color: #6730c5 } /* Keyword.Constant */
+html[data-theme="light"] .highlight .kd { color: #6730c5 } /* Keyword.Declaration */
+html[data-theme="light"] .highlight .kn { color: #6730c5 } /* Keyword.Namespace */
+html[data-theme="light"] .highlight .kp { color: #6730c5 } /* Keyword.Pseudo */
+html[data-theme="light"] .highlight .kr { color: #6730c5 } /* Keyword.Reserved */
+html[data-theme="light"] .highlight .kt { color: #7f4707 } /* Keyword.Type */
+html[data-theme="light"] .highlight .ld { color: #7f4707 } /* Literal.Date */
+html[data-theme="light"] .highlight .m { color: #7f4707 } /* Literal.Number */
+html[data-theme="light"] .highlight .s { color: #00622f } /* Literal.String */
+html[data-theme="light"] .highlight .na { color: #912583 } /* Name.Attribute */
+html[data-theme="light"] .highlight .nb { color: #7f4707 } /* Name.Builtin */
+html[data-theme="light"] .highlight .nc { color: #005b82 } /* Name.Class */
+html[data-theme="light"] .highlight .no { color: #005b82 } /* Name.Constant */
+html[data-theme="light"] .highlight .nd { color: #7f4707 } /* Name.Decorator */
+html[data-theme="light"] .highlight .ni { color: #00622f } /* Name.Entity */
+html[data-theme="light"] .highlight .ne { color: #6730c5 } /* Name.Exception */
+html[data-theme="light"] .highlight .nf { color: #005b82 } /* Name.Function */
+html[data-theme="light"] .highlight .nl { color: #7f4707 } /* Name.Label */
+html[data-theme="light"] .highlight .nn { color: #080808 } /* Name.Namespace */
+html[data-theme="light"] .highlight .nx { color: #080808 } /* Name.Other */
+html[data-theme="light"] .highlight .py { color: #005b82 } /* Name.Property */
+html[data-theme="light"] .highlight .nt { color: #005b82 } /* Name.Tag */
+html[data-theme="light"] .highlight .nv { color: #a12236 } /* Name.Variable */
+html[data-theme="light"] .highlight .ow { color: #6730c5 } /* Operator.Word */
+html[data-theme="light"] .highlight .pm { color: #080808 } /* Punctuation.Marker */
+html[data-theme="light"] .highlight .w { color: #080808 } /* Text.Whitespace */
+html[data-theme="light"] .highlight .mb { color: #7f4707 } /* Literal.Number.Bin */
+html[data-theme="light"] .highlight .mf { color: #7f4707 } /* Literal.Number.Float */
+html[data-theme="light"] .highlight .mh { color: #7f4707 } /* Literal.Number.Hex */
+html[data-theme="light"] .highlight .mi { color: #7f4707 } /* Literal.Number.Integer */
+html[data-theme="light"] .highlight .mo { color: #7f4707 } /* Literal.Number.Oct */
+html[data-theme="light"] .highlight .sa { color: #00622f } /* Literal.String.Affix */
+html[data-theme="light"] .highlight .sb { color: #00622f } /* Literal.String.Backtick */
+html[data-theme="light"] .highlight .sc { color: #00622f } /* Literal.String.Char */
+html[data-theme="light"] .highlight .dl { color: #00622f } /* Literal.String.Delimiter */
+html[data-theme="light"] .highlight .sd { color: #00622f } /* Literal.String.Doc */
+html[data-theme="light"] .highlight .s2 { color: #00622f } /* Literal.String.Double */
+html[data-theme="light"] .highlight .se { color: #00622f } /* Literal.String.Escape */
+html[data-theme="light"] .highlight .sh { color: #00622f } /* Literal.String.Heredoc */
+html[data-theme="light"] .highlight .si { color: #00622f } /* Literal.String.Interpol */
+html[data-theme="light"] .highlight .sx { color: #00622f } /* Literal.String.Other */
+html[data-theme="light"] .highlight .sr { color: #a12236 } /* Literal.String.Regex */
+html[data-theme="light"] .highlight .s1 { color: #00622f } /* Literal.String.Single */
+html[data-theme="light"] .highlight .ss { color: #005b82 } /* Literal.String.Symbol */
+html[data-theme="light"] .highlight .bp { color: #7f4707 } /* Name.Builtin.Pseudo */
+html[data-theme="light"] .highlight .fm { color: #005b82 } /* Name.Function.Magic */
+html[data-theme="light"] .highlight .vc { color: #a12236 } /* Name.Variable.Class */
+html[data-theme="light"] .highlight .vg { color: #a12236 } /* Name.Variable.Global */
+html[data-theme="light"] .highlight .vi { color: #a12236 } /* Name.Variable.Instance */
+html[data-theme="light"] .highlight .vm { color: #7f4707 } /* Name.Variable.Magic */
+html[data-theme="light"] .highlight .il { color: #7f4707 } /* Literal.Number.Integer.Long */
 html[data-theme="dark"] .highlight pre { line-height: 125%; }
 html[data-theme="dark"] .highlight td.linenos .normal { color: inherit; background-color: transparent; padding-left: 5px; padding-right: 5px; }
 html[data-theme="dark"] .highlight span.linenos { color: inherit; background-color: transparent; padding-left: 5px; padding-right: 5px; }

examples/howto_custom_gradients.html

@@ -599,7 +599,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(0.97067314, dtype=float32, weak_type=True), logdensity=Array(-2.512219, dtype=float32), logdensity_grad=Array(-1.5988109, dtype=float32, weak_type=True)))
+ HMCState(position=Array(0.860856, dtype=float32, weak_type=True), logdensity=Array(-2.3501818, dtype=float32), logdensity_grad=Array(-1.3549047, dtype=float32, weak_type=True)))
 </pre></div>
 </div>
 </div>

examples/howto_metropolis_within_gibbs.html

@@ -604,8 +604,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 895 ms, sys: 62.6 ms, total: 957 ms
-Wall time: 950 ms
+<div class="output stream highlight-myst-ansi notranslate"><div class="highlight"><pre><span></span>CPU times: user 944 ms, sys: 61 ms, total: 1.01 s
+Wall time: 998 ms
 </pre></div>
 </div>
 </div>
@@ -622,7 +622,7 @@ <h3>Sampling<a class="headerlink" href="#sampling" title="Link to this heading">
 </div>
 </div>
 <div class="cell_output docutils container">
-<img alt="../_images/aa8f91c89c3588232511e4a3bfb7e0f138ecc96be0bc378d707d58f3bca649b6.png" src="../_images/aa8f91c89c3588232511e4a3bfb7e0f138ecc96be0bc378d707d58f3bca649b6.png" />
+<img alt="../_images/6f0983eabaf46c6aadb38f46d263f08a5e081bdaf65bd4ca74800c382db1e359.png" src="../_images/6f0983eabaf46c6aadb38f46d263f08a5e081bdaf65bd4ca74800c382db1e359.png" />
 </div>
 </div>
 </section>
@@ -743,8 +743,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 702 ms, sys: 24 ms, total: 726 ms
-Wall time: 718 ms
+<div class="output stream highlight-myst-ansi notranslate"><div class="highlight"><pre><span></span>CPU times: user 693 ms, sys: 47.1 ms, total: 740 ms
+Wall time: 733 ms
 </pre></div>
 </div>
 </div>

examples/howto_sample_multiple_chains.html

@@ -524,8 +524,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.2 s, sys: 225 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.9 s, sys: 217 ms, total: 15.1 s
+Wall time: 15.1 s
 </pre></div>
 </div>
 </div>
@@ -629,8 +629,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 17.9 s, sys: 52.5 ms, total: 18 s
-Wall time: 5.58 s
+<div class="output stream highlight-myst-ansi notranslate"><div class="highlight"><pre><span></span>CPU times: user 16.6 s, sys: 63.7 ms, total: 16.7 s
+Wall time: 5.5 s
 </pre></div>
 </div>
 </div>

examples/howto_use_aesara.html

@@ -655,7 +655,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 0x7f7edcf605b0&gt;
+<div class="output text_plain highlight-myst-ansi notranslate"><div class="highlight"><pre><span></span>&lt;ipykernel.iostream.OutStream at 0x7fcc6202c5b0&gt;
 </pre></div>
 </div>
 </div>
@@ -757,7 +757,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/baffa78b233e62eaae0b452fedc34dd869b6a850dcf567c3bb47d0bcc867b97c.png" src="../_images/baffa78b233e62eaae0b452fedc34dd869b6a850dcf567c3bb47d0bcc867b97c.png" />
+<img alt="../_images/de8a0c8df0be84ceb399ce5608fc852da666cea4a3a3be46e6ef59a1ae4cd50c.png" src="../_images/de8a0c8df0be84ceb399ce5608fc852da666cea4a3a3be46e6ef59a1ae4cd50c.png" />
 </div>
 </div>
 </section>