Implementation of a single layer of the MMDiT, proposed in Stable Diffusion 3, in Pytorch
☆540Jan 18, 2026Updated 5 months ago
Alternatives and similar repositories for mmdit
Users that are interested in mmdit are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Implementation of rectified flow and some of its followup research / improvements in Pytorch☆468Jun 8, 2026Updated last week
- Implementation of Autoregressive Diffusion in Pytorch☆437Dec 4, 2025Updated 6 months ago
- Official PyTorch Implementation of "SiT: Exploring Flow and Diffusion-based Generative Models with Scalable Interpolant Transformers"☆1,171Dec 22, 2025Updated 5 months ago
- Pytorch implementation of Transfusion, "Predict the Next Token and Diffuse Images with One Multi-Modal Model", from MetaAI☆1,371Jan 27, 2026Updated 4 months ago
- Official PyTorch Implementation of "Scalable Diffusion Models with Transformers"☆8,628May 31, 2024Updated 2 years ago
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- [ICLR'25 Oral] Representation Alignment for Generation: Training Diffusion Transformers Is Easier Than You Think☆1,650Mar 16, 2025Updated last year
- A PyTorch library for implementing flow matching algorithms, featuring continuous and discrete flow matching implementations. It includes…☆4,542Jan 5, 2026Updated 5 months ago
- Implementation of a multimodal diffusion transformer in Pytorch☆108Jun 24, 2024Updated last year
- Minimal implementation of scalable rectified flow transformers, based on SD3's approach☆635Jul 1, 2024Updated last year
- Autoregressive Model Beats Diffusion: 🦙 Llama for Scalable Image Generation☆1,955Aug 15, 2024Updated last year
- Consistency Models Made Easy☆330Oct 13, 2024Updated last year
- EDM2 and Autoguidance -- Official PyTorch implementation☆845Dec 9, 2024Updated last year
- Scaling Diffusion Transformers with Mixture of Experts☆426Sep 9, 2024Updated last year
- A PyTorch implementation of the paper "All are Worth Words: A ViT Backbone for Diffusion Models".☆1,103Mar 25, 2023Updated 3 years ago
- GPUs on demand by Runpod - Special Offer Available • AdRun AI, ML, and HPC workloads on powerful cloud GPUs—without limits or wasted spend. Deploy GPUs in under a minute and pay by the second.
- Open(MM)DiT: An Easy, Fast and Memory-Efficient System for (MM)DiT Training and Inference☆46Mar 13, 2024Updated 2 years ago
- code for "Diffusion Forcing: Next-token Prediction Meets Full-Sequence Diffusion"☆1,260Nov 9, 2025Updated 7 months ago
- PyTorch implementation of MAR+DiffLoss https://arxiv.org/abs/2406.11838☆1,935Feb 20, 2026Updated 3 months ago
- [ECCV 2024, Oral] FMBoost: Boosting Latent Diffusion with Flow Matching☆259Oct 17, 2025Updated 8 months ago
- PixArt-α: Fast Training of Diffusion Transformer for Photorealistic Text-to-Image Synthesis☆3,300Oct 31, 2024Updated last year
- Implementation of TiTok, proposed by Bytedance in "An Image is Worth 32 Tokens for Reconstruction and Generation"☆184Jun 20, 2024Updated last year
- (NeurIPS 2024 Oral 🔥) Improved Distribution Matching Distillation for Fast Image Synthesis☆1,374Mar 5, 2025Updated last year
- Official Implementation of Rectified Flow (ICLR2023 Spotlight)☆1,612Jul 20, 2024Updated last year
- Lumina-T2X is a unified framework for Text to Any Modality Generation☆2,251Feb 16, 2025Updated last year
- End-to-end encrypted cloud storage - Proton Drive • AdSpecial offer: 40% Off Yearly / 80% Off First Month. Protect your most important files, photos, and documents from prying eyes.
- Implementation of Infini-Transformer in Pytorch☆112Jan 4, 2025Updated last year
- Code for Fast Training of Diffusion Models with Masked Transformers☆427May 15, 2024Updated 2 years ago
- Official implementation of Inductive Moment Matching☆585Jul 11, 2025Updated 11 months ago
- Implementation of MagViT2 Tokenizer in Pytorch☆665Jan 12, 2025Updated last year
- A simple way to keep track of an Exponential Moving Average (EMA) version of your Pytorch model☆654Dec 19, 2025Updated 5 months ago
- Masked Diffusion Transformer is the SOTA for image synthesis. (ICCV 2023)☆597Apr 23, 2024Updated 2 years ago
- Exploration into the proposed "Self Reasoning Tokens" by Felipe Bonetto☆57May 17, 2024Updated 2 years ago
- PeRFlow: Piecewise Rectified Flow as Universal Plug-and-Play Accelerator (NeurIPS 2024)☆538Sep 8, 2025Updated 9 months ago
- [NeurIPS 2025] An official implementation of Flow-GRPO: Training Flow Matching Models via Online RL☆2,338May 7, 2026Updated last month
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- InstaFlow! One-Step Stable Diffusion with Rectified Flow (ICLR 2024)☆1,410Jun 7, 2024Updated 2 years ago
- VideoSys: An easy and efficient system for video generation☆2,026Aug 27, 2025Updated 9 months ago
- TorchCFM: a Conditional Flow Matching library☆2,498Apr 20, 2026Updated last month
- Implementation of E2-TTS, "Embarrassingly Easy Fully Non-Autoregressive Zero-Shot TTS", in Pytorch☆517Dec 20, 2025Updated 5 months ago
- [TMLR 2025] Latte: Latent Diffusion Transformer for Video Generation.☆1,943Oct 30, 2025Updated 7 months ago
- A suite of image and video neural tokenizers☆1,725Feb 11, 2025Updated last year
- A curated list of recent diffusion models for video generation, editing, and various other applications.☆5,691Updated this week