Simple VAE implementation in tensorflow, jax and pytorch where both the encoder and decoder model use gaussian distributions.
-
Updated
Mar 4, 2024 - Jupyter Notebook
#
vae-jax
Here is 1 public repository matching this topic...
Add this topic to your repo
To associate your repository with the vae-jax topic, visit your repo's landing page and select "manage topics."