Deploying to gh-pages from @ 5831740 🚀

blackjax-devs · May 25, 2024 · 6576e4c · 6576e4c
1 parent 93b8b48
commit 6576e4c
Show file tree

Hide file tree

Showing 289 changed files with 8,675 additions and 19,159 deletions.
diff --git a/.buildinfo b/.buildinfo
@@ -1,4 +1,4 @@
 # Sphinx build info version 1
 # This file hashes the configuration used when building these files. When it is not found, a full rebuild will be done.
-config: b6dca085176953f0485f15a9264eeb81
+config: ccdeaad7804c3056e1cb695fc7832455
 tags: 645f666f9bcd5a90fca523b33c5a78b7
diff --git a/.doctrees/autoapi/blackjax/_version/index.doctree b/.doctrees/autoapi/blackjax/_version/index.doctree
diff --git a/.doctrees/autoapi/blackjax/adaptation/base/index.doctree b/.doctrees/autoapi/blackjax/adaptation/base/index.doctree
diff --git a/.doctrees/autoapi/blackjax/adaptation/chees_adaptation/index.doctree b/.doctrees/autoapi/blackjax/adaptation/chees_adaptation/index.doctree
diff --git a/.doctrees/autoapi/blackjax/adaptation/index.doctree b/.doctrees/autoapi/blackjax/adaptation/index.doctree
diff --git a/.doctrees/autoapi/blackjax/adaptation/mass_matrix/index.doctree b/.doctrees/autoapi/blackjax/adaptation/mass_matrix/index.doctree
diff --git a/.doctrees/autoapi/blackjax/adaptation/mclmc_adaptation/index.doctree b/.doctrees/autoapi/blackjax/adaptation/mclmc_adaptation/index.doctree
diff --git a/.doctrees/autoapi/blackjax/adaptation/meads_adaptation/index.doctree b/.doctrees/autoapi/blackjax/adaptation/meads_adaptation/index.doctree
diff --git a/.doctrees/autoapi/blackjax/adaptation/pathfinder_adaptation/index.doctree b/.doctrees/autoapi/blackjax/adaptation/pathfinder_adaptation/index.doctree
diff --git a/.doctrees/autoapi/blackjax/adaptation/step_size/index.doctree b/.doctrees/autoapi/blackjax/adaptation/step_size/index.doctree
diff --git a/.doctrees/autoapi/blackjax/adaptation/window_adaptation/index.doctree b/.doctrees/autoapi/blackjax/adaptation/window_adaptation/index.doctree
diff --git a/.doctrees/autoapi/blackjax/base/index.doctree b/.doctrees/autoapi/blackjax/base/index.doctree
diff --git a/.doctrees/autoapi/blackjax/diagnostics/index.doctree b/.doctrees/autoapi/blackjax/diagnostics/index.doctree
diff --git a/.doctrees/autoapi/blackjax/index.doctree b/.doctrees/autoapi/blackjax/index.doctree
diff --git a/.doctrees/autoapi/blackjax/mcmc/barker/index.doctree b/.doctrees/autoapi/blackjax/mcmc/barker/index.doctree
diff --git a/.doctrees/autoapi/blackjax/mcmc/diffusions/index.doctree b/.doctrees/autoapi/blackjax/mcmc/diffusions/index.doctree
diff --git a/.doctrees/autoapi/blackjax/mcmc/dynamic_hmc/index.doctree b/.doctrees/autoapi/blackjax/mcmc/dynamic_hmc/index.doctree
diff --git a/.doctrees/autoapi/blackjax/mcmc/elliptical_slice/index.doctree b/.doctrees/autoapi/blackjax/mcmc/elliptical_slice/index.doctree
diff --git a/.doctrees/autoapi/blackjax/mcmc/ghmc/index.doctree b/.doctrees/autoapi/blackjax/mcmc/ghmc/index.doctree
diff --git a/.doctrees/autoapi/blackjax/mcmc/hmc/index.doctree b/.doctrees/autoapi/blackjax/mcmc/hmc/index.doctree
diff --git a/.doctrees/autoapi/blackjax/mcmc/index.doctree b/.doctrees/autoapi/blackjax/mcmc/index.doctree
diff --git a/.doctrees/autoapi/blackjax/mcmc/integrators/index.doctree b/.doctrees/autoapi/blackjax/mcmc/integrators/index.doctree
diff --git a/.doctrees/autoapi/blackjax/mcmc/mala/index.doctree b/.doctrees/autoapi/blackjax/mcmc/mala/index.doctree
diff --git a/.doctrees/autoapi/blackjax/mcmc/marginal_latent_gaussian/index.doctree b/.doctrees/autoapi/blackjax/mcmc/marginal_latent_gaussian/index.doctree
diff --git a/.doctrees/autoapi/blackjax/mcmc/mclmc/index.doctree b/.doctrees/autoapi/blackjax/mcmc/mclmc/index.doctree
diff --git a/.doctrees/autoapi/blackjax/mcmc/metrics/index.doctree b/.doctrees/autoapi/blackjax/mcmc/metrics/index.doctree
diff --git a/.doctrees/autoapi/blackjax/mcmc/nuts/index.doctree b/.doctrees/autoapi/blackjax/mcmc/nuts/index.doctree
diff --git a/.doctrees/autoapi/blackjax/mcmc/periodic_orbital/index.doctree b/.doctrees/autoapi/blackjax/mcmc/periodic_orbital/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/mcmc/rmhmc/index.doctree b/.doctrees/autoapi/blackjax/mcmc/rmhmc/index.doctree
diff --git a/.doctrees/autoapi/blackjax/mcmc/termination/index.doctree b/.doctrees/autoapi/blackjax/mcmc/termination/index.doctree
diff --git a/.doctrees/autoapi/blackjax/mcmc/trajectory/index.doctree b/.doctrees/autoapi/blackjax/mcmc/trajectory/index.doctree
diff --git a/.doctrees/autoapi/blackjax/optimizers/dual_averaging/index.doctree b/.doctrees/autoapi/blackjax/optimizers/dual_averaging/index.doctree
diff --git a/.doctrees/autoapi/blackjax/optimizers/index.doctree b/.doctrees/autoapi/blackjax/optimizers/index.doctree
diff --git a/.doctrees/autoapi/blackjax/optimizers/lbfgs/index.doctree b/.doctrees/autoapi/blackjax/optimizers/lbfgs/index.doctree
diff --git a/.doctrees/autoapi/blackjax/progress_bar/index.doctree b/.doctrees/autoapi/blackjax/progress_bar/index.doctree
diff --git a/.doctrees/autoapi/blackjax/sgmcmc/csgld/index.doctree b/.doctrees/autoapi/blackjax/sgmcmc/csgld/index.doctree
diff --git a/.doctrees/autoapi/blackjax/sgmcmc/diffusions/index.doctree b/.doctrees/autoapi/blackjax/sgmcmc/diffusions/index.doctree
diff --git a/.doctrees/autoapi/blackjax/sgmcmc/gradients/index.doctree b/.doctrees/autoapi/blackjax/sgmcmc/gradients/index.doctree
diff --git a/.doctrees/autoapi/blackjax/sgmcmc/index.doctree b/.doctrees/autoapi/blackjax/sgmcmc/index.doctree
diff --git a/.doctrees/autoapi/blackjax/sgmcmc/sghmc/index.doctree b/.doctrees/autoapi/blackjax/sgmcmc/sghmc/index.doctree
diff --git a/.doctrees/autoapi/blackjax/sgmcmc/sgld/index.doctree b/.doctrees/autoapi/blackjax/sgmcmc/sgld/index.doctree
diff --git a/.doctrees/autoapi/blackjax/sgmcmc/sgnht/index.doctree b/.doctrees/autoapi/blackjax/sgmcmc/sgnht/index.doctree
diff --git a/.doctrees/autoapi/blackjax/smc/adaptive_tempered/index.doctree b/.doctrees/autoapi/blackjax/smc/adaptive_tempered/index.doctree
diff --git a/.doctrees/autoapi/blackjax/smc/base/index.doctree b/.doctrees/autoapi/blackjax/smc/base/index.doctree
diff --git a/.doctrees/autoapi/blackjax/smc/ess/index.doctree b/.doctrees/autoapi/blackjax/smc/ess/index.doctree
diff --git a/.doctrees/autoapi/blackjax/smc/index.doctree b/.doctrees/autoapi/blackjax/smc/index.doctree
diff --git a/.doctrees/autoapi/blackjax/smc/inner_kernel_tuning/index.doctree b/.doctrees/autoapi/blackjax/smc/inner_kernel_tuning/index.doctree
diff --git a/.doctrees/autoapi/blackjax/smc/resampling/index.doctree b/.doctrees/autoapi/blackjax/smc/resampling/index.doctree
diff --git a/.doctrees/autoapi/blackjax/smc/solver/index.doctree b/.doctrees/autoapi/blackjax/smc/solver/index.doctree
diff --git a/.doctrees/autoapi/blackjax/smc/tempered/index.doctree b/.doctrees/autoapi/blackjax/smc/tempered/index.doctree
diff --git a/.doctrees/autoapi/blackjax/smc/tuning/from_kernel_info/index.doctree b/.doctrees/autoapi/blackjax/smc/tuning/from_kernel_info/index.doctree
diff --git a/.doctrees/autoapi/blackjax/smc/tuning/from_particles/index.doctree b/.doctrees/autoapi/blackjax/smc/tuning/from_particles/index.doctree
diff --git a/.doctrees/autoapi/blackjax/smc/tuning/index.doctree b/.doctrees/autoapi/blackjax/smc/tuning/index.doctree
diff --git a/.doctrees/autoapi/blackjax/types/index.doctree b/.doctrees/autoapi/blackjax/types/index.doctree
diff --git a/.doctrees/autoapi/blackjax/util/index.doctree b/.doctrees/autoapi/blackjax/util/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/meanfield_vi/index.doctree b/.doctrees/autoapi/blackjax/vi/meanfield_vi/index.doctree
diff --git a/.doctrees/autoapi/blackjax/vi/pathfinder/index.doctree b/.doctrees/autoapi/blackjax/vi/pathfinder/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/autoapi/blackjax/vi/svgd/index.doctree b/.doctrees/autoapi/blackjax/vi/svgd/index.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_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/0dd79069f08ae0b9bda1bafaa36ec8e3643f04d9dcdb0728ffec9bf09aa3bb89.png b/_images/0dd79069f08ae0b9bda1bafaa36ec8e3643f04d9dcdb0728ffec9bf09aa3bb89.png
diff --git a/_images/2e1d496adb88f7dd5eaad226865a94410f4befc12f7ba61527a3034fcfa777a0.png b/_images/2e1d496adb88f7dd5eaad226865a94410f4befc12f7ba61527a3034fcfa777a0.png
diff --git a/_images/3ce2a18449e540cce13aaf011de59163abe8b0a05d17e0262b483b13e25b2c5d.png b/_images/3ce2a18449e540cce13aaf011de59163abe8b0a05d17e0262b483b13e25b2c5d.png
diff --git a/_images/4b70005ce948eaaf68bc92646aeec9568ee1f3ff473ad23ed317088f87b16196.png b/_images/4b70005ce948eaaf68bc92646aeec9568ee1f3ff473ad23ed317088f87b16196.png
diff --git a/_images/7f8a1eb8ff37be125eee17b937f1e7658655478b3925d3b3d48d3475d9145909.png b/_images/7f8a1eb8ff37be125eee17b937f1e7658655478b3925d3b3d48d3475d9145909.png
diff --git a/_images/888bf14297c75bcc3d3eb23d52241e9f1196fc55829e3582a84f0d935e33ca5f.png b/_images/888bf14297c75bcc3d3eb23d52241e9f1196fc55829e3582a84f0d935e33ca5f.png
diff --git a/_images/8d4794a71fb034f327be217eafa1a88fda8dd2c589e2f8201731b4944601c724.png b/_images/8d4794a71fb034f327be217eafa1a88fda8dd2c589e2f8201731b4944601c724.png
diff --git a/_images/ab6ee30af1d1ced9d716f758e8164f05693bbf3a54aa751df6a367bd800e07e0.png b/_images/ab6ee30af1d1ced9d716f758e8164f05693bbf3a54aa751df6a367bd800e07e0.png
diff --git a/_images/b624019ca22a337e29cfd7d74db26f280971ab35212502b1162f1bb6c254bc54.png b/_images/b624019ca22a337e29cfd7d74db26f280971ab35212502b1162f1bb6c254bc54.png
diff --git a/_images/c170a04dbd82eff88e534ef23b33a39def1137c9c220c6798c211f27f9def39e.png b/_images/c170a04dbd82eff88e534ef23b33a39def1137c9c220c6798c211f27f9def39e.png
diff --git a/_images/c62fa1a4c28bab041deba256dceb2de24fb0349cea90eb05d4746c39a5593c14.png b/_images/c62fa1a4c28bab041deba256dceb2de24fb0349cea90eb05d4746c39a5593c14.png
diff --git a/_images/c69243ef0ab77f905aeba16090eac6c6dc8952876a9905235c3f7e205e870a7b.png b/_images/c69243ef0ab77f905aeba16090eac6c6dc8952876a9905235c3f7e205e870a7b.png
diff --git a/_images/c6e0a5641b52de1f9c91b91c084ea4189a4250ea4692a0cc88b33b2788c593f3.png b/_images/c6e0a5641b52de1f9c91b91c084ea4189a4250ea4692a0cc88b33b2788c593f3.png
diff --git a/_images/cd0da16a3facebe616d6a672ae876a433a4651fc9b3c4e0d3920199c9ae5411c.png b/_images/cd0da16a3facebe616d6a672ae876a433a4651fc9b3c4e0d3920199c9ae5411c.png
diff --git a/_images/d27780bbab52f6a62869b3c70f80343b0e4438d36b4e7965abe43bb68f230aaa.png b/_images/d27780bbab52f6a62869b3c70f80343b0e4438d36b4e7965abe43bb68f230aaa.png
diff --git a/_images/dbf72c71d2a3aa15b4acea37c2c2a1f4ba26bdbc4ea085a67c5975974d5104bd.png b/_images/dbf72c71d2a3aa15b4acea37c2c2a1f4ba26bdbc4ea085a67c5975974d5104bd.png
diff --git a/_images/e9eb736b48686d94a6226776bd7b6bfb693c753931dc5fe94c4647fb29da12cd.png b/_images/e9eb736b48686d94a6226776bd7b6bfb693c753931dc5fe94c4647fb29da12cd.png
diff --git a/_images/faf08ec80528ecf58f06efe01c0562a75d0812b4008c23e40295de51022174ca.png b/_images/faf08ec80528ecf58f06efe01c0562a75d0812b4008c23e40295de51022174ca.png
diff --git a/_modules/blackjax/_version.html b/_modules/blackjax/_version.html
diff --git a/_modules/blackjax/adaptation/base.html b/_modules/blackjax/adaptation/base.html
diff --git a/_modules/blackjax/adaptation/chees_adaptation.html b/_modules/blackjax/adaptation/chees_adaptation.html
diff --git a/_modules/blackjax/adaptation/mass_matrix.html b/_modules/blackjax/adaptation/mass_matrix.html
diff --git a/_modules/blackjax/adaptation/mclmc_adaptation.html b/_modules/blackjax/adaptation/mclmc_adaptation.html
diff --git a/_modules/blackjax/adaptation/meads_adaptation.html b/_modules/blackjax/adaptation/meads_adaptation.html
diff --git a/_modules/blackjax/adaptation/pathfinder_adaptation.html b/_modules/blackjax/adaptation/pathfinder_adaptation.html
diff --git a/_modules/blackjax/adaptation/step_size.html b/_modules/blackjax/adaptation/step_size.html
diff --git a/_modules/blackjax/adaptation/window_adaptation.html b/_modules/blackjax/adaptation/window_adaptation.html
diff --git a/_modules/blackjax/base.html b/_modules/blackjax/base.html
diff --git a/_modules/blackjax/diagnostics.html b/_modules/blackjax/diagnostics.html
diff --git a/_modules/blackjax/mcmc/barker.html b/_modules/blackjax/mcmc/barker.html
diff --git a/_modules/blackjax/mcmc/diffusions.html b/_modules/blackjax/mcmc/diffusions.html
diff --git a/_modules/blackjax/mcmc/dynamic_hmc.html b/_modules/blackjax/mcmc/dynamic_hmc.html
diff --git a/_modules/blackjax/mcmc/elliptical_slice.html b/_modules/blackjax/mcmc/elliptical_slice.html
diff --git a/_modules/blackjax/mcmc/ghmc.html b/_modules/blackjax/mcmc/ghmc.html
diff --git a/_modules/blackjax/mcmc/hmc.html b/_modules/blackjax/mcmc/hmc.html
diff --git a/_modules/blackjax/mcmc/integrators.html b/_modules/blackjax/mcmc/integrators.html
diff --git a/_modules/blackjax/mcmc/mala.html b/_modules/blackjax/mcmc/mala.html
diff --git a/_modules/blackjax/mcmc/marginal_latent_gaussian.html b/_modules/blackjax/mcmc/marginal_latent_gaussian.html
diff --git a/_modules/blackjax/mcmc/mclmc.html b/_modules/blackjax/mcmc/mclmc.html
diff --git a/_modules/blackjax/mcmc/metrics.html b/_modules/blackjax/mcmc/metrics.html
diff --git a/_modules/blackjax/mcmc/nuts.html b/_modules/blackjax/mcmc/nuts.html
diff --git a/_modules/blackjax/mcmc/periodic_orbital.html b/_modules/blackjax/mcmc/periodic_orbital.html
diff --git a/_modules/blackjax/mcmc/proposal.html b/_modules/blackjax/mcmc/proposal.html
diff --git a/_modules/blackjax/mcmc/random_walk.html b/_modules/blackjax/mcmc/random_walk.html
diff --git a/_modules/blackjax/mcmc/rmhmc.html b/_modules/blackjax/mcmc/rmhmc.html
diff --git a/_modules/blackjax/mcmc/termination.html b/_modules/blackjax/mcmc/termination.html
diff --git a/_modules/blackjax/mcmc/trajectory.html b/_modules/blackjax/mcmc/trajectory.html
diff --git a/_modules/blackjax/optimizers/dual_averaging.html b/_modules/blackjax/optimizers/dual_averaging.html
diff --git a/_modules/blackjax/optimizers/lbfgs.html b/_modules/blackjax/optimizers/lbfgs.html
diff --git a/_modules/blackjax/progress_bar.html b/_modules/blackjax/progress_bar.html
diff --git a/_modules/blackjax/sgmcmc/csgld.html b/_modules/blackjax/sgmcmc/csgld.html
diff --git a/_modules/blackjax/sgmcmc/diffusions.html b/_modules/blackjax/sgmcmc/diffusions.html
diff --git a/_modules/blackjax/sgmcmc/gradients.html b/_modules/blackjax/sgmcmc/gradients.html
diff --git a/_modules/blackjax/sgmcmc/sghmc.html b/_modules/blackjax/sgmcmc/sghmc.html
diff --git a/_modules/blackjax/sgmcmc/sgld.html b/_modules/blackjax/sgmcmc/sgld.html
diff --git a/_modules/blackjax/sgmcmc/sgnht.html b/_modules/blackjax/sgmcmc/sgnht.html
diff --git a/_modules/blackjax/smc/adaptive_tempered.html b/_modules/blackjax/smc/adaptive_tempered.html
diff --git a/_modules/blackjax/smc/base.html b/_modules/blackjax/smc/base.html
diff --git a/_modules/blackjax/smc/ess.html b/_modules/blackjax/smc/ess.html
diff --git a/_modules/blackjax/smc/inner_kernel_tuning.html b/_modules/blackjax/smc/inner_kernel_tuning.html
diff --git a/_modules/blackjax/smc/resampling.html b/_modules/blackjax/smc/resampling.html
diff --git a/_modules/blackjax/smc/solver.html b/_modules/blackjax/smc/solver.html
diff --git a/_modules/blackjax/smc/tempered.html b/_modules/blackjax/smc/tempered.html
diff --git a/_modules/blackjax/smc/tuning/from_kernel_info.html b/_modules/blackjax/smc/tuning/from_kernel_info.html
diff --git a/_modules/blackjax/smc/tuning/from_particles.html b/_modules/blackjax/smc/tuning/from_particles.html
diff --git a/_modules/blackjax/types.html b/_modules/blackjax/types.html
diff --git a/_modules/blackjax/util.html b/_modules/blackjax/util.html
diff --git a/_modules/blackjax/vi/meanfield_vi.html b/_modules/blackjax/vi/meanfield_vi.html
diff --git a/_modules/blackjax/vi/pathfinder.html b/_modules/blackjax/vi/pathfinder.html
diff --git a/_modules/blackjax/vi/schrodinger_follmer.html b/_modules/blackjax/vi/schrodinger_follmer.html
diff --git a/_modules/blackjax/vi/svgd.html b/_modules/blackjax/vi/svgd.html
diff --git a/_modules/index.html b/_modules/index.html
diff --git a/_sources/autoapi/blackjax/_version/index.rst b/_sources/autoapi/blackjax/_version/index.rst
diff --git a/_sources/autoapi/blackjax/adaptation/base/index.rst b/_sources/autoapi/blackjax/adaptation/base/index.rst
@@ -1,67 +1,56 @@
-:py:mod:`blackjax.adaptation.base`
-==================================
+blackjax.adaptation.base
+========================
 
 .. py:module:: blackjax.adaptation.base
 
 
