Minimal JAX implementations of Autoencoder, VAE, GAN, Flow, Diffusion, and Transformer models. Built from scratch for speed and clarity.
adham-elarabawy / jaxgen Goto Github PK
View Code? Open in Web Editor NEWMinimal JAX implementations of Autoencoder, VAE, GAN, Flow, Diffusion, and Transformer models. Built from scratch for speed and clarity.
License: MIT License