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...
Improve this page
Add a description, image, and links to the vae-jax topic page so that developers can more easily learn about it.
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."