Add Mean-Field Variational Inference

xidulu · rlouf · commit 4dce7fe0b633 · 2023-01-12T11:31:48.000+01:00
diff --git a/docs/vi.rst b/docs/vi.rst
@@ -7,9 +7,15 @@ Variational Inference
   :nosignatures:
 
   pathfinder
+  meanfield_vi
 
 
 Pathfinder
 ~~~~~~~~~~
 
 .. autoclass:: blackjax.pathfinder
+
+Mean-field VI
+~~~~~~~~~~~~~
+
+.. autoclass:: blackjax.meanfield_vi
