/jax-vae

simple vae built in jax and flax

jax-vae

simple vae built in jax and flax