lumalabs / immLinks
Official implementation of Inductive Moment Matching
☆561Updated 4 months ago
Alternatives and similar repositories for imm
Users that are interested in imm are comparing it to the libraries listed below
Sorting:
- Minimal implementation of scalable rectified flow transformers, based on SD3's approach☆616Updated last year
- Implementation of a single layer of the MMDiT, proposed in Stable Diffusion 3, in Pytorch☆465Updated 10 months ago
- Train VAE like a boss☆298Updated last year
- DDT: Decoupled Diffusion Transformer☆312Updated 2 months ago
- PeRFlow: Piecewise Rectified Flow as Universal Plug-and-Play Accelerator (NeurIPS 2024)☆527Updated 2 months ago
- FlexTok: Resampling Images into 1D Token Sequences of Flexible Length☆267Updated 5 months ago
- [CVPR2025] PyTorch-based reimplementation of CrossFlow, as proposed in 'Flowing from Words to Pixels: A Noise-Free Framework for Cross-Mo…☆317Updated 5 months ago
- EDM2 and Autoguidance -- Official PyTorch implementation☆786Updated 11 months ago
- [ECCV 2024, Oral] FMBoost: Boosting Latent Diffusion with Flow Matching☆249Updated 3 weeks ago
- Implementation of rectified flow and some of its followup research / improvements in Pytorch☆389Updated 3 months ago
- ☆158Updated 3 weeks ago
- The aim of this repository is to test and implement Flow-Matching-based models☆118Updated 9 months ago
- ConceptAttention: A method for interpreting multi-modal diffusion transformers.☆346Updated 6 months ago
- [ICLR 2025] Rectified Diffusion: Straightness Is Not Your Need☆242Updated 8 months ago
- [ICML 2025] Official PyTorch Implementation of "History-Guided Video Diffusion"☆543Updated 4 months ago
- [ICCV 2025] Official implementation of the paper: REPA-E: Unlocking VAE for End-to-End Tuning of Latent Diffusion Transformers☆385Updated 2 weeks ago
- Official Implementation for Diffusion Models Without Classifier-free Guidance☆161Updated 8 months ago
- ☆296Updated 10 months ago
- ☆677Updated 11 months ago
- HART: Efficient Visual Generation with Hybrid Autoregressive Transformer☆640Updated last year
- Consistency Models Made Easy☆318Updated last year
- [ICML'25] EQ-VAE: Equivariance Regularized Latent Space for Improved Generative Image Modeling.☆159Updated 4 months ago
- ☆310Updated last year
- JAX implementation of MeanFlow☆464Updated 3 months ago
- Official PyTorch Implementation of "Diffusion Autoencoders are Scalable Image Tokenizers"☆158Updated 9 months ago
- code for "Diffusion Forcing: Next-token Prediction Meets Full-Sequence Diffusion"☆1,073Updated this week
- UniDisc: A discrete diffusion model for joint multimodal generation, enabling controllable and efficient text-image synthesis, editing, a…☆131Updated 7 months ago
- A general framework for inference-time scaling and steering of diffusion models with arbitrary rewards.☆194Updated 4 months ago
- Pixel-Space Generative Models☆274Updated 6 months ago
- (NeurIPS 2024 Oral 🔥) Improved Distribution Matching Distillation for Fast Image Synthesis☆1,038Updated 8 months ago