dome272 / Flow-Matching
My take on Flow Matching
☆50Updated 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 3 months ago
- A general framework for inference-time scaling and steering of diffusion models with arbitrary rewards.☆123Updated last month
- ☆22Updated 9 months ago
- ☆53Updated this week
- Flow-matching algorithms in JAX☆87Updated 8 months ago
- FID computation in Jax/Flax.☆27Updated 8 months ago
- Implementation of a multimodal diffusion transformer in Pytorch☆101Updated 9 months ago
- [ICML 2023] Reflected Diffusion Models (https://arxiv.org/abs/2304.04740)☆158Updated last year
- GENIE: Higher-Order Denoising Diffusion Solvers☆94Updated last year
- Synthetic Alphabet Dataset☆18Updated 2 weeks 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
- Implementation of a framework for Genie2 in Pytorch☆145Updated 3 months ago
- ☆32Updated 5 months ago
- Train VAE like a boss☆274Updated 5 months ago
- Implementation of the proposed Spline-Based Transformer from Disney Research☆88Updated 5 months ago
- UniDisc: A discrete diffusion model for joint multimodal generation, enabling controllable and efficient text-image synthesis, editing, a…☆77Updated last week
- A demo for the Direct Ascent Synthesis: Hidden Generative Capabilities in Discriminative Models paper (https://arxiv.org/abs/2502.07753)☆37Updated last month
- The Gaussian Histogram Loss (HL-Gauss) proposed by Imani et al. with a few convenient wrappers for regression, in Pytorch☆58Updated 2 months ago
- WIP☆93Updated 8 months ago
- Official PyTorch Implementation of "Diffusion Autoencoders are Scalable Image Tokenizers"☆110Updated 2 months ago
- The codebase of our paper "Improving the Training of Rectified Flows", NeurIPS 2024☆108Updated 5 months ago
- JAX implementation ViT-VQGAN☆82Updated 2 years ago
- ☆30Updated 6 months ago
- ☆27Updated 11 months ago
- Implementation of Consistency Models (Song et al 2023) for few-step image generation in Jax.☆20Updated last year
- RS-IMLE☆38Updated 4 months ago
- ☆44Updated last month
- ☆30Updated 2 months ago
- Official Jax Implementation of MD4 Masked Diffusion Models☆74Updated last month
- ☆51Updated last year