/jax-vae

simple vae built in jax and flax

Watchers