nmboffi / flow-mapsLinks
Official codebase for the paper "How to build a consistency model: Learning flow maps via self-distillation" (NeurIPS 2025).
☆69Updated 3 months ago
Alternatives and similar repositories for flow-maps
Users that are interested in flow-maps are comparing it to the libraries listed below
Sorting:
- code for "Riemannian Flow Matching on General Geometries".☆282Updated last year
- [NeurIPS 2025] Official repository for "Energy Matching: Unifying Flow Matching and Energy-Based Models for Generative Modeling"☆193Updated 3 months ago
- Reward fine-tuning for Stable Diffusion models based on stochastic optimal control, including Adjoint Matching☆59Updated 7 months ago
- Neural Optimal Transport with Lagrangian Costs☆61Updated 8 months ago
- code for "Generalized Schrödinger Bridge Matching" (ICLR 2024).☆84Updated last year
- ☆174Updated 3 months ago
- [ICML 2025] The official implementation of the paper "On the Guidance of Flow Matching"☆75Updated 4 months ago
- Flow-matching algorithms in JAX☆114Updated last year
- [ICML 2024] LLM and Simulation as Bilevel Optimizers: A New Paradigm to Advance Physical Scientific Discovery☆81Updated last year
- Definitive implementation of the stochastic interpolant framework for generative modeling in jax.☆40Updated 6 months ago
- ☆36Updated last week
- Denoising Hamiltonian Network for Physical Reasoning☆45Updated 9 months ago
- Code for "Kuramoto Orientation Diffusion"☆26Updated 2 months ago
- NF-Layers for constructing neural functionals.☆93Updated 2 years ago
- A comprehensive JAX/NNX library for diffusion and flow matching generative algorithms, featuring DiT (Diffusion Transformer) and its vari…☆128Updated 3 months ago
- ☆66Updated 5 months ago
- My take on Flow Matching☆90Updated last year
- A Tutorial for Diffusion Models☆60Updated 2 years ago
- [ICLR'25] Artificial Kuramoto Oscillatory Neurons☆106Updated 3 months ago
- Code for the paper: Rotating Features for Object Discovery☆53Updated last year
- A convenient way to trigger synchronizations to wandb / Weights & Biases if your compute nodes don't have internet!☆88Updated 2 weeks ago
- A general framework for inference-time scaling and steering of diffusion models with arbitrary rewards.☆205Updated 6 months ago
- ☆26Updated last year
- code for "Adjoint Sampling: Highly Scalable Diffusion Samplers via Adjoint Matching"☆129Updated 5 months ago
- Annotated Flow Matching paper☆227Updated last year
- Erwin: A Tree-based Hierarchical Transformer for Large-scale Physical Systems [ICML'25]☆110Updated 3 months ago
- Meta Flow Matching: Integrating Vector Fields on the Wasserstein Manifold☆69Updated 10 months ago
- Code release for "Stochastic Optimal Control Matching"☆39Updated last year
- Score-based generative models for compact manifolds☆119Updated 2 years ago
- Official JAX implementation of neural isometries - taming transformations for equivariant ML☆36Updated 5 months ago