Deploying to gh-pages from @ f49945d 🚀

blackjax-devs · Dec 4, 2023 · ae352df · ae352df
1 parent 25ebfeb
commit ae352df
Show file tree

Hide file tree

Showing 157 changed files with 1,752 additions and 438 deletions.
diff --git a/.doctrees/autoapi/blackjax/index.doctree b/.doctrees/autoapi/blackjax/index.doctree
diff --git a/.doctrees/autoapi/blackjax/mcmc/proposal/index.doctree b/.doctrees/autoapi/blackjax/mcmc/proposal/index.doctree
diff --git a/.doctrees/autoapi/blackjax/mcmc/random_walk/index.doctree b/.doctrees/autoapi/blackjax/mcmc/random_walk/index.doctree
diff --git a/.doctrees/autoapi/blackjax/vi/index.doctree b/.doctrees/autoapi/blackjax/vi/index.doctree
diff --git a/.doctrees/autoapi/blackjax/vi/schrodinger_follmer/index.doctree b/.doctrees/autoapi/blackjax/vi/schrodinger_follmer/index.doctree
diff --git a/.doctrees/bib.doctree b/.doctrees/bib.doctree
diff --git a/.doctrees/environment.pickle b/.doctrees/environment.pickle
diff --git a/.doctrees/examples/howto_custom_gradients.doctree b/.doctrees/examples/howto_custom_gradients.doctree
diff --git a/.doctrees/examples/howto_metropolis_within_gibbs.doctree b/.doctrees/examples/howto_metropolis_within_gibbs.doctree
diff --git a/.doctrees/examples/howto_other_frameworks.doctree b/.doctrees/examples/howto_other_frameworks.doctree
diff --git a/.doctrees/examples/howto_sample_multiple_chains.doctree b/.doctrees/examples/howto_sample_multiple_chains.doctree
diff --git a/.doctrees/examples/howto_use_aesara.doctree b/.doctrees/examples/howto_use_aesara.doctree
diff --git a/.doctrees/examples/howto_use_numpyro.doctree b/.doctrees/examples/howto_use_numpyro.doctree
diff --git a/.doctrees/examples/howto_use_oryx.doctree b/.doctrees/examples/howto_use_oryx.doctree
diff --git a/.doctrees/examples/howto_use_pymc.doctree b/.doctrees/examples/howto_use_pymc.doctree
diff --git a/.doctrees/examples/howto_use_tfp.doctree b/.doctrees/examples/howto_use_tfp.doctree
diff --git a/.doctrees/examples/quickstart.doctree b/.doctrees/examples/quickstart.doctree
diff --git a/_images/0da7ddf258dd2d06b952107ec6b76f7d229072aeddc2a3c10f1302c379bc4b9e.png b/_images/0da7ddf258dd2d06b952107ec6b76f7d229072aeddc2a3c10f1302c379bc4b9e.png
diff --git a/_images/0e163fa72ed3585f27aee253fdf8b45e219673b2d35eea69153eaf6e67cc14ca.png b/_images/0e163fa72ed3585f27aee253fdf8b45e219673b2d35eea69153eaf6e67cc14ca.png
diff --git a/_images/240c913f234e0ddc85763b3e50641d3207f8d80b7e19ad3ec27884bc2e5bafc8.png b/_images/240c913f234e0ddc85763b3e50641d3207f8d80b7e19ad3ec27884bc2e5bafc8.png
diff --git a/_images/3014d5efa5a6cf3cd2b6c2fff53b1477c35e2a01fbeda8712260daffa655b98c.png b/_images/3014d5efa5a6cf3cd2b6c2fff53b1477c35e2a01fbeda8712260daffa655b98c.png
diff --git a/_images/3e72fab607ae32465b53a94b2d7e3368672458f3e5218b86bb1f3d49af202d47.png b/_images/3e72fab607ae32465b53a94b2d7e3368672458f3e5218b86bb1f3d49af202d47.png
diff --git a/_images/599ea6c97ef624224789e51115ee406d16b62dc046b35390b23c7b7771d440fe.png b/_images/599ea6c97ef624224789e51115ee406d16b62dc046b35390b23c7b7771d440fe.png
diff --git a/_images/5a1a5b550756ff42435172808ae6aa9f0098372e045e149ab55a46c26d800913.png b/_images/5a1a5b550756ff42435172808ae6aa9f0098372e045e149ab55a46c26d800913.png
diff --git a/_images/892493e54ea93083da7fadb55cdf206096ed9c338d63dd0186667c03e80e1c76.png b/_images/892493e54ea93083da7fadb55cdf206096ed9c338d63dd0186667c03e80e1c76.png
diff --git a/_images/92955b7ff28d0d81a7bb9e42c1ab4667e21577dfc431bb9e12f7870ae14ad6df.png b/_images/92955b7ff28d0d81a7bb9e42c1ab4667e21577dfc431bb9e12f7870ae14ad6df.png
diff --git a/_images/982cff51779cc6dccfda8063ecd863b99752c83f9791095e1b46133c392846aa.png b/_images/982cff51779cc6dccfda8063ecd863b99752c83f9791095e1b46133c392846aa.png
diff --git a/_images/9f53c036839d47dae5ff85037ac77d4d072613b7fbfe2dca537f03c55a6df801.png b/_images/9f53c036839d47dae5ff85037ac77d4d072613b7fbfe2dca537f03c55a6df801.png
diff --git a/_images/a3f0a4e000a56d759a0a1bbb6acbf81998fda0771e0bcefc6c5fe44e98b5d15f.png b/_images/a3f0a4e000a56d759a0a1bbb6acbf81998fda0771e0bcefc6c5fe44e98b5d15f.png
diff --git a/_images/aed5e2a7c85d8ec3b35d3bb9a336129791893e7fa992ee38b5487c26acfb1056.png b/_images/aed5e2a7c85d8ec3b35d3bb9a336129791893e7fa992ee38b5487c26acfb1056.png
diff --git a/_images/afd8ed0bc07621443f2b5b12c766c892a62564da9434b6a34af72eac6edb008a.png b/_images/afd8ed0bc07621443f2b5b12c766c892a62564da9434b6a34af72eac6edb008a.png
diff --git a/_images/d9933aaebd94f7125a9f3a32390a11547b6e003aea6637a7ecde8c5f4f967780.png b/_images/d9933aaebd94f7125a9f3a32390a11547b6e003aea6637a7ecde8c5f4f967780.png
diff --git a/_images/e54e23c5009eb078b6793f10a67c32fd228e57b305884f07b41c6666dea28262.png b/_images/e54e23c5009eb078b6793f10a67c32fd228e57b305884f07b41c6666dea28262.png
diff --git a/_images/e99798b0eb6a6b72c0b2e1ac3b6f811c1b95f375a3c2b3bd81c84e3ad3cfc18f.png b/_images/e99798b0eb6a6b72c0b2e1ac3b6f811c1b95f375a3c2b3bd81c84e3ad3cfc18f.png
diff --git a/_images/eb6a32e3c16c075acd2e494b07f69133e8fbe31104337eaec960fe5028cc3ce6.png b/_images/eb6a32e3c16c075acd2e494b07f69133e8fbe31104337eaec960fe5028cc3ce6.png
diff --git a/_modules/blackjax/_version.html b/_modules/blackjax/_version.html
@@ -215,6 +215,7 @@
 <li class="toctree-l2 has-children"><a class="reference internal" href="../../autoapi/blackjax/vi/index.html"><code class="xref py py-mod docutils literal notranslate"><span class="pre">blackjax.vi</span></code></a><input class="toctree-checkbox" id="toctree-checkbox-6" name="toctree-checkbox-6" type="checkbox"/><label class="toctree-toggle" for="toctree-checkbox-6"><i class="fa-solid fa-chevron-down"></i></label><ul>
 <li class="toctree-l3"><a class="reference internal" href="../../autoapi/blackjax/vi/meanfield_vi/index.html"><code class="xref py py-mod docutils literal notranslate"><span class="pre">blackjax.vi.meanfield_vi</span></code></a></li>
 <li class="toctree-l3"><a class="reference internal" href="../../autoapi/blackjax/vi/pathfinder/index.html"><code class="xref py py-mod docutils literal notranslate"><span class="pre">blackjax.vi.pathfinder</span></code></a></li>
