Expose contrib.funsor inference algorithms in docs
Funsor-based Pyro
==========================
+
+Primitives
+----------
+
.. automodule:: pyro.contrib.funsor
:members:
:undoc-members:
@@ -33,3 +37,31 @@ Effect handlers
:undoc-members:
:show-inheritance:
:member-order: bysource
+
+Inference algorithms
+--------------------
+
+.. automodule:: pyro.contrib.funsor.infer.elbo
+ :members:
+ :undoc-members:
+ :show-inheritance:
+ :member-order: bysource
+
+.. automodule:: pyro.contrib.funsor.infer.trace_elbo
+ :members:
+ :undoc-members:
+ :show-inheritance:
+ :member-order: bysource
+
+.. automodule:: pyro.contrib.funsor.infer.traceenum_elbo
+ :members:
+ :undoc-members:
+ :show-inheritance:
+ :member-order: bysource
+
+.. automodule:: pyro.contrib.funsor.infer.tracetmc_elbo
+ :members:
+ :undoc-members:
+ :show-inheritance:
+ :member-order: bysource
+
