smsharma / minified-generative-models

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

Related projects

Alternatives and complementary repositories for minified-generative-models