m1balcerak / EnergyMatchingLinks
Official repository for "Energy Matching: Unifying Flow Matching and Energy-Based Models for Generative Modeling" [NeurIPS 2025]
☆172Updated last week
Alternatives and similar repositories for EnergyMatching
Users that are interested in EnergyMatching are comparing it to the libraries listed below
Sorting:
- code for "Generalized Schrödinger Bridge Matching" (ICLR 2024).☆80Updated last year
- ☆289Updated 10 months ago
- ☆137Updated 3 weeks ago
- My take on Flow Matching☆80Updated 9 months ago
- Reward fine-tuning for Stable Diffusion models based on stochastic optimal control, including Adjoint Matching☆46Updated 5 months ago
- Flow-matching algorithms in JAX☆106Updated last year
- Meta Flow Matching: Integrating Vector Fields on the Wasserstein Manifold☆66Updated 7 months ago
- code for "Adjoint Sampling: Highly Scalable Diffusion Samplers via Adjoint Matching"☆122Updated 3 months ago
- code for "Riemannian Flow Matching on General Geometries".☆266Updated last year
- PyTorch Implementation of Diffusion Schrodinger Bridge Matching☆185Updated 8 months ago
- A general framework for inference-time scaling and steering of diffusion models with arbitrary rewards.☆193Updated 4 months ago
- ☆183Updated last year
- Neural Optimal Transport with Lagrangian Costs☆60Updated 5 months ago
- The official repository for the paper "Optimal Flow Matching: Learning Straight Trajectories in Just One Step" (NeurIPS 2024)☆95Updated 10 months ago
- Annotated Flow Matching paper☆215Updated last year
- Clifford-Steerable Convolutional Neural Networks [ICML'24]☆49Updated 5 months ago
- Code for PnP-Flow: Plug-and-Play image restoration with Flow Matching (ICLR 2025)☆124Updated last month
- ☆34Updated 2 months ago
- Code release for the paper Generative Modeling With Inverse Heat Dissipation☆79Updated 2 years ago
- A Tutorial for Diffusion Models☆59Updated 2 years ago
- Educational implementation of the Discrete Flow Matching paper☆122Updated last year
- Official PyTorch implementation of the paper: Flow Matching in Latent Space☆314Updated 9 months ago
- [ICML 2024]: Official implementation for the paper: "Consistent Diffusion Meets Tweedie"☆53Updated last year
- A repo where I play with conditional flow approaches for learning time-varying vector-fields.☆23Updated last year
- Erwin: A Tree-based Hierarchical Transformer for Large-scale Physical Systems [ICML'25]☆104Updated 3 weeks ago
- The Superposition of Diffusion Models Using the Itô Density Estimator☆51Updated 7 months ago
- ☆12Updated last year
- Code for the paper: Rotating Features for Object Discovery☆53Updated last year
- Official implementation of Metric Flow Matching (NeurIPS 2024)☆54Updated 2 months ago
- The official PyTorch implementation of Manifold Preserving Guided Diffusion (MPGD)☆37Updated last year