GitHub repository: smsharma/minified-generative-models
Primary language: Jupyter Notebook
Description: Bare-bones implementations of some generative models in Jax: diffusion, normalizing flows, consistency models, flow matching, (beta)-VAEs.