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