Quick & dirty implementation of a generative adversarial network with jax, optax and equinox.
Primary LanguageJupyter Notebook
Implementation of GANs in JAX, for learning purposes. The model is not really working yet.