/jax-image-autoencoder

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

Primary LanguageJupyter Notebook

No issues in this repository yet.