ChEES-HMC

blackjax-devs · Oct 30, 2023 · 43ed564 · 43ed564
1 parent 29dc2eb
commit 43ed564
Show file tree

Hide file tree

Showing 5 changed files with 490 additions and 5 deletions.
diff --git a/blackjax/__init__.py b/blackjax/__init__.py
@@ -1,5 +1,6 @@
 from blackjax._version import __version__
 
+from .adaptation.chees_adaptation import chees_adaptation
 from .adaptation.meads_adaptation import meads_adaptation
 from .adaptation.pathfinder_adaptation import pathfinder_adaptation
 from .adaptation.window_adaptation import window_adaptation
@@ -44,6 +45,7 @@
     "csgld",
     "window_adaptation",  # mcmc adaptation
     "meads_adaptation",
+    "chees_adaptation",
     "pathfinder_adaptation",
     "adaptive_tempered_smc",  # smc
     "tempered_smc",

diff --git a/blackjax/adaptation/__init__.py b/blackjax/adaptation/__init__.py
@@ -1,3 +1,13 @@
-from . import meads_adaptation, pathfinder_adaptation, window_adaptation
+from . import (
+    chees_adaptation,
+    meads_adaptation,
+    pathfinder_adaptation,
+    window_adaptation,
+)
 
-__all__ = ["meads_adaptation", "window_adaptation", "pathfinder_adaptation"]
+__all__ = [
+    "chees_adaptation",
+    "meads_adaptation",
+    "window_adaptation",
+    "pathfinder_adaptation",
+]