helblazer811 / DiffusionLab
Interactive visualizations of the theory behind diffusion models.
☆373Updated this week
Alternatives and similar repositories for DiffusionLab
Users that are interested in DiffusionLab are comparing it to the libraries listed below
Sorting:
- Simple and readable code for training and sampling from diffusion models☆493Updated 4 months ago
- Implementation of Diffusion Transformer (DiT) in JAX☆275Updated 11 months ago
- Annotated Flow Matching paper☆183Updated 8 months ago
- Flow-matching algorithms in JAX☆90Updated 9 months ago
- A minimal PyTorch implementation of probabilistic diffusion models for 2D datasets.☆734Updated last year
- Minimal Implementation of a D3PM in pytorch☆222Updated last year
- A simple implimentation of Bayesian Flow Networks (BFN)☆240Updated last year
- A PyTorch library for implementing flow matching algorithms, featuring continuous and discrete flow matching implementations. It includes…☆2,580Updated 4 months ago
- Official implementation of Inductive Moment Matching☆465Updated 2 months ago
- Train VAE like a boss☆278Updated 6 months ago
- Official Implementation of "ADOPT: Modified Adam Can Converge with Any β2 with the Optimal Rate"☆424Updated 5 months ago
- The boundary of neural network trainability is fractal☆198Updated last year
- ☆150Updated 9 months ago
- Minimal implementation of scalable rectified flow transformers, based on SD3's approach☆528Updated 10 months ago
- A curated list for awesome discrete diffusion models resources.☆313Updated 3 months ago
- Getting crystal-like representations with harmonic loss☆183Updated last month
- CIFAR-10 speedruns: 94% in 2.6 seconds and 96% in 27 seconds☆233Updated 2 months ago
- A mini-library for training consistency models.☆243Updated last year
- From-scratch diffusion model implemented in PyTorch.☆87Updated last year
- code for "Diffusion Forcing: Next-token Prediction Meets Full-Sequence Diffusion"☆855Updated last month
- This repo contains the code for the paper "Intuitive physics understanding emerges fromself-supervised pretraining on natural videos"☆154Updated 2 months ago
- code for "Riemannian Flow Matching on General Geometries".☆224Updated last year
- ☆36Updated 5 months ago
- A Tutorial for Diffusion Models☆51Updated last year
- My take on Flow Matching☆55Updated 4 months ago
- For optimization algorithm research and development.☆513Updated this week
- Universal Tensor Operations in Einstein-Inspired Notation for Python.☆369Updated last month
- EDM2 and Autoguidance -- Official PyTorch implementation☆706Updated 5 months ago
- Bare-bones implementations of some generative models in Jax: diffusion, normalizing flows, consistency models, flow matching, (beta)-VAEs…☆129Updated last year
- ☆182Updated 5 months ago