+<li class="toctree-l3"><a class="reference internal" href="../../autoapi/blackjax/vi/schrodinger_follmer/index.html"><code class="xref py py-mod docutils literal notranslate"><span class="pre">blackjax.vi.schrodinger_follmer</span></code></a></li>
 <li class="toctree-l3"><a class="reference internal" href="../../autoapi/blackjax/vi/svgd/index.html"><code class="xref py py-mod docutils literal notranslate"><span class="pre">blackjax.vi.svgd</span></code></a></li>
 </ul>
 </li>
@@ -354,8 +355,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+g3845635&#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;g3845635&#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+gf49945d&#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;gf49945d&#39;</span><span class="p">)</span>
 </pre></div>
 
                 </article>

diff --git a/_modules/blackjax/adaptation/base.html b/_modules/blackjax/adaptation/base.html
@@ -215,6 +215,7 @@
 <li class="toctree-l2 has-children"><a class="reference internal" href="../../../autoapi/blackjax/vi/index.html"><code class="xref py py-mod docutils literal notranslate"><span class="pre">blackjax.vi</span></code></a><input class="toctree-checkbox" id="toctree-checkbox-6" name="toctree-checkbox-6" type="checkbox"/><label class="toctree-toggle" for="toctree-checkbox-6"><i class="fa-solid fa-chevron-down"></i></label><ul>
 <li class="toctree-l3"><a class="reference internal" href="../../../autoapi/blackjax/vi/meanfield_vi/index.html"><code class="xref py py-mod docutils literal notranslate"><span class="pre">blackjax.vi.meanfield_vi</span></code></a></li>
 <li class="toctree-l3"><a class="reference internal" href="../../../autoapi/blackjax/vi/pathfinder/index.html"><code class="xref py py-mod docutils literal notranslate"><span class="pre">blackjax.vi.pathfinder</span></code></a></li>
