Alternatives and similar repositories for DiT-PyTorch:

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

• cloneofsimo / vqgan-training
  Train VAE like a boss
  ☆270Updated 5 months ago
• lucidrains / rectified-flow-pytorch
  Implementation of rectified flow and some of its followup research / improvements in Pytorch
  ☆265Updated last month
• feizc / DiT-MoE
  Scaling Diffusion Transformers with Mixture of Experts
  ☆294Updated 6 months ago
• lucidrains / mmdit
  Implementation of a single layer of the MMDiT, proposed in Stable Diffusion 3, in Pytorch
  ☆331Updated 2 months ago
• apapiu / transformer_latent_diffusion
  Text to Image Latent Diffusion using a Transformer core
  ☆174Updated 6 months ago
• lucidrains / maskbit-pytorch
  Implementation of the proposed MaskBit from Bytedance AI
  ☆75Updated 4 months ago
• cloneofsimo / minRF
  Minimal implementation of scalable rectified flow transformers, based on SD3's approach
  ☆491Updated 8 months ago
• fal-ai / diffusion-speedrun
  Focused on fast experimentation and simplicity
  ☆70Updated 3 months ago
• cloneofsimo / d3pm
  Minimal Implementation of a D3PM in pytorch
  ☆206Updated 11 months ago
• lucidrains / multimodal-dit-pytorch
  Implementation of a multimodal diffusion transformer in Pytorch
  ☆101Updated 9 months ago
• mit-han-lab / hart
  HART: Efficient Visual Generation with Hybrid Autoregressive Transformer
  ☆441Updated 5 months ago
• hkproj / pytorch-ddpm
  Implementation of the paper "Denoising Diffusion Probabilistic Models" in PyTorch
  ☆54Updated last year
• SwayStar123 / microdiffusion
  ☆43Updated last month
• fal-ai-community / nano-mdm
  Tiny re-implementation of MDM in style of LLaDA and nano-gpt speedrun
  ☆44Updated 2 weeks ago
• ai-forever / MoVQGAN
  MoVQGAN - model for the image encoding and reconstruction
  ☆223Updated last year
• zacharyhorvitz / Fk-Diffusion-Steering
  A general framework for inference-time scaling and steering of diffusion models with arbitrary rewards.
  ☆112Updated last month
• swookey-thinky / mindiffusion
  Repository of lessons exploring image diffusion models, focused on understanding and education.
  ☆46Updated 2 months ago
• kuleshov-group / bd3lms
  Block Diffusion: Interpolating Between Autoregressive and Diffusion Language Models
  ☆397Updated this week
• G-U-N / Rectified-Diffusion
  [ICLR 2025] Rectified Diffusion: Straightness Is Not Your Need
  ☆202Updated 2 weeks ago
• lucidrains / autoregressive-diffusion-pytorch
  Implementation of Autoregressive Diffusion in Pytorch
  ☆364Updated 4 months ago
• lucidrains / lumiere-pytorch
  Implementation of Lumiere, SOTA text-to-video generation from Google Deepmind, in Pytorch
  ☆269Updated 8 months ago
• cfifty / rotation_trick
  ☆85Updated this week
• sayakpaul / tt-scale-flux
  Inference-time scaling of diffusion-based image and video generation models.
  ☆117Updated 3 weeks ago
• lucidrains / hyper-connections
  Attempt to make multiple residual streams from Bytedance's Hyper-Connections paper accessible to the public
  ☆78Updated last month
• NVlabs / DiffiT
  [ECCV 2024] Official Repository for DiffiT: Diffusion Vision Transformers for Image Generation
  ☆488Updated 4 months ago
• sihyun-yu / REPA
  [ICLR'25 Oral] Representation Alignment for Generation: Training Diffusion Transformers Is Easier Than You Think
  ☆887Updated last week
• CompVis / fm-boosting
  FMBoost: Boosting Latent Diffusion with Flow Matching (ECCV 2024 Oral)
  ☆223Updated 3 months ago
• leffff / FlowModels
  The aim of this repository is to test and implement Flow-Matching-based models
  ☆80Updated 2 months ago
• kvfrans / shortcut-models
  ☆413Updated 3 months ago
• Anima-Lab / MaskDiT
  Code for Fast Training of Diffusion Models with Masked Transformers
  ☆396Updated 10 months ago