Implementation of a single layer of the MMDiT, proposed in Stable Diffusion 3, in Pytorch
☆523Jan 18, 2026Updated 3 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☆445Apr 28, 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,160Dec 22, 2025Updated 4 months ago
- Pytorch implementation of Transfusion, "Predict the Next Token and Diffuse Images with One Multi-Modal Model", from MetaAI☆1,356Jan 27, 2026Updated 3 months ago
- Official PyTorch Implementation of "Scalable Diffusion Models with Transformers"☆8,545May 31, 2024Updated last year
- Serverless GPU API endpoints on Runpod - Get Bonus Credits • AdSkip the infrastructure headaches. Auto-scaling, pay-as-you-go, no-ops approach lets you focus on innovating your application.
- [ICLR'25 Oral] Representation Alignment for Generation: Training Diffusion Transformers Is Easier Than You Think☆1,621Mar 16, 2025Updated last year
- A PyTorch library for implementing flow matching algorithms, featuring continuous and discrete flow matching implementations. It includes…☆4,387Jan 5, 2026Updated 4 months ago
- Implementation of a multimodal diffusion transformer in Pytorch☆107Jun 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,947Aug 15, 2024Updated last year
- Consistency Models Made Easy☆330Oct 13, 2024Updated last year
- EDM2 and Autoguidance -- Official PyTorch implementation☆842Dec 9, 2024Updated last year
- Scaling Diffusion Transformers with Mixture of Experts☆420Sep 9, 2024Updated last year
- A PyTorch implementation of the paper "All are Worth Words: A ViT Backbone for Diffusion Models".☆1,107Mar 25, 2023Updated 3 years ago
- Bare Metal GPUs on DigitalOcean Gradient AI • AdPurpose-built for serious AI teams training foundational models, running large-scale inference, and pushing the boundaries of what's possible.
- 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,236Nov 9, 2025Updated 6 months ago
- PyTorch implementation of MAR+DiffLoss https://arxiv.org/abs/2406.11838☆1,905Feb 20, 2026Updated 2 months ago
- [ECCV 2024, Oral] FMBoost: Boosting Latent Diffusion with Flow Matching☆258Oct 17, 2025Updated 6 months ago
- 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,316Mar 5, 2025Updated last year
- PixArt-α: Fast Training of Diffusion Transformer for Photorealistic Text-to-Image Synthesis☆3,298Oct 31, 2024Updated last year
- Official Implementation of Rectified Flow (ICLR2023 Spotlight)☆1,597Jul 20, 2024Updated last year
- Lumina-T2X is a unified framework for Text to Any Modality Generation☆2,253Feb 16, 2025Updated last year
- Wordpress hosting with auto-scaling - Free Trial Offer • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- Implementation of Infini-Transformer in Pytorch☆112Jan 4, 2025Updated last year
- Code for Fast Training of Diffusion Models with Masked Transformers☆424May 15, 2024Updated last year
- Official implementation of Inductive Moment Matching☆584Jul 11, 2025Updated 9 months ago
- Implementation of MagViT2 Tokenizer in Pytorch☆660Jan 12, 2025Updated last year
- A simple way to keep track of an Exponential Moving Average (EMA) version of your Pytorch model☆650Dec 19, 2025Updated 4 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 last year
- [NeurIPS 2025] An official implementation of Flow-GRPO: Training Flow Matching Models via Online RL☆2,245Updated this week
- PeRFlow: Piecewise Rectified Flow as Universal Plug-and-Play Accelerator (NeurIPS 2024)☆536Sep 8, 2025Updated 8 months ago
- AI Agents on DigitalOcean Gradient AI Platform • AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- InstaFlow! One-Step Stable Diffusion with Rectified Flow (ICLR 2024)☆1,400Jun 7, 2024Updated last year
- VideoSys: An easy and efficient system for video generation☆2,021Aug 27, 2025Updated 8 months ago
- TorchCFM: a Conditional Flow Matching library☆2,440Apr 20, 2026Updated 2 weeks ago
- Implementation of E2-TTS, "Embarrassingly Easy Fully Non-Autoregressive Zero-Shot TTS", in Pytorch☆516Dec 20, 2025Updated 4 months ago
- [TMLR 2025] Latte: Latent Diffusion Transformer for Video Generation.☆1,935Oct 30, 2025Updated 6 months ago
- A suite of image and video neural tokenizers☆1,723Feb 11, 2025Updated last year
- A curated list of recent diffusion models for video generation, editing, and various other applications.☆5,620Apr 3, 2026Updated last month