Alternatives and similar repositories for shortcut-models

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

• lucidrains / rectified-flow-pytorch
  Implementation of rectified flow and some of its followup research / improvements in Pytorch
  ☆398Updated this week
• Gsunshine / meanflow
  JAX implementation of MeanFlow
  ☆472Updated 4 months ago
• sihyun-yu / REPA
  [ICLR'25 Oral] Representation Alignment for Generation: Training Diffusion Transformers Is Easier Than You Think
  ☆1,430Updated 8 months ago
• buoyancy99 / diffusion-forcing
  code for "Diffusion Forcing: Next-token Prediction Meets Full-Sequence Diffusion"
  ☆1,091Updated 3 weeks ago
• sony / ctm
  ☆313Updated 2 weeks ago
• haidog-yaqub / MeanFlow
  Pytorch Implementation (unofficial) of the paper "Mean Flows for One-step Generative Modeling" by Geng et al.
  ☆941Updated last month
• NVlabs / edm2
  EDM2 and Autoguidance -- Official PyTorch implementation
  ☆795Updated 11 months ago
• lucidrains / autoregressive-diffusion-pytorch
  Implementation of Autoregressive Diffusion in Pytorch
  ☆418Updated last year
• zhuyu-cs / MeanFlow
  Pytorch implementation of MeanFlow on ImageNet and CIFAR10
  ☆354Updated 3 months ago
• willisma / SiT
  Official PyTorch Implementation of "SiT: Exploring Flow and Diffusion-based Generative Models with Scalable Interpolant Transformers"
  ☆1,032Updated 3 weeks ago
• locuslab / ect
  Consistency Models Made Easy
  ☆317Updated last year
• dongzhuoyao / awesome-flow-matching
  A summary of related works about flow matching, stochastic interpolants
  ☆604Updated 8 months ago
• gnobitab / RectifiedFlow
  Official Implementation of Rectified Flow (ICLR2023 Spotlight)
  ☆1,488Updated last year
• lucidrains / mmdit
  Implementation of a single layer of the MMDiT, proposed in Stable Diffusion 3, in Pytorch
  ☆482Updated 10 months ago
• jannerm / ddpo
  Code for the paper "Training Diffusion Models with Reinforcement Learning"
  ☆537Updated 2 years ago
• hustvl / LightningDiT
  [CVPR 2025 Oral] Reconstruction vs. Generation: Taming Optimization Dilemma in Latent Diffusion Models
  ☆1,302Updated 5 months ago
• YangLing0818 / consistency_flow_matching
  Official Implementation for "Consistency Flow Matching: Defining Straight Flows with Velocity Consistency"
  ☆251Updated 10 months ago
• kvablack / ddpo-pytorch
  DDPO for finetuning diffusion models, implemented in PyTorch with LoRA support
  ☆708Updated last year
• Gsunshine / py-meanflow
  Pytorch implementation for MeanFlow
  ☆240Updated 4 months ago
• bytedance / 1d-tokenizer
  This repo contains the code for 1D tokenizer and generator
  ☆1,078Updated 8 months ago
• LTH14 / JiT
  PyTorch implementation of JiT https://arxiv.org/abs/2511.13720
  ☆1,374Updated 2 weeks ago
• bytetriper / RAE
  Official PyTorch Implementation of "Diffusion Transformers with Representation Autoencoders"
  ☆1,594Updated 2 weeks ago
• G-U-N / Rectified-Diffusion
  [ICLR 2025] Rectified Diffusion: Straightness Is Not Your Need
  ☆241Updated 8 months ago
• LTH14 / mar
  PyTorch implementation of MAR+DiffLoss https://arxiv.org/abs/2406.11838
  ☆1,802Updated last year
• apple / ml-tarflow
  ☆299Updated 11 months ago
• TongTong313 / rectified-flow
  从零手搓Flow Matching（Rectified Flow）
  ☆546Updated 11 months ago
• xandergos / sCM-mnist
  Unofficial implementation of "Simplifying, Stabilizing & Scaling Continuous-Time Consistency Models" for MNIST
  ☆81Updated 8 months ago
• TencentARC / SEED-Voken
  SEED-Voken: A Series of Powerful Visual Tokenizers
  ☆981Updated last week
• Anima-Lab / MaskDiT
  Code for Fast Training of Diffusion Models with Masked Transformers
  ☆417Updated last year
• End2End-Diffusion / REPA-E
  [ICCV 2025] Official implementation of the paper: REPA-E: Unlocking VAE for End-to-End Tuning of Latent Diffusion Transformers
  ☆407Updated last month