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
  ☆381Updated last month
• Gsunshine / meanflow
  JAX implementation of MeanFlow
  ☆431Updated 2 months ago
• buoyancy99 / diffusion-forcing
  code for "Diffusion Forcing: Next-token Prediction Meets Full-Sequence Diffusion"
  ☆1,028Updated 6 months ago
• sihyun-yu / REPA
  [ICLR'25 Oral] Representation Alignment for Generation: Training Diffusion Transformers Is Easier Than You Think
  ☆1,326Updated 6 months ago
• NVlabs / edm2
  EDM2 and Autoguidance -- Official PyTorch implementation
  ☆771Updated 9 months ago
• haidog-yaqub / MeanFlow
  Pytorch Implementation (unofficial) of the paper "Mean Flows for One-step Generative Modeling" by Geng et al.
  ☆865Updated 3 months ago
• sony / ctm
  ☆303Updated 11 months ago
• lucidrains / autoregressive-diffusion-pytorch
  Implementation of Autoregressive Diffusion in Pytorch
  ☆409Updated 11 months ago
• locuslab / ect
  Consistency Models Made Easy
  ☆312Updated 11 months ago
• willisma / SiT
  Official PyTorch Implementation of "SiT: Exploring Flow and Diffusion-based Generative Models with Scalable Interpolant Transformers"
  ☆968Updated last year
• zhuyu-cs / MeanFlow
  Pytorch implementation of MeanFlow on ImageNet and CIFAR10
  ☆262Updated last month
• dongzhuoyao / awesome-flow-matching
  A summary of related works about flow matching, stochastic interpolants
  ☆570Updated 6 months ago
• YangLing0818 / consistency_flow_matching
  Official Implementation for "Consistency Flow Matching: Defining Straight Flows with Velocity Consistency"
  ☆242Updated 8 months ago
• hustvl / LightningDiT
  [CVPR 2025 Oral] Reconstruction vs. Generation: Taming Optimization Dilemma in Latent Diffusion Models
  ☆1,193Updated 3 months ago
• lucidrains / mmdit
  Implementation of a single layer of the MMDiT, proposed in Stable Diffusion 3, in Pytorch
  ☆452Updated 8 months ago
• G-U-N / Rectified-Diffusion
  [ICLR 2025] Rectified Diffusion: Straightness Is Not Your Need
  ☆238Updated 6 months ago
• gnobitab / RectifiedFlow
  Official Implementation of Rectified Flow (ICLR2023 Spotlight)
  ☆1,424Updated last year
• bytedance / 1d-tokenizer
  This repo contains the code for 1D tokenizer and generator
  ☆1,039Updated 6 months ago
• yifan123 / flow_grpo
  An official implementation of Flow-GRPO: Training Flow Matching Models via Online RL
  ☆1,373Updated 2 weeks ago
• kvablack / ddpo-pytorch
  DDPO for finetuning diffusion models, implemented in PyTorch with LoRA support
  ☆685Updated last year
• xandergos / sCM-mnist
  Unofficial implementation of "Simplifying, Stabilizing & Scaling Continuous-Time Consistency Models" for MNIST
  ☆77Updated 6 months ago
• TongTong313 / rectified-flow
  从零手搓Flow Matching（Rectified Flow）
  ☆493Updated 9 months ago
• jannerm / ddpo
  Code for the paper "Training Diffusion Models with Reinforcement Learning"
  ☆514Updated 2 years ago
• apple / ml-tarflow
  ☆281Updated 9 months ago
• lumalabs / imm
  Official implementation of Inductive Moment Matching
  ☆549Updated 2 months ago
• LTH14 / mar
  PyTorch implementation of MAR+DiffLoss https://arxiv.org/abs/2406.11838
  ☆1,750Updated last year
• Anima-Lab / MaskDiT
  Code for Fast Training of Diffusion Models with Masked Transformers
  ☆412Updated 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…
  ☆301Updated 3 months ago
• kwsong0113 / diffusion-forcing-transformer
  [ICML 2025] Official PyTorch Implementation of "History-Guided Video Diffusion"
  ☆503Updated 3 months ago
• valeoai / Halton-MaskGIT
  [ICLR2025] Halton Scheduler for Masked Generative Image Transformer
  ☆264Updated 3 months ago