kvfrans / shortcut-modelsLinks
☆709Updated last year
Alternatives and similar repositories for shortcut-models
Users that are interested in shortcut-models are comparing it to the libraries listed below
Sorting:
- JAX implementation of MeanFlow☆510Updated 5 months ago
- code for "Diffusion Forcing: Next-token Prediction Meets Full-Sequence Diffusion"☆1,120Updated last month
- Implementation of rectified flow and some of its followup research / improvements in Pytorch☆403Updated this week
- [ICLR'25 Oral] Representation Alignment for Generation: Training Diffusion Transformers Is Easier Than You Think☆1,476Updated 9 months ago
- Pytorch Implementation (unofficial) of the paper "Mean Flows for One-step Generative Modeling" by Geng et al.☆987Updated 2 weeks ago
- Pytorch implementation of MeanFlow on ImageNet and CIFAR10☆387Updated this week
- ☆315Updated last month
- Implementation of Autoregressive Diffusion in Pytorch☆430Updated 3 weeks ago
- EDM2 and Autoguidance -- Official PyTorch implementation☆807Updated last year
- Consistency Models Made Easy☆322Updated last year
- Official PyTorch Implementation of "Diffusion Transformers with Representation Autoencoders"☆1,654Updated last month
- Official PyTorch Implementation of "SiT: Exploring Flow and Diffusion-based Generative Models with Scalable Interpolant Transformers"☆1,057Updated last week
- This repo contains the code for 1D tokenizer and generator☆1,087Updated 9 months ago
- A summary of related works about flow matching, stochastic interpolants☆621Updated 9 months ago
- Implementation of a single layer of the MMDiT, proposed in Stable Diffusion 3, in Pytorch☆494Updated 11 months ago
- Official Implementation of Rectified Flow (ICLR2023 Spotlight)☆1,507Updated last year
- [CVPR 2025 Oral] Reconstruction vs. Generation: Taming Optimization Dilemma in Latent Diffusion Models☆1,346Updated 2 weeks ago
- Official Implementation for "Consistency Flow Matching: Defining Straight Flows with Velocity Consistency"☆255Updated 11 months ago
- Pytorch implementation for MeanFlow☆277Updated 5 months ago
- PyTorch implementation of MAR+DiffLoss https://arxiv.org/abs/2406.11838☆1,830Updated last year
- SEED-Voken: A Series of Powerful Visual Tokenizers☆984Updated last month
- Code for the paper "Training Diffusion Models with Reinforcement Learning"☆546Updated 2 years ago
- [NeurIPS 2025] An official implementation of Flow-GRPO: Training Flow Matching Models via Online RL☆1,808Updated last month
- [ICLR 2025] Rectified Diffusion: Straightness Is Not Your Need☆243Updated 9 months ago
- DDPO for finetuning diffusion models, implemented in PyTorch with LoRA support☆722Updated last year
- ☆318Updated last year
- [ICCV 2025] Official implementation of the paper: REPA-E: Unlocking VAE for End-to-End Tuning of Latent Diffusion Transformers☆425Updated 3 weeks ago
- Unofficial implementation of "Simplifying, Stabilizing & Scaling Continuous-Time Consistency Models" for MNIST☆84Updated 9 months ago
- [ICML 2025] Official PyTorch Implementation of "History-Guided Video Diffusion"☆587Updated 5 months ago
- PyTorch implementation of JiT https://arxiv.org/abs/2511.13720☆1,864Updated 3 weeks ago