-Module Contents
----------------
-
 Classes
-~~~~~~~
+-------
 
 .. autoapisummary::
 
    blackjax.adaptation.base.AdaptationResults
    blackjax.adaptation.base.AdaptationInfo
 
 
-
 Functions
-~~~~~~~~~
+---------
 
 .. autoapisummary::
 
    blackjax.adaptation.base.return_all_adapt_info
    blackjax.adaptation.base.get_filter_adapt_info_fn
 
 
+Module Contents
+---------------
 
 .. py:class:: AdaptationResults
 
 
 
-
    .. py:attribute:: state
-      :type: blackjax.types.ArrayTree
+      :type:  blackjax.types.ArrayTree
 
-
 
    .. py:attribute:: parameters
-      :type: dict
-
-
+      :type:  dict
 
 
 .. py:class:: AdaptationInfo
 
 
 
-
    .. py:attribute:: state
-      :type: NamedTuple
+      :type:  NamedTuple
 
-
 
    .. py:attribute:: info
-      :type: NamedTuple
+      :type:  NamedTuple
 
-
 
    .. py:attribute:: adaptation_state
-      :type: NamedTuple
-
-
+      :type:  NamedTuple
 
 
 .. py:function:: return_all_adapt_info(state, info, adaptation_state)

diff --git a/_sources/autoapi/blackjax/adaptation/chees_adaptation/index.rst b/_sources/autoapi/blackjax/adaptation/chees_adaptation/index.rst
@@ -1,5 +1,5 @@
-:py:mod:`blackjax.adaptation.chees_adaptation`
-==============================================
+blackjax.adaptation.chees_adaptation
+====================================
 
 .. py:module:: blackjax.adaptation.chees_adaptation
 
