andrinr / jax-gan

Quick & dirty implementation of a generative adversarial network with jax, optax and equinox.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

jax-gan

Implementation of GANs in JAX, for learning purposes. The model is not really working yet.

About

Quick & dirty implementation of a generative adversarial network with jax, optax and equinox.


Languages

Language:Jupyter Notebook 100.0%