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:
- ☆182Updated 4 months ago
- A mini-library for training consistency models.☆251Updated 2 years ago
- A general framework for inference-time scaling and steering of diffusion models with arbitrary rewards.☆210Updated 7 months ago
- Flow-matching algorithms in JAX☆115Updated last year
- Minimal Implementation of a D3PM in pytorch☆285Updated last year
- Reward fine-tuning for Stable Diffusion models based on stochastic optimal control, including Adjoint Matching☆63Updated 8 months ago
- Implementation of rectified flow and some of its followup research / improvements in Pytorch☆424Updated last week
- ☆320Updated last year
- Official Jax Implementation of MD4 Masked Diffusion Models☆153Updated 11 months ago
- Train VAE like a boss☆313Updated last year
- [ICML 2023] Reflected Diffusion Models (https://arxiv.org/abs/2304.04740)☆159Updated 2 years ago
- Implementation of Diffusion Transformer Model in Pytorch☆74Updated last month
- Annotated Flow Matching paper☆227Updated last year
- [ICML 2025] Implementation of Spatial Reasoning with Denoising Models☆86Updated 6 months ago
- ☆167Updated last month
- Implementation of a framework for Genie2 in Pytorch☆156Updated last year
- The codebase of our paper "Improving the Training of Rectified Flows", NeurIPS 2024☆128Updated last year
- Flash Attention Triton kernel with support for second-order derivatives☆138Updated last month
- Implementation of a multimodal diffusion transformer in Pytorch☆107Updated last year
- Implementation of Consistency Models (Song et al 2023) for few-step image generation in Jax.☆19Updated 2 years ago
- [ECCV 2024, Oral] FMBoost: Boosting Latent Diffusion with Flow Matching☆256Updated 3 months ago
- Consistency Models Made Easy☆324Updated last year
- A minimal implementation of a denoising diffusion model in PyTorch.☆131Updated 2 months ago
- A repo where I play with conditional flow approaches for learning time-varying vector-fields.☆23Updated last year
- A demo for the Direct Ascent Synthesis: Hidden Generative Capabilities in Discriminative Models paper (https://arxiv.org/abs/2502.07753)☆41Updated 11 months ago
- PyTorch implementation for "Parallel Sampling of Diffusion Models", NeurIPS 2023 Spotlight☆153Updated 2 years ago
- ☆123Updated 7 months ago
- ☆85Updated last year
- Official Implementation for "Consistency Flow Matching: Defining Straight Flows with Velocity Consistency"☆257Updated last year
- A Tutorial for Diffusion Models☆60Updated 2 years ago