deploy: f73c906

danielward27 · Oct 10, 2024 · 063c411 · 063c411
commit 063c411
Show file tree

Hide file tree

Showing 278 changed files with 37,146 additions and 0 deletions.
diff --git a/.buildinfo b/.buildinfo
@@ -0,0 +1,4 @@
+# Sphinx build info version 1
+# This file records the configuration used when building these files. When it is not found, a full rebuild will be done.
+config: 0f868d9d0fc9144c26a2bd2b83014639
+tags: 645f666f9bcd5a90fca523b33c5a78b7
diff --git a/.doctrees/api/bijections.doctree b/.doctrees/api/bijections.doctree
diff --git a/.doctrees/api/distributions.doctree b/.doctrees/api/distributions.doctree
diff --git a/.doctrees/api/experimental.doctree b/.doctrees/api/experimental.doctree
diff --git a/.doctrees/api/flows.doctree b/.doctrees/api/flows.doctree
diff --git a/.doctrees/api/losses.doctree b/.doctrees/api/losses.doctree
diff --git a/.doctrees/api/training.doctree b/.doctrees/api/training.doctree
diff --git a/.doctrees/api/wrappers.doctree b/.doctrees/api/wrappers.doctree
diff --git a/.doctrees/environment.pickle b/.doctrees/environment.pickle
diff --git a/.doctrees/examples/conditional.doctree b/.doctrees/examples/conditional.doctree
diff --git a/.doctrees/examples/constrained.doctree b/.doctrees/examples/constrained.doctree
diff --git a/.doctrees/examples/examples.doctree b/.doctrees/examples/examples.doctree
diff --git a/.doctrees/examples/snpe.doctree b/.doctrees/examples/snpe.doctree
diff --git a/.doctrees/examples/unconditional.doctree b/.doctrees/examples/unconditional.doctree
diff --git a/.doctrees/examples/variational_inference.doctree b/.doctrees/examples/variational_inference.doctree
diff --git a/.doctrees/faq.doctree b/.doctrees/faq.doctree
diff --git a/.doctrees/getting_started.doctree b/.doctrees/getting_started.doctree
diff --git a/.doctrees/index.doctree b/.doctrees/index.doctree
diff --git a/.doctrees/nbsphinx/examples/conditional.ipynb b/.doctrees/nbsphinx/examples/conditional.ipynb
diff --git a/.doctrees/nbsphinx/examples/constrained.ipynb b/.doctrees/nbsphinx/examples/constrained.ipynb
diff --git a/.doctrees/nbsphinx/examples/snpe.ipynb b/.doctrees/nbsphinx/examples/snpe.ipynb
diff --git a/.doctrees/nbsphinx/examples/unconditional.ipynb b/.doctrees/nbsphinx/examples/unconditional.ipynb
diff --git a/.doctrees/nbsphinx/examples/variational_inference.ipynb b/.doctrees/nbsphinx/examples/variational_inference.ipynb
diff --git a/.doctrees/nbsphinx/examples_conditional_8_0.png b/.doctrees/nbsphinx/examples_conditional_8_0.png
diff --git a/.doctrees/nbsphinx/examples_constrained_10_0.png b/.doctrees/nbsphinx/examples_constrained_10_0.png
diff --git a/.doctrees/nbsphinx/examples_snpe_12_0.png b/.doctrees/nbsphinx/examples_snpe_12_0.png
diff --git a/.doctrees/nbsphinx/examples_unconditional_12_0.png b/.doctrees/nbsphinx/examples_unconditional_12_0.png
diff --git a/.doctrees/nbsphinx/examples_variational_inference_7_1.png b/.doctrees/nbsphinx/examples_variational_inference_7_1.png
diff --git a/.doctrees/nbsphinx/examples_variational_inference_9_0.png b/.doctrees/nbsphinx/examples_variational_inference_9_0.png
diff --git a/.nojekyll b/.nojekyll
diff --git a/_images/examples_conditional_8_0.png b/_images/examples_conditional_8_0.png
diff --git a/_images/examples_constrained_10_0.png b/_images/examples_constrained_10_0.png
diff --git a/_images/examples_snpe_12_0.png b/_images/examples_snpe_12_0.png
diff --git a/_images/examples_unconditional_12_0.png b/_images/examples_unconditional_12_0.png
diff --git a/_images/examples_variational_inference_7_1.png b/_images/examples_variational_inference_7_1.png
diff --git a/_images/examples_variational_inference_9_0.png b/_images/examples_variational_inference_9_0.png
diff --git a/_modules/flowjax/bijections/affine.html b/_modules/flowjax/bijections/affine.html
diff --git a/_modules/flowjax/bijections/bijection.html b/_modules/flowjax/bijections/bijection.html
diff --git a/_modules/flowjax/bijections/block_autoregressive_network.html b/_modules/flowjax/bijections/block_autoregressive_network.html
diff --git a/_modules/flowjax/bijections/chain.html b/_modules/flowjax/bijections/chain.html
diff --git a/_modules/flowjax/bijections/concatenate.html b/_modules/flowjax/bijections/concatenate.html
diff --git a/_modules/flowjax/bijections/coupling.html b/_modules/flowjax/bijections/coupling.html
diff --git a/_modules/flowjax/bijections/exp.html b/_modules/flowjax/bijections/exp.html
diff --git a/_modules/flowjax/bijections/jax_transforms.html b/_modules/flowjax/bijections/jax_transforms.html
diff --git a/_modules/flowjax/bijections/masked_autoregressive.html b/_modules/flowjax/bijections/masked_autoregressive.html
diff --git a/_modules/flowjax/bijections/planar.html b/_modules/flowjax/bijections/planar.html
diff --git a/_modules/flowjax/bijections/power.html b/_modules/flowjax/bijections/power.html
diff --git a/_modules/flowjax/bijections/rational_quadratic_spline.html b/_modules/flowjax/bijections/rational_quadratic_spline.html
diff --git a/_modules/flowjax/bijections/sigmoid.html b/_modules/flowjax/bijections/sigmoid.html
diff --git a/_modules/flowjax/bijections/softplus.html b/_modules/flowjax/bijections/softplus.html
diff --git a/_modules/flowjax/bijections/tanh.html b/_modules/flowjax/bijections/tanh.html
diff --git a/_modules/flowjax/bijections/utils.html b/_modules/flowjax/bijections/utils.html
diff --git a/_modules/flowjax/distributions.html b/_modules/flowjax/distributions.html
diff --git a/_modules/flowjax/experimental/numpyro.html b/_modules/flowjax/experimental/numpyro.html
diff --git a/_modules/flowjax/flows.html b/_modules/flowjax/flows.html
diff --git a/_modules/flowjax/train/data_fit.html b/_modules/flowjax/train/data_fit.html
diff --git a/_modules/flowjax/train/losses.html b/_modules/flowjax/train/losses.html
diff --git a/_modules/flowjax/train/train_utils.html b/_modules/flowjax/train/train_utils.html
diff --git a/_modules/flowjax/train/variational_fit.html b/_modules/flowjax/train/variational_fit.html
diff --git a/_modules/flowjax/wrappers.html b/_modules/flowjax/wrappers.html
diff --git a/_modules/index.html b/_modules/index.html
@@ -0,0 +1,389 @@
+
+<!DOCTYPE html>
+
+
+<html lang="en" data-content_root="../" >
+
+  <head>
+    <meta charset="utf-8" />
+    <meta name="viewport" content="width=device-width, initial-scale=1.0" />
+    <title>Overview: module code &#8212; FlowJAX</title>
+
+
+
+  <script data-cfasync="false">
+    document.documentElement.dataset.mode = localStorage.getItem("mode") || "";
+    document.documentElement.dataset.theme = localStorage.getItem("theme") || "";
+  </script>
+
+  <!-- Loaded before other Sphinx assets -->
+  <link href="../_static/styles/theme.css?digest=dfe6caa3a7d634c4db9b" rel="stylesheet" />
+<link href="../_static/styles/bootstrap.css?digest=dfe6caa3a7d634c4db9b" rel="stylesheet" />
+<link href="../_static/styles/pydata-sphinx-theme.css?digest=dfe6caa3a7d634c4db9b" rel="stylesheet" />
+
+
+  <link href="../_static/vendor/fontawesome/6.5.2/css/all.min.css?digest=dfe6caa3a7d634c4db9b" rel="stylesheet" />
+  <link rel="preload" as="font" type="font/woff2" crossorigin href="../_static/vendor/fontawesome/6.5.2/webfonts/fa-solid-900.woff2" />
+<link rel="preload" as="font" type="font/woff2" crossorigin href="../_static/vendor/fontawesome/6.5.2/webfonts/fa-brands-400.woff2" />
+<link rel="preload" as="font" type="font/woff2" crossorigin href="../_static/vendor/fontawesome/6.5.2/webfonts/fa-regular-400.woff2" />
+
+    <link rel="stylesheet" type="text/css" href="../_static/pygments.css?v=3d3fbe02" />
+    <link rel="stylesheet" type="text/css" href="../_static/styles/sphinx-book-theme.css?v=a3416100" />
+    <link rel="stylesheet" type="text/css" href="../_static/copybutton.css?v=76b2166b" />
+    <link rel="stylesheet" type="text/css" href="../_static/style.css?v=a31e0150" />
+
+  <!-- Pre-loaded scripts that we'll load fully later -->
+  <link rel="preload" as="script" href="../_static/scripts/bootstrap.js?digest=dfe6caa3a7d634c4db9b" />
+<link rel="preload" as="script" href="../_static/scripts/pydata-sphinx-theme.js?digest=dfe6caa3a7d634c4db9b" />
+  <script src="../_static/vendor/fontawesome/6.5.2/js/all.min.js?digest=dfe6caa3a7d634c4db9b"></script>
+
+    <script src="../_static/documentation_options.js?v=9eb32ce0"></script>
+    <script src="../_static/doctools.js?v=9bcbadda"></script>
+    <script src="../_static/sphinx_highlight.js?v=dc90522c"></script>
+    <script src="../_static/clipboard.min.js?v=a7894cd8"></script>
+    <script src="../_static/copybutton.js?v=ccdb6887"></script>
+    <script src="../_static/scripts/sphinx-book-theme.js?v=887ef09a"></script>
+    <script crossorigin="anonymous" integrity="sha256-Ae2Vz/4ePdIu6ZyI/5ZGsYnb+m0JlOmKPjt6XZ9JJkA=" src="https://cdnjs.cloudflare.com/ajax/libs/require.js/2.3.4/require.min.js"></script>
+    <script>DOCUMENTATION_OPTIONS.pagename = '_modules/index';</script>
+    <link rel="icon" href="../_static/icon.svg"/>
+    <link rel="index" title="Index" href="../genindex.html" />
+    <link rel="search" title="Search" href="../search.html" />
+  <meta name="viewport" content="width=device-width, initial-scale=1"/>
+  <meta name="docsearch:language" content="en"/>
+  </head>
+
+
+  <body data-bs-spy="scroll" data-bs-target=".bd-toc-nav" data-offset="180" data-bs-root-margin="0px 0px -60%" data-default-mode="">
+
+
+
+  <div id="pst-skip-link" class="skip-link d-print-none"><a href="#main-content">Skip to main content</a></div>
+
+  <div id="pst-scroll-pixel-helper"></div>
+
+  <button type="button" class="btn rounded-pill" id="pst-back-to-top">
+    <i class="fa-solid fa-arrow-up"></i>Back to top</button>
+
+
+  <input type="checkbox"
+          class="sidebar-toggle"
+          id="pst-primary-sidebar-checkbox"/>
+  <label class="overlay overlay-primary" for="pst-primary-sidebar-checkbox"></label>
+
+  <input type="checkbox"
+          class="sidebar-toggle"
+          id="pst-secondary-sidebar-checkbox"/>
+  <label class="overlay overlay-secondary" for="pst-secondary-sidebar-checkbox"></label>
+
+  <div class="search-button__wrapper">
+    <div class="search-button__overlay"></div>
+    <div class="search-button__search-container">
+<form class="bd-search d-flex align-items-center"
+      action="../search.html"
+      method="get">
+  <i class="fa-solid fa-magnifying-glass"></i>
+  <input type="search"
+         class="form-control"
+         name="q"
+         id="search-input"
+         placeholder="Search..."
+         aria-label="Search..."
+         autocomplete="off"
+         autocorrect="off"
+         autocapitalize="off"
+         spellcheck="false"/>
+  <span class="search-button__kbd-shortcut"><kbd class="kbd-shortcut__modifier">Ctrl</kbd>+<kbd>K</kbd></span>
+</form></div>
+  </div>
+
+  <div class="pst-async-banner-revealer d-none">
+  <aside id="bd-header-version-warning" class="d-none d-print-none" aria-label="Version warning"></aside>
+</div>
+
+
+    <header class="bd-header navbar navbar-expand-lg bd-navbar d-print-none">
+    </header>
+
+
+  <div class="bd-container">
+    <div class="bd-container__inner bd-page-width">
+
+
+
+
+
+      <div class="bd-sidebar-primary bd-sidebar">
+
+
+
+  <div class="sidebar-header-items sidebar-primary__section">
+
+
+
+
+  </div>
+
+    <div class="sidebar-primary-items__start sidebar-primary__section">
+        <div class="sidebar-primary-item">
+
+
+
+
+
+<a class="navbar-brand logo" href="../index.html">
+
+
+
+
+
+
+
+
+
+
+    <img src="../_static/logo_light.svg" class="logo__image only-light" alt="FlowJAX - Home"/>
+    <script>document.write(`<img src="../_static/logo_dark.svg" class="logo__image only-dark" alt="FlowJAX - Home"/>`);</script>
+
+
+</a></div>
+        <div class="sidebar-primary-item">
+
+ <script>
+ document.write(`
+   <button class="btn search-button-field search-button__button" title="Search" aria-label="Search" data-bs-placement="bottom" data-bs-toggle="tooltip">
+    <i class="fa-solid fa-magnifying-glass"></i>
+    <span class="search-button__default-text">Search</span>
+    <span class="search-button__kbd-shortcut"><kbd class="kbd-shortcut__modifier">Ctrl</kbd>+<kbd class="kbd-shortcut__modifier">K</kbd></span>
+   </button>
+ `);
+ </script></div>
+        <div class="sidebar-primary-item"><nav class="bd-links bd-docs-nav" aria-label="Main">
+    <div class="bd-toc-item navbar-nav active">
+
+        <ul class="nav bd-sidenav bd-sidenav__home-link">
+            <li class="toctree-l1">
+                <a class="reference internal" href="../index.html">
+                    FlowJAX
+                </a>
+            </li>
+        </ul>
+        <p aria-level="2" class="caption" role="heading"><span class="caption-text">Getting started</span></p>
+<ul class="nav bd-sidenav">
+<li class="toctree-l1"><a class="reference internal" href="../getting_started.html">Getting started</a></li>
+</ul>
+<p aria-level="2" class="caption" role="heading"><span class="caption-text">Examples</span></p>
+<ul class="nav bd-sidenav">
+<li class="toctree-l1"><a class="reference internal" href="../examples/unconditional.html">Unconditional density estmation</a></li>
+<li class="toctree-l1"><a class="reference internal" href="../examples/conditional.html">Conditional density estimation</a></li>
+<li class="toctree-l1"><a class="reference internal" href="../examples/variational_inference.html">Variational inference</a></li>
+<li class="toctree-l1"><a class="reference internal" href="../examples/constrained.html">Flows with constrained support</a></li>
+<li class="toctree-l1"><a class="reference internal" href="../examples/snpe.html">Sequential neural posterior estimation</a></li>
+</ul>
+<p aria-level="2" class="caption" role="heading"><span class="caption-text">API</span></p>
+<ul class="nav bd-sidenav">
+<li class="toctree-l1"><a class="reference internal" href="../api/distributions.html">Distributions</a></li>
+<li class="toctree-l1"><a class="reference internal" href="../api/bijections.html">Bijections</a></li>
+<li class="toctree-l1"><a class="reference internal" href="../api/flows.html">Flows</a></li>
+<li class="toctree-l1"><a class="reference internal" href="../api/training.html">Training</a></li>
+<li class="toctree-l1"><a class="reference internal" href="../api/losses.html">Loss functions</a></li>
+<li class="toctree-l1"><a class="reference internal" href="../api/wrappers.html">Wrappers</a></li>
+<li class="toctree-l1"><a class="reference internal" href="../api/experimental.html">Experimental numpyro interface</a></li>
+</ul>
+<p aria-level="2" class="caption" role="heading"><span class="caption-text">Miscellaneous</span></p>
+<ul class="nav bd-sidenav">
+<li class="toctree-l1"><a class="reference internal" href="../faq.html">FAQ</a></li>
+</ul>
+
+    </div>
+</nav></div>
+    </div>
+
+
+  <div class="sidebar-primary-items__end sidebar-primary__section">
+  </div>
+
+  <div id="rtd-footer-container"></div>
+
+
+      </div>
+
+      <main id="main-content" class="bd-main" role="main">
+
+
+
+<div class="sbt-scroll-pixel-helper"></div>
+
+          <div class="bd-content">
+            <div class="bd-article-container">
+
+              <div class="bd-header-article d-print-none">
+<div class="header-article-items header-article__inner">
+
+    <div class="header-article-items__start">
+
+        <div class="header-article-item"><button class="sidebar-toggle primary-toggle btn btn-sm" title="Toggle primary sidebar" data-bs-placement="bottom" data-bs-toggle="tooltip">
+  <span class="fa-solid fa-bars"></span>
+</button></div>
+
+    </div>
+
+
+    <div class="header-article-items__end">
+
+        <div class="header-article-item">
+
+<div class="article-header-buttons">
+
+
+<a href="https://github.com/danielward27/flowjax" target="_blank"
+   class="btn btn-sm btn-source-repository-button"
+   title="Source repository"
+   data-bs-placement="bottom" data-bs-toggle="tooltip"
+>
+
+
+<span class="btn__icon-container">
+  <i class="fab fa-github"></i>
+  </span>
+
+</a>
+
+
+
+<script>
+document.write(`
+  <button class="btn btn-sm nav-link pst-navbar-icon theme-switch-button" title="light/dark" aria-label="light/dark" data-bs-placement="bottom" data-bs-toggle="tooltip">
+    <i class="theme-switch fa-solid fa-sun fa-lg" data-mode="light"></i>
+    <i class="theme-switch fa-solid fa-moon fa-lg" data-mode="dark"></i>
+    <i class="theme-switch fa-solid fa-circle-half-stroke fa-lg" data-mode="auto"></i>
+  </button>
+`);
+</script>
+
+
+<script>
+document.write(`
+  <button class="btn btn-sm pst-navbar-icon search-button search-button__button" title="Search" aria-label="Search" data-bs-placement="bottom" data-bs-toggle="tooltip">
+    <i class="fa-solid fa-magnifying-glass fa-lg"></i>
+  </button>
+`);
+</script>
+
+</div></div>
+
+    </div>
+
+</div>
+</div>
+
+
+
+<div id="jb-print-docs-body" class="onlyprint">
+    <h1></h1>
+    <!-- Table of contents -->
+    <div id="print-main-content">
+        <div id="jb-print-toc">
+
+        </div>
+    </div>
+</div>
+
+
+
+<div id="searchbox"></div>
+                <article class="bd-article">
+
+  <h1>All modules for which code is available</h1>
+<ul><li><a href="flowjax/bijections/affine.html">flowjax.bijections.affine</a></li>
+<li><a href="flowjax/bijections/bijection.html">flowjax.bijections.bijection</a></li>
+<li><a href="flowjax/bijections/block_autoregressive_network.html">flowjax.bijections.block_autoregressive_network</a></li>
+<li><a href="flowjax/bijections/chain.html">flowjax.bijections.chain</a></li>
+<li><a href="flowjax/bijections/concatenate.html">flowjax.bijections.concatenate</a></li>
+<li><a href="flowjax/bijections/coupling.html">flowjax.bijections.coupling</a></li>
+<li><a href="flowjax/bijections/exp.html">flowjax.bijections.exp</a></li>
+<li><a href="flowjax/bijections/jax_transforms.html">flowjax.bijections.jax_transforms</a></li>
+<li><a href="flowjax/bijections/masked_autoregressive.html">flowjax.bijections.masked_autoregressive</a></li>
+<li><a href="flowjax/bijections/planar.html">flowjax.bijections.planar</a></li>
+<li><a href="flowjax/bijections/power.html">flowjax.bijections.power</a></li>
+<li><a href="flowjax/bijections/rational_quadratic_spline.html">flowjax.bijections.rational_quadratic_spline</a></li>
+<li><a href="flowjax/bijections/sigmoid.html">flowjax.bijections.sigmoid</a></li>
+<li><a href="flowjax/bijections/softplus.html">flowjax.bijections.softplus</a></li>
+<li><a href="flowjax/bijections/tanh.html">flowjax.bijections.tanh</a></li>
+<li><a href="flowjax/bijections/utils.html">flowjax.bijections.utils</a></li>
+<li><a href="flowjax/distributions.html">flowjax.distributions</a></li>
+<li><a href="flowjax/experimental/numpyro.html">flowjax.experimental.numpyro</a></li>
+<li><a href="flowjax/flows.html">flowjax.flows</a></li>
+<li><a href="flowjax/train/data_fit.html">flowjax.train.data_fit</a></li>
+<li><a href="flowjax/train/losses.html">flowjax.train.losses</a></li>
+<li><a href="flowjax/train/train_utils.html">flowjax.train.train_utils</a></li>
+<li><a href="flowjax/train/variational_fit.html">flowjax.train.variational_fit</a></li>
+<li><a href="flowjax/wrappers.html">flowjax.wrappers</a></li>
+</ul>
+
+                </article>
+
+
+
+
+
+
+                <footer class="prev-next-footer d-print-none">
+
+<div class="prev-next-area">
+</div>
+                </footer>
+
+            </div>
+
+
+
+
+          </div>
+          <footer class="bd-footer-content">
+
+<div class="bd-footer-content__inner container">
+
+  <div class="footer-item">
+
+<p class="component-author">
+By Daniel Ward
+</p>
+
+  </div>
+
+  <div class="footer-item">
+
+
+  <p class="copyright">
+
+      © Copyright 2022, Daniel Ward.
+      <br/>
+
+  </p>
+
+  </div>
+
+  <div class="footer-item">
+
+  </div>
+
+  <div class="footer-item">
+
+  </div>
+
+</div>
+          </footer>
+
+
+      </main>
+    </div>
+  </div>
+
+  <!-- Scripts loaded after <body> so the DOM is not blocked -->
+  <script src="../_static/scripts/bootstrap.js?digest=dfe6caa3a7d634c4db9b"></script>
+<script src="../_static/scripts/pydata-sphinx-theme.js?digest=dfe6caa3a7d634c4db9b"></script>
+
+  <footer class="bd-footer">
+  </footer>
+  </body>
+</html>