facebookresearch / lagrangian-otLinks
Neural Optimal Transport with Lagrangian Costs
☆58Updated 4 months ago
Alternatives and similar repositories for lagrangian-ot
Users that are interested in lagrangian-ot are comparing it to the libraries listed below
Sorting:
- Free-form flows are a generative model training a pair of neural networks via maximum likelihood☆47Updated 3 months ago
- NF-Layers for constructing neural functionals.☆89Updated last year
- Flow-matching algorithms in JAX☆105Updated last year
- JAX implementation of the JKOnet* architecture presented in "Learning Diffusion at Lightspeed".☆31Updated 6 months ago
- ☆35Updated 3 years ago
- ☆32Updated last month
- ☆23Updated 2 years ago
- Deep Generalized Schrödinger Bridge, NeurIPS 2022 Oral☆52Updated 2 years ago
- Denoising Hamiltonian Network for Physical Reasoning☆37Updated 6 months ago
- Code release for "Stochastic Optimal Control Matching"☆37Updated last year
- Official repository for the paper "Neural Differential Equations for Learning to Program Neural Nets Through Continuous Learning Rules" (…☆23Updated 3 months ago
- ☆31Updated last year
- [ICLR 2022] Path integral sampler☆52Updated 2 years ago
- ☆13Updated 2 years ago
- Neural Diffusion Processes☆81Updated last year
- Implementation of Action Matching☆46Updated 2 years ago
- ☆33Updated 6 months ago
- The Wasserstein Distance and Optimal Transport Map of Gaussian Processes☆53Updated 5 years ago
- code for "Riemannian Flow Matching on General Geometries".☆263Updated last year
- Meta Flow Matching: Integrating Vector Fields on the Wasserstein Manifold☆64Updated 7 months ago
- code for "Generalized Schrödinger Bridge Matching" (ICLR 2024).☆79Updated last year
- Clifford-Steerable Convolutional Neural Networks [ICML'24]☆49Updated 4 months ago
- Modern Fixed Point Systems using Pytorch☆116Updated last year
- Reward fine-tuning for Stable Diffusion models based on stochastic optimal control, including Adjoint Matching☆44Updated 4 months ago
- Official repository for "Energy Matching: Unifying Flow Matching and Energy-Based Models for Generative Modeling" [NeurIPS 2025]☆152Updated this week
- Official implementation of Deep Momentum Schrödinger Bridge☆28Updated last year
- By introducing a differentiable contact model, DiffCoSim extends the applicability of Lagrangian/Hamiltonian-inspired neural networks to …☆36Updated 2 years ago
- Code for GFlowNet-EM, a novel algorithm for fitting latent variable models with compositional latents and an intractable true posterior.☆41Updated last year
- Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.☆58Updated 2 months ago
- [ICLR'25] Artificial Kuramoto Oscillatory Neurons☆102Updated last month