/VAE-MNIST-Haiku-Jax

Simple VAE example with Jax.

Primary LanguageJupyter NotebookMIT LicenseMIT

Stargazers