tanelp / tiny-diffusionLinks
A minimal PyTorch implementation of probabilistic diffusion models for 2D datasets.
☆923Updated 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☆590Updated 2 months ago
- Self-contained, minimalistic implementation of diffusion models with Pytorch.☆1,086Updated 3 years ago
- Interactive visualizations of the geometric intuition behind diffusion models.☆811Updated 2 months ago
- Schedule-Free Optimization in PyTorch☆2,202Updated 3 months ago
- Official implementation of Cold-Diffusion for different transformations in pytorch.☆1,090Updated 2 years ago
- TorchCFM: a Conditional Flow Matching library☆1,915Updated last month
- Tensors, for human consumption☆1,275Updated 2 months ago
- A minimal yet resourceful implementation of diffusion models (along with pretrained models + synthetic images for nine datasets)☆290Updated 11 months ago
- A PyTorch library for implementing flow matching algorithms, featuring continuous and discrete flow matching implementations. It includes…☆3,226Updated 2 months ago
- DiffusionFastForward: a free course and experimental framework for diffusion-based generative models☆659Updated 2 years ago
- Implementation of Diffusion Transformer (DiT) in JAX☆291Updated last year
- Creating a diffusion model from scratch in PyTorch to learn exactly how they work.☆377Updated 4 months ago
- Minimal implementation of scalable rectified flow transformers, based on SD3's approach☆603Updated last year
- A series of tutorial notebooks on denoising diffusion probabilistic models in PyTorch☆696Updated 2 years ago
- Elucidating the Design Space of Diffusion-Based Generative Models (EDM)☆1,750Updated last year
- TensorDict is a pytorch dedicated tensor container.☆955Updated this week
- A simple implimentation of Bayesian Flow Networks (BFN)☆239Updated last year
- Code for NeurIPS 2022 Paper, "Poisson Flow Generative Models" (PFGM)☆867Updated 2 years ago
- Diffusion Reading Group at EleutherAI☆325Updated 2 years ago
- Official repository for our work on micro-budget training of large-scale diffusion models.☆1,507Updated 7 months ago
- Code for our NeurIPS 2022 paper☆369Updated 2 years ago
- Train to 94% on CIFAR-10 in <6.3 seconds on a single A100. Or ~95.79% in ~110 seconds (or less!)☆1,277Updated 8 months ago
- Text to Image Latent Diffusion using a Transformer core☆199Updated 11 months ago
- Official Implementation of "ADOPT: Modified Adam Can Converge with Any β2 with the Optimal Rate"☆428Updated 8 months ago
- A simple way to keep track of an Exponential Moving Average (EMA) version of your Pytorch model☆601Updated 8 months ago
- ☆783Updated 2 months ago
- Code for ICML 2023 paper, "PFGM++: Unlocking the Potential of Physics-Inspired Generative Models"☆375Updated last year
- Universal Tensor Operations in Einstein-Inspired Notation for Python.☆399Updated 4 months ago
- For optimization algorithm research and development.☆530Updated this week
- Annotated version of the Mamba paper☆487Updated last year