facebookresearch / lagrangian-ot
Neural Optimal Transport with Lagrangian Costs
☆55Updated 8 months ago
Alternatives and similar repositories for lagrangian-ot:
Users that are interested in lagrangian-ot are comparing it to the libraries listed below
- Meta Flow Matching: Integrating Vector Fields on the Wasserstein Manifold☆52Updated last month
- Code release for "Stochastic Optimal Control Matching"☆32Updated 8 months ago
- ☆17Updated last year
- JAX implementation of the JKOnet* architecture presented in "Learning Diffusion at Lightspeed".☆30Updated 3 weeks ago
- Flow-matching algorithms in JAX☆87Updated 8 months ago
- Free-form flows are a generative model training a pair of neural networks via maximum likelihood☆43Updated 2 months ago
- ☆23Updated last week
- ☆32Updated 10 months ago
- Official implementation of Deep Momentum Schrödinger Bridge☆24Updated last year
- Deep Generalized Schrödinger Bridge, NeurIPS 2022 Oral☆47Updated 2 years ago
- Code repository of the paper "Clifford-Steerable Convolutional Neural Networks"☆47Updated 8 months ago
- [ICLR 2022] Path integral sampler☆45Updated last year
- A set of tests for evaluating large-scale algorithms for Wasserstein-2 transport maps computation (NeurIPS 2021)☆40Updated 2 years ago
- A repo where I play with conditional flow approaches for learning time-varying vector-fields.☆18Updated 10 months ago
- NF-Layers for constructing neural functionals.☆82Updated last year
- Official repository for the paper "Neural Differential Equations for Learning to Program Neural Nets Through Continuous Learning Rules" (…☆22Updated 2 years ago
- Code for "Invariance Learning in Deep Neural Networks with Differentiable Laplace Approximations"☆23Updated 2 years ago
- Official PyTorch and JAX Implementation of "Harmonics of Learning: Universal Fourier Features Emerge in Invariant Networks"☆38Updated last year
- Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.☆51Updated last week
- LieTransformer: Equivariant Self-Attention for Lie Groups☆69Updated 8 months ago
- Euclidean Wasserstein-2 optimal transportation