+<li class="toctree-l3"><a class="reference internal" href="../../../autoapi/blackjax/vi/schrodinger_follmer/index.html"><code class="xref py py-mod docutils literal notranslate"><span class="pre">blackjax.vi.schrodinger_follmer</span></code></a></li>
 <li class="toctree-l3"><a class="reference internal" href="../../../autoapi/blackjax/vi/svgd/index.html"><code class="xref py py-mod docutils literal notranslate"><span class="pre">blackjax.vi.svgd</span></code></a></li>
 </ul>
 </li>

diff --git a/_modules/blackjax/adaptation/chees_adaptation.html b/_modules/blackjax/adaptation/chees_adaptation.html
@@ -215,6 +215,7 @@
 <li class="toctree-l2 has-children"><a class="reference internal" href="../../../autoapi/blackjax/vi/index.html"><code class="xref py py-mod docutils literal notranslate"><span class="pre">blackjax.vi</span></code></a><input class="toctree-checkbox" id="toctree-checkbox-6" name="toctree-checkbox-6" type="checkbox"/><label class="toctree-toggle" for="toctree-checkbox-6"><i class="fa-solid fa-chevron-down"></i></label><ul>
 <li class="toctree-l3"><a class="reference internal" href="../../../autoapi/blackjax/vi/meanfield_vi/index.html"><code class="xref py py-mod docutils literal notranslate"><span class="pre">blackjax.vi.meanfield_vi</span></code></a></li>
 <li class="toctree-l3"><a class="reference internal" href="../../../autoapi/blackjax/vi/pathfinder/index.html"><code class="xref py py-mod docutils literal notranslate"><span class="pre">blackjax.vi.pathfinder</span></code></a></li>
+<li class="toctree-l3"><a class="reference internal" href="../../../autoapi/blackjax/vi/schrodinger_follmer/index.html"><code class="xref py py-mod docutils literal notranslate"><span class="pre">blackjax.vi.schrodinger_follmer</span></code></a></li>
 <li class="toctree-l3"><a class="reference internal" href="../../../autoapi/blackjax/vi/svgd/index.html"><code class="xref py py-mod docutils literal notranslate"><span class="pre">blackjax.vi.svgd</span></code></a></li>
 </ul>
 </li>
