smsharma/minified-generative-models

Bare-bones implementations of some generative models in Jax: diffusion, normalizing flows, consistency models, flow matching, (beta)-VAEs, etc

Jupyter NotebookMIT