Simple and readable code for training and sampling from diffusion models
☆714Jun 14, 2025Updated 8 months ago
Alternatives and similar repositories for smalldiffusion
Users that are interested in smalldiffusion are comparing it to the libraries listed below
Sorting:
- Minimal implementation of scalable rectified flow transformers, based on SD3's approach☆635Jul 1, 2024Updated last year
- Official repository for our work on micro-budget training of large-scale diffusion models.☆1,549Jan 12, 2025Updated last year
- A PyTorch library for implementing flow matching algorithms, featuring continuous and discrete flow matching implementations. It includes…☆4,189Jan 5, 2026Updated 2 months ago
- Supercharge huggingface transformers with model parallelism.☆78Jul 23, 2025Updated 7 months ago
- ConceptAttention: A method for interpreting multi-modal diffusion transformers.☆425Jan 16, 2026Updated last month
- Minimal Implementation of a D3PM in pytorch☆287Apr 22, 2024Updated last year
- A transformer-based multimodal model for music.☆29Aug 15, 2024Updated last year
- Minimalistic 4D-parallelism distributed training framework for education purpose☆2,099Aug 26, 2025Updated 6 months ago
- TorchCFM: a Conditional Flow Matching library☆2,331Nov 11, 2025Updated 3 months ago
- ☆48Feb 23, 2025Updated last year
- Single File, Single GPU, From Scratch, Efficient, Full Parameter Tuning library for "RL for LLMs"☆599Oct 7, 2025Updated 5 months ago
- LeanRL is a fork of CleanRL, where selected PyTorch scripts optimized for performance using compile and cudagraphs.☆677Aug 22, 2025Updated 6 months ago
- An interactive exploration of Transformer programming.☆274Nov 15, 2023Updated 2 years ago
- Implementation of Diffusion Transformer (DiT) in JAX☆308Jun 11, 2024Updated last year
- Interactive visualizations of the geometric intuition behind diffusion models.☆1,073Jan 31, 2026Updated last month
- [ACM MM24] Official implementation of ACM MM 2024 paper: "ZePo: Zero-Shot Portrait Stylization with Faster Sampling"☆44Aug 22, 2024Updated last year
- Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.☆32Jun 5, 2025Updated 9 months ago
- NanoGPT (124M) in 2 minutes☆4,734Feb 27, 2026Updated last week
- Train VAE like a boss☆313Oct 21, 2024Updated last year
- FlexTok: Resampling Images into 1D Token Sequences of Flexible Length☆300Jun 2, 2025Updated 9 months ago
- Focused on fast experimentation and simplicity☆80Dec 24, 2024Updated last year
- ☆292Jul 15, 2024Updated last year
- ☆239Nov 24, 2025Updated 3 months ago
- My take on Flow Matching☆92Jan 11, 2025Updated last year
- Huggingface-compatible SDXL Unet implementation that is readily hackable☆435Aug 9, 2023Updated 2 years ago
- 🚀 Efficient implementations of state-of-the-art linear attention models☆4,474Updated this week
- Schedule-Free Optimization in PyTorch☆2,262May 21, 2025Updated 9 months ago
- Track and Collaborate on ML & AI Experiments.☆44Mar 10, 2025Updated 11 months ago
- UNet diffusion model in pure CUDA☆657Jun 28, 2024Updated last year
- Rust implementation of Surya☆65Mar 1, 2025Updated last year
- EDM2 and Autoguidance -- Official PyTorch implementation☆824Dec 9, 2024Updated last year
- Compositional Linear Algebra☆508Aug 1, 2025Updated 7 months ago
- T-GATE: Temporally Gating Attention to Accelerate Diffusion Model for Free!☆415Feb 26, 2025Updated last year
- A minimal PyTorch implementation of probabilistic diffusion models for 2D datasets.☆987May 7, 2024Updated last year
- HunyuanVideo: A Systematic Framework For Large Video Generation Model☆48Dec 14, 2024Updated last year
- A JAX research toolkit for building, editing, and visualizing neural networks.☆1,869Jun 22, 2025Updated 8 months ago
- A streamlined implementation of Grounding DINO and SAM for advanced image segmentation. This lightweight solution simplifies the integrat…☆67Sep 30, 2024Updated last year
- Diffusion models in PyTorch☆128Feb 2, 2026Updated last month
- The simplest, fastest repository for training/finetuning small-sized VLMs.☆4,685Oct 27, 2025Updated 4 months ago