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
  ☆364Updated last week
• sihyun-yu / REPA
  [ICLR'25 Oral] Representation Alignment for Generation: Training Diffusion Transformers Is Easier Than You Think
  ☆1,263Updated 5 months ago
• buoyancy99 / diffusion-forcing
  code for "Diffusion Forcing: Next-token Prediction Meets Full-Sequence Diffusion"
  ☆973Updated 4 months ago
• haidog-yaqub / MeanFlow
  Pytorch Implementation (unofficial) of the paper "Mean Flows for One-step Generative Modeling" by Geng et al.
  ☆708Updated last month
• NVlabs / edm2
  EDM2 and Autoguidance -- Official PyTorch implementation
  ☆752Updated 8 months ago
• sony / ctm
  ☆296Updated 10 months ago
• lucidrains / autoregressive-diffusion-pytorch
  Implementation of Autoregressive Diffusion in Pytorch
  ☆399Updated 9 months ago
• Gsunshine / meanflow
  JAX implementation of MeanFlow
  ☆327Updated 3 weeks ago
• locuslab / ect
  Consistency Models Made Easy
  ☆306Updated 10 months ago
• willisma / SiT
  Official PyTorch Implementation of "SiT: Exploring Flow and Diffusion-based Generative Models with Scalable Interpolant Transformers"
  ☆943Updated last year
• dongzhuoyao / awesome-flow-matching
  A summary of related works about flow matching, stochastic interpolants
  ☆537Updated 4 months ago
• gnobitab / RectifiedFlow
  Official Implementation of Rectified Flow (ICLR2023 Spotlight)
  ☆1,376Updated last year
• G-U-N / Rectified-Diffusion
  [ICLR 2025] Rectified Diffusion: Straightness Is Not Your Need
  ☆236Updated 5 months ago
• zhuyu-cs / MeanFlow
  Pytorch implementation of MeanFlow on ImageNet and CIFAR10
  ☆210Updated last week
• yifan123 / flow_grpo
  An official implementation of Flow-GRPO: Training Flow Matching Models via Online RL
  ☆1,116Updated last week
• apple / ml-tarflow
  ☆265Updated 8 months ago
• YangLing0818 / consistency_flow_matching
  Official Implementation for "Consistency Flow Matching: Defining Straight Flows with Velocity Consistency"
  ☆237Updated 7 months ago
• TongTong313 / rectified-flow
  从零手搓Flow Matching（Rectified Flow）
  ☆454Updated 8 months ago
• kvablack / ddpo-pytorch
  DDPO for finetuning diffusion models, implemented in PyTorch with LoRA support
  ☆660Updated last year
• lucidrains / mmdit
  Implementation of a single layer of the MMDiT, proposed in Stable Diffusion 3, in Pytorch
  ☆410Updated 7 months ago
• hustvl / LightningDiT
  [CVPR 2025 Oral] Reconstruction vs. Generation: Taming Optimization Dilemma in Latent Diffusion Models
  ☆1,122Updated 2 months ago
• jannerm / ddpo
  Code for the paper "Training Diffusion Models with Reinforcement Learning"
  ☆493Updated 2 years ago
• bytedance / 1d-tokenizer
  This repo contains the code for 1D tokenizer and generator
  ☆996Updated 5 months ago
• LTH14 / mar
  PyTorch implementation of MAR+DiffLoss https://arxiv.org/abs/2406.11838
  ☆1,702Updated 10 months ago
• TencentARC / SEED-Voken
  SEED-Voken: A Series of Powerful Visual Tokenizers
  ☆931Updated last month
• Anima-Lab / MaskDiT
  Code for Fast Training of Diffusion Models with Masked Transformers
  ☆406Updated last year
• kwsong0113 / diffusion-forcing-transformer
  [ICML 2025] Official PyTorch Implementation of "History-Guided Video Diffusion"
  ☆446Updated last month
• qihao067 / CrossFlow
  [CVPR2025] PyTorch-based reimplementation of CrossFlow, as proposed in 'Flowing from Words to Pixels: A Noise-Free Framework for Cross-Mo…
  ☆295Updated 2 months ago
• xandergos / sCM-mnist
  Unofficial implementation of "Simplifying, Stabilizing & Scaling Continuous-Time Consistency Models" for MNIST
  ☆72Updated 4 months ago
• lumalabs / imm
  Official implementation of Inductive Moment Matching
  ☆537Updated last month