@@ -9,46 +9,42 @@
 
 
 
-Module Contents
----------------
+Attributes
+----------
+
+.. autoapisummary::
+
+   blackjax.adaptation.chees_adaptation.OPTIMAL_TARGET_ACCEPTANCE_RATE
+
 
 Classes
-~~~~~~~
+-------
 
 .. autoapisummary::
 
    blackjax.adaptation.chees_adaptation.ChEESAdaptationState
 
 
-
 Functions
-~~~~~~~~~
+---------
 
 .. autoapisummary::
 
    blackjax.adaptation.chees_adaptation.base
    blackjax.adaptation.chees_adaptation.chees_adaptation
 
 
-
-Attributes
-~~~~~~~~~~
-
-.. autoapisummary::
-
-   blackjax.adaptation.chees_adaptation.OPTIMAL_TARGET_ACCEPTANCE_RATE
-
+Module Contents
+---------------
 
 .. py:data:: OPTIMAL_TARGET_ACCEPTANCE_RATE
    :value: 0.651
 
-
 
 .. py:class:: ChEESAdaptationState
 
 
 
-
    State of the ChEES-HMC adaptation scheme.
 
    step_size
@@ -70,45 +66,37 @@ Attributes
        Current iteration number.
 
 
+
    .. py:attribute:: step_size
