yuanchenyang / smalldiffusionView external linksLinks
Simple and readable code for training and sampling from diffusion models
☆703Jun 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☆632Jul 1, 2024Updated last year
- Official repository for our work on micro-budget training of large-scale diffusion models.☆1,550Jan 12, 2025Updated last year
- A PyTorch library for implementing flow matching algorithms, featuring continuous and discrete flow matching implementations. It includes…☆4,134Jan 5, 2026Updated last month
- Supercharge huggingface transformers with model parallelism.☆78Jul 23, 2025Updated 6 months ago
- ConceptAttention: A method for interpreting multi-modal diffusion transformers.☆418Jan 16, 2026Updated last month
- Minimal Implementation of a D3PM in pytorch☆286Apr 22, 2024Updated last year
- A transformer-based multimodal model for music.☆29Aug 15, 2024Updated last year
- TorchCFM: a Conditional Flow Matching library☆2,294Nov 11, 2025Updated 3 months ago
- ☆48Feb 23, 2025Updated 11 months ago
- Minimalistic 4D-parallelism distributed training framework for education purpose☆2,076Aug 26, 2025Updated 5 months ago
- An interactive exploration of Transformer programming.☆271Nov 15, 2023Updated 2 years ago
- Implementation of Diffusion Transformer (DiT) in JAX☆304Jun 11, 2024Updated last year
- Interactive visualizations of the geometric intuition behind diffusion models.☆1,064Jan 31, 2026Updated 2 weeks ago
- [ACM MM24] Official implementation of ACM MM 2024 paper: "ZePo: Zero-Shot Portrait Stylization with Faster Sampling"☆43Aug 22, 2024Updated last year
- Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.☆32Jun 5, 2025Updated 8 months ago
- Single File, Single GPU, From Scratch, Efficient, Full Parameter Tuning library for "RL for LLMs"☆595Oct 7, 2025Updated 4 months ago
- LeanRL is a fork of CleanRL, where selected PyTorch scripts optimized for performance using compile and cudagraphs.☆671Aug 22, 2025Updated 5 months ago
- NanoGPT (124M) in 2 minutes☆4,624Updated this week
- FlexTok: Resampling Images into 1D Token Sequences of Flexible Length☆292Jun 2, 2025Updated 8 months ago
- Train VAE like a boss☆313Oct 21, 2024Updated last year
- ☆234Nov 24, 2025Updated 2 months ago
- Focused on fast experimentation and simplicity☆80Dec 24, 2024Updated last year
- ☆292Jul 15, 2024Updated last year
- My take on Flow Matching☆91Jan 11, 2025Updated last year
- 🚀 Efficient implementations of state-of-the-art linear attention models☆4,379Updated this week
- Huggingface-compatible SDXL Unet implementation that is readily hackable☆435Aug 9, 2023Updated 2 years ago
- Schedule-Free Optimization in PyTorch☆2,256May 21, 2025Updated 8 months ago
- Track and Collaborate on ML & AI Experiments.☆44Mar 10, 2025Updated 11 months ago
- EDM2 and Autoguidance -- Official PyTorch implementation☆819Dec 9, 2024Updated last year
- Rust implementation of Surya☆65Mar 1, 2025Updated 11 months ago
- Compositional Linear Algebra☆507Aug 1, 2025Updated 6 months ago
- T-GATE: Temporally Gating Attention to Accelerate Diffusion Model for Free!☆415Feb 26, 2025Updated 11 months ago
- 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,863Jun 22, 2025Updated 7 months ago
- A streamlined implementation of Grounding DINO and SAM for advanced image segmentation. This lightweight solution simplifies the integrat…☆66Sep 30, 2024Updated last year
- Diffusion models in PyTorch☆128Feb 2, 2026Updated 2 weeks ago
- An interactive HTML pretty-printer for machine learning research in IPython notebooks.☆458Aug 8, 2025Updated 6 months ago
- UNet diffusion model in pure CUDA☆660Jun 28, 2024Updated last year
- ☆30Oct 7, 2024Updated last year