facebookresearch / lagrangian-otLinks
Neural Optimal Transport with Lagrangian Costs
☆59Updated 5 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:
- JAX implementation of the JKOnet* architecture presented in "Learning Diffusion at Lightspeed".☆31Updated 7 months ago
- Free-form flows are a generative model training a pair of neural networks via maximum likelihood☆49Updated 4 months ago
- NF-Layers for constructing neural functionals.☆90Updated last year
- ☆23Updated 2 years ago
- ☆35Updated 3 years ago
- ☆34Updated 2 months ago
- Deep Generalized Schrödinger Bridge, NeurIPS 2022 Oral☆52Updated 2 years ago
- Flow-matching algorithms in JAX☆106Updated last year
- Deep Networks Grok All the Time and Here is Why☆37Updated last year
- Neural Diffusion Processes☆81Updated last year
- Code release for "Stochastic Optimal Control Matching"☆37Updated last year
- Clifford-Steerable Convolutional Neural Networks [ICML'24]☆49Updated 5 months ago
- Variational Autoencoder with non-euclidean (hyperbolic) latent space☆11Updated 2 years ago
- [ICLR 2022] Path integral sampler☆52Updated 2 years ago
- Neural Laplace: Differentiable Laplace Reconstructions for modelling any time observation with O(1) complexity.☆81Updated 6 months ago
- Official repository for "Energy Matching: Unifying Flow Matching and Energy-Based Models for Generative Modeling" [NeurIPS 2025]☆170Updated last week
- LieTransformer: Equivariant Self-Attention for Lie Groups☆74Updated last year
- The Energy Transformer block, in JAX☆58Updated last year
- Meta Flow Matching: Integrating Vector Fields on the Wasserstein Manifold☆66Updated 7 months ago
- code for "Riemannian Flow Matching on General Geometries".☆266Updated last year
- Supplementary code for the paper "Stationary Kernels and Gaussian Processes on Lie Groups and their Homogeneous Spaces"☆43Updated last year
- ☆31Updated last year
- Official source code for "Graph Neural Networks for Learning Equivariant Representations of Neural Networks". In ICLR 2024 (oral).☆82Updated last year
- Euclidean Wasserstein-2 optimal transportation☆47Updated 2 years ago
- ☆12Updated last year
- This repository contains the source code to perform Geometry-aware Bayesian Optimization (GaBO) on Riemannian manifolds.☆53Updated 3 years ago
- Official repository for the paper "Neural Differential Equations for Learning to Program Neural Nets Through Continuous Learning Rules" (…☆23Updated 4 months ago
- The Wasserstein Distance and Optimal Transport Map of Gaussian Processes☆52Updated 5 years ago
- ICML 2022: Learning Iterative Reasoning through Energy Minimization☆48Updated 2 years ago
- Generative Equilibrium Transformer☆25Updated last year