-      :type: float
+      :type:  float
 
-
 
    .. py:attribute:: log_step_size_moving_average
-      :type: float
+      :type:  float
 
-
 
    .. py:attribute:: trajectory_length
-      :type: float
+      :type:  float
 
-
 
    .. py:attribute:: log_trajectory_length_moving_average
-      :type: float
+      :type:  float
 
-
 
    .. py:attribute:: da_state
-      :type: blackjax.optimizers.dual_averaging.DualAveragingState
+      :type:  blackjax.optimizers.dual_averaging.DualAveragingState
 
-
 
    .. py:attribute:: optim_state
-      :type: optax.OptState
+      :type:  optax.OptState
 
-
 
    .. py:attribute:: random_generator_arg
-      :type: blackjax.types.Array
+      :type:  blackjax.types.Array
 
-
 
    .. py:attribute:: step
-      :type: int
-
-
+      :type:  int
 
 
 .. py:function:: base(jitter_generator: Callable, next_random_arg_fn: Callable, optim: optax.GradientTransformation, target_acceptance_rate: float, decay_rate: float) -> Tuple[Callable, Callable]

diff --git a/_sources/autoapi/blackjax/adaptation/index.rst b/_sources/autoapi/blackjax/adaptation/index.rst
@@ -1,22 +1,22 @@
-:py:mod:`blackjax.adaptation`
-=============================
+blackjax.adaptation
+===================
 
 .. py:module:: blackjax.adaptation
 
 
 Submodules
 ----------
