facebookresearch / lagrangian-ot
Neural Optimal Transport with Lagrangian Costs
☆50Updated 6 months ago
Alternatives and similar repositories for lagrangian-ot:
Users that are interested in lagrangian-ot are comparing it to the libraries listed below
- ☆18Updated last month
- Meta Flow Matching: Integrating Vector Fields on the Wasserstein Manifold☆43Updated 4 months ago
- ☆16Updated last year
- Code release for "Stochastic Optimal Control Matching"☆30Updated 5 months ago
- Flow-matching algorithms in JAX☆82Updated 5 months ago
- ☆31Updated 7 months ago
- Code repository of the paper "Clifford-Steerable Convolutional Neural Networks"☆43Updated 5 months ago
- Official implementation of Deep Momentum Schrödinger Bridge☆22Updated last year
- JAX implementation of the JKOnet* architecture presented in "Learning Diffusion at Lightspeed".☆26Updated 3 weeks ago
- Deep Generalized Schrödinger Bridge, NeurIPS 2022 Oral☆47Updated 2 years ago
- [ICLR 2022] Path integral sampler☆43Updated last year
- ☆49Updated 7 months ago
- code for "Generalized Schrödinger Bridge Matching" (ICLR 2024).☆51Updated 9 months ago
- Free-form flows are a generative model training a pair of neural networks via maximum likelihood☆39Updated 7 months ago
- code for "Riemannian Flow Matching on General Geometries".☆192Updated 10 months ago
- Code repository for Trajectory Flow Matching☆50Updated 2 months ago
- Supplementary code for the paper "Stationary Kernels and Gaussian Processes on Lie Groups and their Homogeneous Spaces"☆42Updated last year
- Euclidean Wasserstein-2 optimal transportation☆44Updated last year
- The Energy Transformer block, in JAX☆54Updated last year
- Visualizations of the theory behind diffusion models.☆77Updated 8 months ago
- Official PyTorch and JAX Implementation of "Harmonics of Learning: Universal Fourier Features Emerge in Invariant Networks"☆32Updated 11 months ago
- NF-Layers for constructing neural functionals.☆79Updated last year
- 3D Gaussian Splatting in JAX☆55Updated 7 months ago
- Official implementation of Metric Flow Matching (NeurIPS 2024)☆26Updated 2 months ago
- Implementation of Action Matching for the Schrödinger equation☆24Updated last year
- Code for Paper "Think While You Generate: Discrete Diffusion with Planned Denoising"☆35Updated 2 months ago
- ☆19Updated last year
- Flow Annealed Importance Sampling Bootstrap (FAB). ICLR 2023.☆53Updated 10 months ago
- Code for https://arxiv.org/abs/2406.04329☆51Updated last month
- LieTransformer: Equivariant Self-Attention for Lie Groups☆67Updated 5 months ago