tanelp / tiny-diffusionLinks
A minimal PyTorch implementation of probabilistic diffusion models for 2D datasets.
☆912Updated last year
Alternatives and similar repositories for tiny-diffusion
Users that are interested in tiny-diffusion are comparing it to the libraries listed below
Sorting:
- Simple and readable code for training and sampling from diffusion models☆511Updated 3 weeks ago
- Self-contained, minimalistic implementation of diffusion models with Pytorch.☆1,074Updated 3 years ago
- Interactive visualizations of the geometric intuition behind diffusion models.☆783Updated 3 weeks ago
- Official implementation of Cold-Diffusion for different transformations in pytorch.☆1,087Updated 2 years ago
- DiffusionFastForward: a free course and experimental framework for diffusion-based generative models☆649Updated 2 years ago
- A series of tutorial notebooks on denoising diffusion probabilistic models in PyTorch☆695Updated 2 years ago
- Tensors, for human consumption☆1,263Updated 3 weeks ago
- Schedule-Free Optimization in PyTorch☆2,189Updated last month
- Elucidating the Design Space of Diffusion-Based Generative Models (EDM)☆1,718Updated last year
- TorchCFM: a Conditional Flow Matching library☆1,825Updated 4 months ago
- 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,980Updated last month
- Creating a diffusion model from scratch in PyTorch to learn exactly how they work.☆369Updated 3 months ago
- "Deep Generative Modeling": Introductory Examples☆1,197Updated 9 months ago
- Code for our NeurIPS 2022 paper☆369Updated 2 years ago
- Implementation of Diffusion Transformer (DiT) in JAX☆279Updated last year
- A minimal yet resourceful implementation of diffusion models (along with pretrained models + synthetic images for nine datasets)☆289Updated 10 months ago
- Minimal implementation of scalable rectified flow transformers, based on SD3's approach☆596Updated last year
- Code for ICML 2023 paper, "PFGM++: Unlocking the Potential of Physics-Inspired Generative Models"☆376Updated last year
- TensorDict is a pytorch dedicated tensor container.☆937Updated this week
- Diffusion Reading Group at EleutherAI☆321Updated last year
- Official Implementation of "ADOPT: Modified Adam Can Converge with Any β2 with the Optimal Rate"☆426Updated 7 months ago
- Code for NeurIPS 2022 Paper, "Poisson Flow Generative Models" (PFGM)☆864Updated 2 years ago
- For optimization algorithm research and development.☆521Updated this week
- A simple way to keep track of an Exponential Moving Average (EMA) version of your Pytorch model☆594Updated 7 months ago
- UNet diffusion model in pure CUDA☆610Updated last year
- Train to 94% on CIFAR-10 in <6.3 seconds on a single A100. Or ~95.79% in ~110 seconds (or less!)☆1,271Updated 6 months ago
- ☆780Updated last month
- Official repository for our work on micro-budget training of large-scale diffusion models.☆1,493Updated 6 months ago
- Python library for designing and training your own Diffusion Models with PyTorch.☆285Updated 3 weeks ago