Quick & dirty implementation of a generative adversarial network with jax, optax and equinox.
Geek Repo:Geek Repo
Github PK Tool:Github PK Tool
Implementation of GANs in JAX, for learning purposes. The model is not really working yet.