@@ -748,8 +749,8 @@ <h1>Source code for blackjax.adaptation.chees_adaptation</h1><div class="highlig
             <span class="n">new_states</span><span class="p">,</span> <span class="n">info</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="n">_step_fn</span><span class="p">)(</span><span class="n">keys</span><span class="p">,</span> <span class="n">states</span><span class="p">)</span>
             <span class="n">new_adaptation_state</span> <span class="o">=</span> <span class="n">update</span><span class="p">(</span>
                 <span class="n">adaptation_state</span><span class="p">,</span>
-                <span class="n">info</span><span class="o">.</span><span class="n">proposal</span><span class="o">.</span><span class="n">state</span><span class="o">.</span><span class="n">position</span><span class="p">,</span>
-                <span class="n">info</span><span class="o">.</span><span class="n">proposal</span><span class="o">.</span><span class="n">state</span><span class="o">.</span><span class="n">momentum</span><span class="p">,</span>
+                <span class="n">info</span><span class="o">.</span><span class="n">proposal</span><span class="o">.</span><span class="n">position</span><span class="p">,</span>
+                <span class="n">info</span><span class="o">.</span><span class="n">proposal</span><span class="o">.</span><span class="n">momentum</span><span class="p">,</span>
                 <span class="n">states</span><span class="o">.</span><span class="n">position</span><span class="p">,</span>
                 <span class="n">info</span><span class="o">.</span><span class="n">acceptance_rate</span><span class="p">,</span>
                 <span class="n">info</span><span class="o">.</span><span class="n">is_divergent</span><span class="p">,</span>

diff --git a/_modules/blackjax/adaptation/mass_matrix.html b/_modules/blackjax/adaptation/mass_matrix.html
@@ -215,6 +215,7 @@
 <li class="toctree-l2 has-children"><a class="reference internal" href="../../../autoapi/blackjax/vi/index.html"><code class="xref py py-mod docutils literal notranslate"><span class="pre">blackjax.vi</span></code></a><input class="toctree-checkbox" id="toctree-checkbox-6" name="toctree-checkbox-6" type="checkbox"/><label class="toctree-toggle" for="toctree-checkbox-6"><i class="fa-solid fa-chevron-down"></i></label><ul>
 <li class="toctree-l3"><a class="reference internal" href="../../../autoapi/blackjax/vi/meanfield_vi/index.html"><code class="xref py py-mod docutils literal notranslate"><span class="pre">blackjax.vi.meanfield_vi</span></code></a></li>
 <li class="toctree-l3"><a class="reference internal" href="../../../autoapi/blackjax/vi/pathfinder/index.html"><code class="xref py py-mod docutils literal notranslate"><span class="pre">blackjax.vi.pathfinder</span></code></a></li>
+<li class="toctree-l3"><a class="reference internal" href="../../../autoapi/blackjax/vi/schrodinger_follmer/index.html"><code class="xref py py-mod docutils literal notranslate"><span class="pre">blackjax.vi.schrodinger_follmer</span></code></a></li>
 <li class="toctree-l3"><a class="reference internal" href="../../../autoapi/blackjax/vi/svgd/index.html"><code class="xref py py-mod docutils literal notranslate"><span class="pre">blackjax.vi.svgd</span></code></a></li>
 </ul>
 </li>

diff --git a/_modules/blackjax/adaptation/meads_adaptation.html b/_modules/blackjax/adaptation/meads_adaptation.html
@@ -215,6 +215,7 @@
 <li class="toctree-l2 has-children"><a class="reference internal" href="../../../autoapi/blackjax/vi/index.html"><code class="xref py py-mod docutils literal notranslate"><span class="pre">blackjax.vi</span></code></a><input class="toctree-checkbox" id="toctree-checkbox-6" name="toctree-checkbox-6" type="checkbox"/><label class="toctree-toggle" for="toctree-checkbox-6"><i class="fa-solid fa-chevron-down"></i></label><ul>
 <li class="toctree-l3"><a class="reference internal" href="../../../autoapi/blackjax/vi/meanfield_vi/index.html"><code class="xref py py-mod docutils literal notranslate"><span class="pre">blackjax.vi.meanfield_vi</span></code></a></li>
 <li class="toctree-l3"><a class="reference internal" href="../../../autoapi/blackjax/vi/pathfinder/index.html"><code class="xref py py-mod docutils literal notranslate"><span class="pre">blackjax.vi.pathfinder</span></code></a></li>
+<li class="toctree-l3"><a class="reference internal" href="../../../autoapi/blackjax/vi/schrodinger_follmer/index.html"><code class="xref py py-mod docutils literal notranslate"><span class="pre">blackjax.vi.schrodinger_follmer</span></code></a></li>
 <li class="toctree-l3"><a class="reference internal" href="../../../autoapi/blackjax/vi/svgd/index.html"><code class="xref py py-mod docutils literal notranslate"><span class="pre">blackjax.vi.svgd</span></code></a></li>
 </ul>
 </li>

diff --git a/_modules/blackjax/adaptation/pathfinder_adaptation.html b/_modules/blackjax/adaptation/pathfinder_adaptation.html
@@ -215,6 +215,7 @@
 <li class="toctree-l2 has-children"><a class="reference internal" href="../../../autoapi/blackjax/vi/index.html"><code class="xref py py-mod docutils literal notranslate"><span class="pre">blackjax.vi</span></code></a><input class="toctree-checkbox" id="toctree-checkbox-6" name="toctree-checkbox-6" type="checkbox"/><label class="toctree-toggle" for="toctree-checkbox-6"><i class="fa-solid fa-chevron-down"></i></label><ul>
 <li class="toctree-l3"><a class="reference internal" href="../../../autoapi/blackjax/vi/meanfield_vi/index.html"><code class="xref py py-mod docutils literal notranslate"><span class="pre">blackjax.vi.meanfield_vi</span></code></a></li>
 <li class="toctree-l3"><a class="reference internal" href="../../../autoapi/blackjax/vi/pathfinder/index.html"><code class="xref py py-mod docutils literal notranslate"><span class="pre">blackjax.vi.pathfinder</span></code></a></li>
+<li class="toctree-l3"><a class="reference internal" href="../../../autoapi/blackjax/vi/schrodinger_follmer/index.html"><code class="xref py py-mod docutils literal notranslate"><span class="pre">blackjax.vi.schrodinger_follmer</span></code></a></li>
 <li class="toctree-l3"><a class="reference internal" href="../../../autoapi/blackjax/vi/svgd/index.html"><code class="xref py py-mod docutils literal notranslate"><span class="pre">blackjax.vi.svgd</span></code></a></li>
 </ul>
 </li>

diff --git a/_modules/blackjax/adaptation/step_size.html b/_modules/blackjax/adaptation/step_size.html
@@ -215,6 +215,7 @@
 <li class="toctree-l2 has-children"><a class="reference internal" href="../../../autoapi/blackjax/vi/index.html"><code class="xref py py-mod docutils literal notranslate"><span class="pre">blackjax.vi</span></code></a><input class="toctree-checkbox" id="toctree-checkbox-6" name="toctree-checkbox-6" type="checkbox"/><label class="toctree-toggle" for="toctree-checkbox-6"><i class="fa-solid fa-chevron-down"></i></label><ul>
 <li class="toctree-l3"><a class="reference internal" href="../../../autoapi/blackjax/vi/meanfield_vi/index.html"><code class="xref py py-mod docutils literal notranslate"><span class="pre">blackjax.vi.meanfield_vi</span></code></a></li>
 <li class="toctree-l3"><a class="reference internal" href="../../../autoapi/blackjax/vi/pathfinder/index.html"><code class="xref py py-mod docutils literal notranslate"><span class="pre">blackjax.vi.pathfinder</span></code></a></li>
+<li class="toctree-l3"><a class="reference internal" href="../../../autoapi/blackjax/vi/schrodinger_follmer/index.html"><code class="xref py py-mod docutils literal notranslate"><span class="pre">blackjax.vi.schrodinger_follmer</span></code></a></li>
 <li class="toctree-l3"><a class="reference internal" href="../../../autoapi/blackjax/vi/svgd/index.html"><code class="xref py py-mod docutils literal notranslate"><span class="pre">blackjax.vi.svgd</span></code></a></li>
 </ul>
 </li>

diff --git a/_modules/blackjax/adaptation/window_adaptation.html b/_modules/blackjax/adaptation/window_adaptation.html
@@ -215,6 +215,7 @@
 <li class="toctree-l2 has-children"><a class="reference internal" href="../../../autoapi/blackjax/vi/index.html"><code class="xref py py-mod docutils literal notranslate"><span class="pre">blackjax.vi</span></code></a><input class="toctree-checkbox" id="toctree-checkbox-6" name="toctree-checkbox-6" type="checkbox"/><label class="toctree-toggle" for="toctree-checkbox-6"><i class="fa-solid fa-chevron-down"></i></label><ul>
 <li class="toctree-l3"><a class="reference internal" href="../../../autoapi/blackjax/vi/meanfield_vi/index.html"><code class="xref py py-mod docutils literal notranslate"><span class="pre">blackjax.vi.meanfield_vi</span></code></a></li>
 <li class="toctree-l3"><a class="reference internal" href="../../../autoapi/blackjax/vi/pathfinder/index.html"><code class="xref py py-mod docutils literal notranslate"><span class="pre">blackjax.vi.pathfinder</span></code></a></li>
+<li class="toctree-l3"><a class="reference internal" href="../../../autoapi/blackjax/vi/schrodinger_follmer/index.html"><code class="xref py py-mod docutils literal notranslate"><span class="pre">blackjax.vi.schrodinger_follmer</span></code></a></li>
 <li class="toctree-l3"><a class="reference internal" href="../../../autoapi/blackjax/vi/svgd/index.html"><code class="xref py py-mod docutils literal notranslate"><span class="pre">blackjax.vi.svgd</span></code></a></li>
 </ul>
 </li>

diff --git a/_modules/blackjax/base.html b/_modules/blackjax/base.html
@@ -215,6 +215,7 @@
 <li class="toctree-l2 has-children"><a class="reference internal" href="../../autoapi/blackjax/vi/index.html"><code class="xref py py-mod docutils literal notranslate"><span class="pre">blackjax.vi</span></code></a><input class="toctree-checkbox" id="toctree-checkbox-6" name="toctree-checkbox-6" type="checkbox"/><label class="toctree-toggle" for="toctree-checkbox-6"><i class="fa-solid fa-chevron-down"></i></label><ul>
 <li class="toctree-l3"><a class="reference internal" href="../../autoapi/blackjax/vi/meanfield_vi/index.html"><code class="xref py py-mod docutils literal notranslate"><span class="pre">blackjax.vi.meanfield_vi</span></code></a></li>
 <li class="toctree-l3"><a class="reference internal" href="../../autoapi/blackjax/vi/pathfinder/index.html"><code class="xref py py-mod docutils literal notranslate"><span class="pre">blackjax.vi.pathfinder</span></code></a></li>
+<li class="toctree-l3"><a class="reference internal" href="../../autoapi/blackjax/vi/schrodinger_follmer/index.html"><code class="xref py py-mod docutils literal notranslate"><span class="pre">blackjax.vi.schrodinger_follmer</span></code></a></li>
 <li class="toctree-l3"><a class="reference internal" href="../../autoapi/blackjax/vi/svgd/index.html"><code class="xref py py-mod docutils literal notranslate"><span class="pre">blackjax.vi.svgd</span></code></a></li>
 </ul>
 </li>

diff --git a/_modules/blackjax/diagnostics.html b/_modules/blackjax/diagnostics.html
@@ -215,6 +215,7 @@
 <li class="toctree-l2 has-children"><a class="reference internal" href="../../autoapi/blackjax/vi/index.html"><code class="xref py py-mod docutils literal notranslate"><span class="pre">blackjax.vi</span></code></a><input class="toctree-checkbox" id="toctree-checkbox-6" name="toctree-checkbox-6" type="checkbox"/><label class="toctree-toggle" for="toctree-checkbox-6"><i class="fa-solid fa-chevron-down"></i></label><ul>
 <li class="toctree-l3"><a class="reference internal" href="../../autoapi/blackjax/vi/meanfield_vi/index.html"><code class="xref py py-mod docutils literal notranslate"><span class="pre">blackjax.vi.meanfield_vi</span></code></a></li>
 <li class="toctree-l3"><a class="reference internal" href="../../autoapi/blackjax/vi/pathfinder/index.html"><code class="xref py py-mod docutils literal notranslate"><span class="pre">blackjax.vi.pathfinder</span></code></a></li>
+<li class="toctree-l3"><a class="reference internal" href="../../autoapi/blackjax/vi/schrodinger_follmer/index.html"><code class="xref py py-mod docutils literal notranslate"><span class="pre">blackjax.vi.schrodinger_follmer</span></code></a></li>
 <li class="toctree-l3"><a class="reference internal" href="../../autoapi/blackjax/vi/svgd/index.html"><code class="xref py py-mod docutils literal notranslate"><span class="pre">blackjax.vi.svgd</span></code></a></li>
 </ul>
 </li>