Alternatives and similar repositories for DiT-PyTorch

Users that are interested in DiT-PyTorch are comparing it to the libraries listed below

• apapiu / transformer_latent_diffusion
  Text to Image Latent Diffusion using a Transformer core
  ☆192Updated 10 months ago
• cloneofsimo / minRF
  Minimal implementation of scalable rectified flow transformers, based on SD3's approach
  ☆599Updated last year
• lucidrains / mmdit
  Implementation of a single layer of the MMDiT, proposed in Stable Diffusion 3, in Pytorch
  ☆383Updated 6 months ago
• lumalabs / imm
  Official implementation of Inductive Moment Matching
  ☆515Updated last week
• hkproj / pytorch-ddpm
  Implementation of the paper "Denoising Diffusion Probabilistic Models" in PyTorch
  ☆65Updated 2 years ago
• lucidrains / rectified-flow-pytorch
  Implementation of rectified flow and some of its followup research / improvements in Pytorch
  ☆326Updated last month
• feizc / DiT-MoE
  Scaling Diffusion Transformers with Mixture of Experts
  ☆347Updated 10 months ago
• mit-han-lab / hart
  HART: Efficient Visual Generation with Hybrid Autoregressive Transformer
  ☆609Updated 9 months ago
• cloneofsimo / vqgan-training
  Train VAE like a boss
  ☆283Updated 8 months ago
• sihyun-yu / REPA
  [ICLR'25 Oral] Representation Alignment for Generation: Training Diffusion Transformers Is Easier Than You Think
  ☆1,194Updated 4 months ago
• kvfrans / shortcut-models
  ☆572Updated 7 months ago
• leffff / FlowModels
  The aim of this repository is to test and implement Flow-Matching-based models
  ☆99Updated 6 months ago
• explainingai-code / StableDiffusion-PyTorch
  This repo implements a Stable Diffusion model in PyTorch with all the essential components.
  ☆216Updated 7 months ago
• yifan123 / flow_grpo
  An official implementation of Flow-GRPO: Training Flow Matching Models via Online RL
  ☆884Updated this week
• zju-pi / diff-sampler
  An open-source toolbox for fast sampling of diffusion models. Official implementations of our works published in ICML, NeurIPS, CVPR.
  ☆303Updated 4 months ago
• sony / ctm
  ☆294Updated 9 months ago
• NVlabs / DiffiT
  [ECCV 2024] Official Repository for DiffiT: Diffusion Vision Transformers for Image Generation
  ☆495Updated 8 months ago
• lucidrains / autoregressive-diffusion-pytorch
  Implementation of Autoregressive Diffusion in Pytorch
  ☆393Updated 8 months ago
• cloneofsimo / d3pm
  Minimal Implementation of a D3PM in pytorch
  ☆239Updated last year
• lucidrains / lumiere-pytorch
  Implementation of Lumiere, SOTA text-to-video generation from Google Deepmind, in Pytorch
  ☆278Updated 11 months ago
• NVlabs / edm2
  EDM2 and Autoguidance -- Official PyTorch implementation
  ☆743Updated 7 months ago
• G-U-N / Rectified-Diffusion
  [ICLR 2025] Rectified Diffusion: Straightness Is Not Your Need
  ☆234Updated 4 months ago
• gojasper / flash-diffusion
  ⚡ Flash Diffusion ⚡: Accelerating Any Conditional Diffusion Model for Few Steps Image Generation (AAAI 2025 Oral)
  ☆603Updated 4 months ago
• dome272 / Flow-Matching
  My take on Flow Matching
  ☆66Updated 6 months ago
• G-U-N / Awesome-Consistency-Models
  Awesome List of Consistency Models
  ☆111Updated 2 months ago
• gmongaras / Diffusion_models_from_scratch
  Creating a diffusion model from scratch in PyTorch to learn exactly how they work.
  ☆370Updated 3 months ago
• kvablack / ddpo-pytorch
  DDPO for finetuning diffusion models, implemented in PyTorch with LoRA support
  ☆626Updated last year
• MCG-NJU / DDT
  DDT: Decoupled Diffusion Transformer
  ☆264Updated 2 weeks ago
• magic-research / piecewise-rectified-flow
  PeRFlow: Piecewise Rectified Flow as Universal Plug-and-Play Accelerator (NeurIPS 2024)
  ☆520Updated last year
• xandergos / sCM-mnist
  Unofficial implementation of "Simplifying, Stabilizing & Scaling Continuous-Time Consistency Models" for MNIST
  ☆67Updated 3 months ago