andrinr / jax-image-autoencoder

Quick dirty implementation of a convolutional image autoencoder with JAX, Flax and optax.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

jax-image-autoencoder

Strictly for learning purposes. No guarantee for correctness.

Reconstructing images from their latent space: reconstruction

Generating new images from a random vector in the latent space: generation

About

Quick dirty implementation of a convolutional image autoencoder with JAX, Flax and optax.


Languages

Language:Jupyter Notebook 100.0%