Strictly for learning purposes. No guarantee for correctness.
Reconstructing images from their latent space:
Generating new images from a random vector in the latent space:
andrinr/jax-image-autoencoder
Quick dirty implementation of a convolutional image autoencoder with JAX, Flax and optax.
Jupyter Notebook