Porting the GMM tutorial (#1562)

* GMM tutorial * hook_optax * import jax.pure_callback * address comments * remove optax_to_numpyro * remove sample_subs dict
pyro-ppl · Mar 27, 2023 · a7267d9 · a7267d9
1 parent fa5ce64
commit a7267d9
Show file tree

Hide file tree

Showing 3 changed files with 808 additions and 16 deletions.
diff --git a/docs/source/index.rst b/docs/source/index.rst
@@ -41,6 +41,7 @@ NumPyro documentation
    :caption: Discrete Latent Variables
    :name: discrete-latent-variables
 
+   tutorials/gmm
    examples/annotation
    examples/hmm_enum
    examples/capture_recapture