Implementation of a single layer of the MMDiT, proposed in Stable Diffusion 3, in Pytorch
☆536Jan 18, 2026Updated 4 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☆459May 19, 2026Updated last week
- Implementation of Autoregressive Diffusion in Pytorch☆437Dec 4, 2025Updated 5 months ago
- Official PyTorch Implementation of "SiT: Exploring Flow and Diffusion-based Generative Models with Scalable Interpolant Transformers"☆1,166Dec 22, 2025Updated 5 months ago
- Pytorch implementation of Transfusion, "Predict the Next Token and Diffuse Images with One Multi-Modal Model", from MetaAI☆1,361Jan 27, 2026Updated 4 months ago
- Official PyTorch Implementation of "Scalable Diffusion Models with Transformers"☆8,594May 31, 2024Updated last year
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- [ICLR'25 Oral] Representation Alignment for Generation: Training Diffusion Transformers Is Easier Than You Think☆1,635Mar 16, 2025Updated last year
- A PyTorch library for implementing flow matching algorithms, featuring continuous and discrete flow matching implementations. It includes…☆4,457Jan 5, 2026Updated 4 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☆634Jul 1, 2024Updated last year
- Autoregressive Model Beats Diffusion: 🦙 Llama for Scalable Image Generation☆1,951Aug 15, 2024Updated last year
- Consistency Models Made Easy☆330Oct 13, 2024Updated last year
- EDM2 and Autoguidance -- Official PyTorch implementation☆844Dec 9, 2024Updated last year
- Scaling Diffusion Transformers with Mixture of Experts☆422Sep 9, 2024Updated last year
- A PyTorch implementation of the paper "All are Worth Words: A ViT Backbone for Diffusion Models".☆1,106Mar 25, 2023Updated 3 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.
- 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,248Nov 9, 2025Updated 6 months ago
- PyTorch implementation of MAR+DiffLoss https://arxiv.org/abs/2406.11838☆1,928Feb 20, 2026Updated 3 months ago
- [ECCV 2024, Oral] FMBoost: Boosting Latent Diffusion with Flow Matching☆259Oct 17, 2025Updated 7 months ago
- PixArt-α: Fast Training of Diffusion Transformer for Photorealistic Text-to-Image Synthesis☆3,299Oct 31, 2024Updated last year
- Implementation of TiTok, proposed by Bytedance in "An Image is Worth 32 Tokens for Reconstruction and Generation"☆182Jun 20, 2024Updated last year
- (NeurIPS 2024 Oral 🔥) Improved Distribution Matching Distillation for Fast Image Synthesis☆1,353Mar 5, 2025Updated last year
- Official Implementation of Rectified Flow (ICLR2023 Spotlight)☆1,604Jul 20, 2024Updated last year
- Lumina-T2X is a unified framework for Text to Any Modality Generation☆2,252Feb 16, 2025Updated last year
- GPU virtual machines on DigitalOcean Gradient AI • AdGet to production fast with high-performance AMD and NVIDIA GPUs you can spin up in seconds. The definition of operational simplicity.
- 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 10 months ago
- Implementation of MagViT2 Tokenizer in Pytorch☆662Jan 12, 2025Updated last year
- A simple way to keep track of an Exponential Moving Average (EMA) version of your Pytorch model☆650Dec 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
- [NeurIPS 2025] An official implementation of Flow-GRPO: Training Flow Matching Models via Online RL☆2,287May 7, 2026Updated 3 weeks ago
- PeRFlow: Piecewise Rectified Flow as Universal Plug-and-Play Accelerator (NeurIPS 2024)☆538Sep 8, 2025Updated 8 months ago
- 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,405Jun 7, 2024Updated last year
- VideoSys: An easy and efficient system for video generation☆2,023Aug 27, 2025Updated 9 months ago
- TorchCFM: a Conditional Flow Matching library☆2,474Apr 20, 2026Updated last month
- Implementation of E2-TTS, "Embarrassingly Easy Fully Non-Autoregressive Zero-Shot TTS", in Pytorch☆516Dec 20, 2025Updated 5 months ago
- [TMLR 2025] Latte: Latent Diffusion Transformer for Video Generation.☆1,938Oct 30, 2025Updated 6 months ago
- A suite of image and video neural tokenizers☆1,726Feb 11, 2025Updated last year
- A curated list of recent diffusion models for video generation, editing, and various other applications.☆5,663May 8, 2026Updated 3 weeks ago