dome272 / Flow-Matching
My take on Flow Matching
☆52Updated 3 months ago
Alternatives and similar repositories for Flow-Matching:
Users that are interested in Flow-Matching are comparing it to the libraries listed below
- Focused on fast experimentation and simplicity☆71Updated 4 months ago
- A general framework for inference-time scaling and steering of diffusion models with arbitrary rewards.☆137Updated 2 months ago
- [ICML 2023] Reflected Diffusion Models (https://arxiv.org/abs/2304.04740)☆157Updated last year
- WIP☆93Updated 8 months ago
- Flow-matching algorithms in JAX☆90Updated 8 months ago
- Train VAE like a boss☆276Updated 6 months ago
- A JAX implementation of the continuous time formulation of Consistency Models☆84Updated 2 years ago
- Sparse Autoencoders for Stable Diffusion XL models.☆55Updated 3 weeks ago
- ☆27Updated last year
- [ICML 2025] Gaussian Mixture Flow Matching Models (GMFlow)☆84Updated this week
- The aim of this repository is to test and implement Flow-Matching-based models☆90Updated 3 months ago
- Implementation of Consistency Models (Song et al 2023) for few-step image generation in Jax.☆20Updated last year
- The codebase of our paper "Improving the Training of Rectified Flows", NeurIPS 2024☆110Updated 6 months ago
- NeuMeta transforms neural networks by allowing a single model to adapt on the fly to different sizes, generating the right weights when n…☆42Updated 5 months ago
- FMBoost: Boosting Latent Diffusion with Flow Matching (ECCV 2024 Oral)☆226Updated 5 months ago
- UniDisc: A discrete diffusion model for joint multimodal generation, enabling controllable and efficient text-image synthesis, editing, a…☆94Updated last month
- Official PyTorch Implementation of "Diffusion Autoencoders are Scalable Image Tokenizers"☆112Updated 3 months ago
- Implementation of a framework for Genie2 in Pytorch☆145Updated 3 months ago
- Reward fine-tuning for Stable Diffusion models based on stochastic optimal control☆24Updated this week
- Implementation of the proposed Spline-Based Transformer from Disney Research☆88Updated 5 months ago
- GENIE: Higher-Order Denoising Diffusion Solvers☆94Updated last year
- ☆33Updated 6 months ago
- Official Implementation for "Consistency Flow Matching: Defining Straight Flows with Velocity Consistency"☆210Updated 3 months ago
- Implementation of rectified flow and some of its followup research / improvements in Pytorch☆285Updated last week
- Code and weights for the paper "Cluster and Predict Latents Patches for Improved Masked Image Modeling"☆101Updated 3 weeks ago
- [ICML 2024]: Official implementation for the paper: "Consistent Diffusion Meets Tweedie"☆53Updated last year
- ☆22Updated 10 months ago
- ☆174Updated 4 months ago
- Implementation of a multimodal diffusion transformer in Pytorch☆101Updated 10 months ago
- The Gaussian Histogram Loss (HL-Gauss) proposed by Imani et al. with a few convenient wrappers for regression, in Pytorch☆58Updated last week