Alternatives and similar repositories for shortcut-models

Users that are interested in shortcut-models are comparing it to the libraries listed below

• sihyun-yu / REPA
  [ICLR'25 Oral] Representation Alignment for Generation: Training Diffusion Transformers Is Easier Than You Think
  ☆1,126Updated 3 months ago
• buoyancy99 / diffusion-forcing
  code for "Diffusion Forcing: Next-token Prediction Meets Full-Sequence Diffusion"
  ☆892Updated 2 months ago
• lucidrains / rectified-flow-pytorch
  Implementation of rectified flow and some of its followup research / improvements in Pytorch
  ☆309Updated this week
• lucidrains / autoregressive-diffusion-pytorch
  Implementation of Autoregressive Diffusion in Pytorch
  ☆390Updated 7 months ago
• NVlabs / edm2
  EDM2 and Autoguidance -- Official PyTorch implementation
  ☆729Updated 6 months ago
• dongzhuoyao / awesome-flow-matching
  A summary of related works about flow matching, stochastic interpolants
  ☆482Updated 2 months ago
• yifan123 / flow_grpo
  An official implementation of Flow-GRPO: Training Flow Matching Models via Online RL
  ☆752Updated this week
• haidog-yaqub / MeanFlow
  Pytorch Implementation (unofficial) of the paper "Mean Flows for One-step Generative Modeling" by Geng et al.
  ☆419Updated last week
• locuslab / ect
  Consistency Models Made Easy
  ☆288Updated 8 months ago
• willisma / SiT
  Official PyTorch Implementation of "SiT: Exploring Flow and Diffusion-based Generative Models with Scalable Interpolant Transformers"
  ☆873Updated last year
• lucidrains / mmdit
  Implementation of a single layer of the MMDiT, proposed in Stable Diffusion 3, in Pytorch
  ☆369Updated 5 months ago
• sony / ctm
  ☆291Updated 8 months ago
• YangLing0818 / consistency_flow_matching
  Official Implementation for "Consistency Flow Matching: Defining Straight Flows with Velocity Consistency"
  ☆221Updated 5 months ago
• hustvl / LightningDiT
  [CVPR 2025 Oral] Reconstruction vs. Generation: Taming Optimization Dilemma in Latent Diffusion Models
  ☆936Updated last week
• TongTong313 / rectified-flow
  从零手搓Flow Matching（Rectified Flow）
  ☆396Updated 6 months ago
• jannerm / ddpo
  Code for the paper "Training Diffusion Models with Reinforcement Learning"
  ☆473Updated last year
• G-U-N / Rectified-Diffusion
  [ICLR 2025] Rectified Diffusion: Straightness Is Not Your Need
  ☆227Updated 3 months ago
• gnobitab / RectifiedFlow
  Official Implementation of Rectified Flow (ICLR2023 Spotlight)
  ☆1,288Updated 11 months ago
• apple / ml-tarflow
  ☆200Updated 6 months ago
• Anima-Lab / MaskDiT
  Code for Fast Training of Diffusion Models with Masked Transformers
  ☆403Updated last year
• kvablack / ddpo-pytorch
  DDPO for finetuning diffusion models, implemented in PyTorch with LoRA support
  ☆606Updated last year
• lumalabs / imm
  Official implementation of Inductive Moment Matching
  ☆484Updated 3 months ago
• LTH14 / mar
  PyTorch implementation of MAR+DiffLoss https://arxiv.org/abs/2406.11838
  ☆1,613Updated 8 months ago
• TencentARC / SEED-Voken
  SEED-Voken: A Series of Powerful Visual Tokenizers
  ☆893Updated 4 months ago
• kwsong0113 / diffusion-forcing-transformer
  [ICML 2025] Official PyTorch Implementation of "History-Guided Video Diffusion"
  ☆359Updated 3 weeks ago
• VinAIResearch / LFM
  Official PyTorch implementation of the paper: Flow Matching in Latent Space
  ☆271Updated 5 months ago
• sail-sg / MDT
  Masked Diffusion Transformer is the SOTA for image synthesis. (ICCV 2023)
  ☆568Updated last year
• qihao067 / CrossFlow
  [CVPR2025] PyTorch-based reimplementation of CrossFlow, as proposed in 'Flowing from Words to Pixels: A Noise-Free Framework for Cross-Mo…
  ☆254Updated 2 weeks ago
• bytedance / 1d-tokenizer
  This repo contains the code for 1D tokenizer and generator
  ☆908Updated 3 months ago
• feizc / DiT-MoE
  Scaling Diffusion Transformers with Mixture of Experts
  ☆339Updated 9 months ago