+
 .. toctree::
-   :titlesonly:
    :maxdepth: 1
 
-   base/index.rst
-   chees_adaptation/index.rst
-   mass_matrix/index.rst
-   mclmc_adaptation/index.rst
-   meads_adaptation/index.rst
-   pathfinder_adaptation/index.rst
-   step_size/index.rst
-   window_adaptation/index.rst
+   /autoapi/blackjax/adaptation/base/index
+   /autoapi/blackjax/adaptation/chees_adaptation/index
+   /autoapi/blackjax/adaptation/mass_matrix/index
+   /autoapi/blackjax/adaptation/mclmc_adaptation/index
+   /autoapi/blackjax/adaptation/meads_adaptation/index
+   /autoapi/blackjax/adaptation/pathfinder_adaptation/index
+   /autoapi/blackjax/adaptation/step_size/index
+   /autoapi/blackjax/adaptation/window_adaptation/index
 
 
diff --git a/_sources/autoapi/blackjax/adaptation/mass_matrix/index.rst b/_sources/autoapi/blackjax/adaptation/mass_matrix/index.rst
@@ -1,5 +1,5 @@
-:py:mod:`blackjax.adaptation.mass_matrix`
-=========================================
+blackjax.adaptation.mass_matrix
+===============================
 
 .. py:module:: blackjax.adaptation.mass_matrix
 
