facebookresearch / lagrangian-otLinks
Neural Optimal Transport with Lagrangian Costs
☆56Updated 2 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 4 months ago
- ☆29Updated last month
- ☆22Updated last year
- NF-Layers for constructing neural functionals.☆87Updated last year
- Flow-matching algorithms in JAX☆100Updated 11 months ago
- Free-form flows are a generative model training a pair of neural networks via maximum likelihood☆47Updated last month
- ☆35Updated 3 years ago
- Code release for "Stochastic Optimal Control Matching"☆36Updated 11 months ago
- Deep Generalized Schrödinger Bridge, NeurIPS 2022 Oral☆49Updated 2 years ago
- Clifford-Steerable Convolutional Neural Networks [ICML'24]☆48Updated 2 months ago
- ☆32Updated last year
- code for "Riemannian Flow Matching on General Geometries".☆245Updated last year
- The Wasserstein Distance and Optimal Transport Map of Gaussian Processes☆53Updated 5 years ago
- Neural Diffusion Processes☆81Updated last year
- Tutorial on amortized optimization for learning to optimize over continuous domains☆243Updated 5 months ago
- Neural Laplace: Differentiable Laplace Reconstructions for modelling any time observation with O(1) complexity.☆80Updated 3 months ago
- Meta Flow Matching: Integrating Vector Fields on the Wasserstein Manifold☆57Updated 5 months ago
- Modern Fixed Point Systems using Pytorch☆102Updated last year
- Denoising Hamiltonian Network for Physical Reasoning☆32Updated 4 months ago
- ☆26Updated 2 years ago
- [ICLR 2022] Path integral sampler☆50Updated last year
- ☆12Updated last year
- Meta Optimal Transport☆103Updated 2 years ago
- LieTransformer: Equivariant Self-Attention for Lie Groups☆72Updated last year
- Deep Networks Grok All the Time and Here is Why☆37Updated last year
- Official repository for the paper "Neural Differential Equations for Learning to Program Neural Nets Through Continuous Learning Rules" (…☆23Updated last month
- Code for GFlowNet-EM, a novel algorithm for fitting latent variable models with compositional latents and an intractable true posterior.☆41Updated last year
- Official implementation of Deep Momentum Schrödinger Bridge☆27Updated last year
- Euclidean Wasserstein-2 optimal transportation☆47Updated last year
- Experiment with diffusion models that you can run on your local jupyter instances☆63Updated 9 months ago