SUZ-tsinghua / jit_toy_exampleLinks
Unofficial implementation of the toy example in JiT https://arxiv.org/abs/2511.13720
☆57Updated last month
Alternatives and similar repositories for jit_toy_example
Users that are interested in jit_toy_example are comparing it to the libraries listed below
Sorting:
- ☆167Updated last week
- The aim of this repository is to test and implement Flow-Matching-based models☆120Updated last year
- Official Implementation for Diffusion Models Without Classifier-free Guidance☆168Updated 11 months ago
- [ICCV 2025] Official Implementation of Contrastive Flow Matching☆153Updated 6 months ago
- [CVPR2025] PyTorch-based reimplementation of CrossFlow, as proposed in 'Flowing from Words to Pixels: A Noise-Free Framework for Cross-Mo…☆328Updated 7 months ago
- Official Repo of From Masks to Worlds: A Hitchhiker’s Guide to World Models.☆67Updated 2 months ago
- PyTorch implementation of Shortcut Models [Frans, 2025] with little modification☆68Updated 6 months ago
- DDT: Decoupled Diffusion Transformer☆352Updated 4 months ago
- [Preprint] UCGM: Unified Continuous Generative Models☆175Updated 7 months ago
- PyTorch Implementation of Transfusion: Predict the Next Token and Diffuse Images with One Multi-Modal Model☆27Updated last year
- FlexTok: Resampling Images into 1D Token Sequences of Flexible Length☆284Updated 7 months ago
- JAX implementation of MeanFlow☆518Updated 5 months ago
- [ICML 2025] Gaussian Mixture Flow Matching Models (GMFlow)☆174Updated 2 months ago
- Implementation of a single layer of the MMDiT, proposed in Stable Diffusion 3, in Pytorch☆500Updated last year
- ConceptAttention: A method for interpreting multi-modal diffusion transformers.☆413Updated this week
- PyTorch re-implementation for MeanFlow☆115Updated 6 months ago
- Implementation of rectified flow and some of its followup research / improvements in Pytorch☆414Updated 2 weeks ago
- Official implementation of Inductive Moment Matching☆570Updated 6 months ago
- [ICCV 2025] Official implementation of the paper: REPA-E: Unlocking VAE for End-to-End Tuning of Latent Diffusion Transformers☆439Updated last month
- Pytorch implementation for MeanFlow☆291Updated 5 months ago
- Official PyTorch implementation of FlowMo.☆108Updated 9 months ago
- ElasticTok: Adaptive Tokenization for Image and Video☆87Updated last year
- DiffusionNFT: Online Diffusion Reinforcement with Forward Process☆542Updated last week
- Official PyTorch Implementation of "Diffusion Autoencoders are Scalable Image Tokenizers"☆164Updated 11 months ago
- [ICML 2025] Official PyTorch Implementation of "History-Guided Video Diffusion"☆598Updated 6 months ago
- ☆719Updated last year
- An ML research template with good documentation by Boyuan Chen, an MIT PhD student☆118Updated 10 months ago
- ☆162Updated last year
- Unofficial implementation of "Simplifying, Stabilizing & Scaling Continuous-Time Consistency Models" for MNIST☆84Updated 9 months ago
- Minimal unofficial implementation of Consistency Trajectory models on a 1D toy task.☆22Updated last year