@@ -13,34 +13,31 @@
 
 
 
-Module Contents
----------------
-
 Classes
-~~~~~~~
+-------
 
 .. autoapisummary::
 
    blackjax.adaptation.mass_matrix.WelfordAlgorithmState
    blackjax.adaptation.mass_matrix.MassMatrixAdaptationState
 
 
-
 Functions
-~~~~~~~~~
+---------
 
 .. autoapisummary::
 
    blackjax.adaptation.mass_matrix.mass_matrix_adaptation
    blackjax.adaptation.mass_matrix.welford_algorithm
 
 
+Module Contents
+---------------
 
 .. py:class:: WelfordAlgorithmState
 
 
 
-
    State carried through the Welford algorithm.
 
    mean
@@ -53,27 +50,23 @@ Functions
        also the current number of iterations of the algorithm.
 
 
+
    .. py:attribute:: mean
-      :type: blackjax.types.Array
+      :type:  blackjax.types.Array
 
-
 
    .. py:attribute:: m2
-      :type: blackjax.types.Array
+      :type:  blackjax.types.Array
 
-
 
    .. py:attribute:: sample_size
-      :type: int
-
-
+      :type:  int
 
 
 .. py:class:: MassMatrixAdaptationState
 
 
 
-
    State carried through the mass matrix adaptation.
 
    inverse_mass_matrix
@@ -82,15 +75,13 @@ Functions
        The current state of the Welford Algorithm.
 
 
+
    .. py:attribute:: inverse_mass_matrix
-      :type: blackjax.types.Array
+      :type:  blackjax.types.Array
 
-
 
    .. py:attribute:: wc_state
-      :type: WelfordAlgorithmState
-
-
+      :type:  WelfordAlgorithmState
 
 
 .. py:function:: mass_matrix_adaptation(is_diagonal_matrix: bool = True) -> tuple[Callable, Callable, Callable]