dome272 / Flow-MatchingLinks
My take on Flow Matching
☆90Updated last year
Alternatives and similar repositories for Flow-Matching
Users that are interested in Flow-Matching are comparing it to the libraries listed below
Sorting:
- ☆181Updated 3 months ago
- A mini-library for training consistency models.☆251Updated 2 years ago
- Implementation of Diffusion Transformer Model in Pytorch☆73Updated 3 weeks ago
- [ICML 2023] Reflected Diffusion Models (https://arxiv.org/abs/2304.04740)☆159Updated 2 years ago
- ☆320Updated last year
- Implementation of a multimodal diffusion transformer in Pytorch☆107Updated last year
- Flow-matching algorithms in JAX☆114Updated last year
- Official Jax Implementation of MD4 Masked Diffusion Models☆153Updated 11 months ago
- A general framework for inference-time scaling and steering of diffusion models with arbitrary rewards.☆207Updated 7 months ago
- Implementation of rectified flow and some of its followup research / improvements in Pytorch☆421Updated last week
- PyTorch implementation of Shortcut Models [Frans, 2025] with little modification☆70Updated 6 months ago
- The codebase of our paper "Improving the Training of Rectified Flows", NeurIPS 2024☆127Updated last year
- Implementation of a framework for Genie2 in Pytorch☆156Updated last year
- [ICML 2025] Implementation of Spatial Reasoning with Denoising Models☆86Updated 6 months ago
- Flash Attention Triton kernel with support for second-order derivatives☆138Updated last month
- Minimal Implementation of a D3PM in pytorch☆285Updated last year
- Annotated Flow Matching paper☆227Updated last year
- A repo where I play with conditional flow approaches for learning time-varying vector-fields.☆23Updated last year
- ☆123Updated 7 months ago
- [ECCV 2024, Oral] FMBoost: Boosting Latent Diffusion with Flow Matching☆256Updated 3 months ago
- [NeurIPS 2025] Official repository for "Energy Matching: Unifying Flow Matching and Energy-Based Models for Generative Modeling"☆198Updated 3 months ago
- Consistency Models Made Easy☆324Updated last year
- Implementation of the new SOTA for model based RL, from the paper "Improving Transformer World Models for Data-Efficient RL", in Pytorch☆153Updated 9 months ago
- The Gaussian Histogram Loss (HL-Gauss) proposed by Imani et al. with a few convenient wrappers for regression, in Pytorch☆72Updated 2 months ago
- PyTorch implementation for "Parallel Sampling of Diffusion Models", NeurIPS 2023 Spotlight☆153Updated 2 years ago
- ☆167Updated 3 weeks ago
- [ICML 2024]: Official implementation for the paper: "Consistent Diffusion Meets Tweedie"☆53Updated last year
- Reward fine-tuning for Stable Diffusion models based on stochastic optimal control, including Adjoint Matching☆62Updated 8 months ago
- ICML 2023: Reduce, Reuse, Recycle: Composing Energy-Based Diffusion Models with MCMC☆148Updated last year
- ☆85Updated last year