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:
- Neural Optimal Transport with Lagrangian Costs☆56Updated last month
- Code release for "Stochastic Optimal Control Matching"☆36Updated 11 months ago
- Flow-matching algorithms in JAX☆97Updated 11 months ago
- Deep Generalized Schrödinger Bridge, NeurIPS 2022 Oral☆48Updated 2 years ago
- Neural Diffusion Processes☆81Updated 11 months ago
- [ICLR 2022] Path integral sampler☆48Updated last year
- ☆32Updated last year
- [ICML2022] Variational Wasserstein gradient flow☆23Updated 2 years ago
- Improved sampling via learned diffusions (ICLR2024) and an optimal control perspective on diffusion-based generative modeling (TMLR2024)☆62Updated 4 months ago
- NF-Layers for constructing neural functionals.☆87Updated last year
- ☆35Updated 3 months ago
- ☆44Updated last year
- PyTorch implementation of Shortcut Models [Frans, 2025] with little modification☆41Updated this week
- A repo where I play with conditional flow approaches for learning time-varying vector-fields.☆20Updated last year
- Euclidean Wasserstein-2 optimal transportation☆47Updated last year
- ☆21Updated last year
- Neural Laplace: Differentiable Laplace Reconstructions for modelling any time observation with O(1) complexity.☆78Updated 2 months ago
- ☆13Updated 2 years ago
- ☆110Updated last month
- Official implementation of Metric Flow Matching (NeurIPS 2024)☆39Updated this week
- ☆32Updated last year
- ☆14Updated last year
- A set of tests for evaluating large-scale algorithms for Wasserstein-2 transport maps computation (NeurIPS 2021)☆42Updated 3 years ago
- ☆18Updated 5 months ago
- ☆28Updated 3 weeks ago
- Official code for "Accelerating Feedforward Computation via Parallel Nonlinear Equation Solving", ICML 2021☆27Updated 3 years ago
- Experiment with diffusion models that you can run on your local jupyter instances☆63Updated 8 months ago
- Code for the paper: Rotating Features for Object Discovery☆52Updated 11 months ago
- The Wasserstein Distance and Optimal Transport Map of Gaussian Processes☆52Updated 4 years ago
- JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)☆30Updated last year