antonioterpin / jkonet-starLinks
JAX implementation of the JKOnet* architecture presented in "Learning Diffusion at Lightspeed".
☆30Updated 3 months ago
Alternatives and similar repositories for jkonet-star
Users that are interested in jkonet-star are comparing it to the libraries listed below
Sorting:
- Code release for "Stochastic Optimal Control Matching"☆35Updated 10 months ago
- Neural Optimal Transport with Lagrangian Costs☆56Updated last month
- Deep Generalized Schrödinger Bridge, NeurIPS 2022 Oral☆48Updated 2 years ago
- A set of tests for evaluating large-scale algorithms for Wasserstein-1 transport computation (NeurIPS'22).☆21Updated 9 months ago
- ☆13Updated 2 years ago
- [ICML2022] Variational Wasserstein gradient flow☆22Updated 2 years ago
- ☆21Updated last year
- Collecting research materials on neural samplers with diffusion/flow models☆52Updated 3 weeks ago
- Official implementation of Deep Momentum Schrödinger Bridge☆26Updated last year
- ☆32Updated last year
- [ICLR 2022] Path integral sampler☆46Updated last year
- Second-Order Neural ODE Optimizer, NeurIPS 2021 spotlight☆52Updated 3 years ago
- Source code for Large-Scale Wasserstein Gradient Flows (NeurIPS 2021)☆32Updated 3 years ago
- NeurIPS'23: Energy Discrepancies: A Score-Independent Loss for Energy-Based Models☆16Updated 8 months ago
- Likelihood Training of Schrödinger Bridge using FBSDEs Theory, ICLR 2022☆84Updated 3 years ago
- Flow-matching algorithms in JAX☆97Updated 10 months ago
- A set of tests for evaluating large-scale algorithms for Wasserstein-2 transport maps computation (NeurIPS 2021)☆41Updated 3 years ago
- ☆34Updated 2 months ago
- The Wasserstein Distance and Optimal Transport Map of Gaussian Processes☆52Updated 4 years ago
- Implementation of Action Matching☆44Updated 2 years ago
- Light and Optimal Schrödinger Bridge Matching (ICML 2024) official PyTorch implementation☆41Updated 10 months ago
- GP Sinkhorn Implementation, paper: https://www.mdpi.com/1099-4300/23/9/1134☆23Updated 3 years ago
- Euclidean Wasserstein-2 optimal transportation☆47Updated last year
- Official implementation for our paper "Unsupervised Learning of Lagrangian Dynamics from Images for Prediction and Control"☆19Updated 2 years ago
- Pytorch implementation of "Entropic Neural Optimal Transport via Diffusion Processes" (NeurIPS 2023, oral).☆37Updated last year
- Official code for "Accelerating Feedforward Computation via Parallel Nonlinear Equation Solving", ICML 2021☆27Updated 3 years ago
- ☆40Updated 2 years ago
- PyTorch Implementation of Diffusion Schrodinger Bridge Matching☆161Updated 4 months ago
- ☆17Updated 2 years ago
- ☆152Updated last year