antonioterpin / jkonet-star
JAX implementation of the JKOnet* architecture presented in "Learning Diffusion at Lightspeed".
☆29Updated last month
Alternatives and similar repositories for jkonet-star:
Users that are interested in jkonet-star are comparing it to the libraries listed below
- Code release for "Stochastic Optimal Control Matching"☆34Updated 8 months ago
- ☆13Updated last year
- Neural Optimal Transport with Lagrangian Costs☆55Updated 9 months ago
- [ICML2022] Variational Wasserstein gradient flow☆22Updated 2 years ago
- Deep Generalized Schrödinger Bridge, NeurIPS 2022 Oral☆47Updated 2 years ago
- ☆19Updated last year
- Official code for "Accelerating Feedforward Computation via Parallel Nonlinear Equation Solving", ICML 2021☆27Updated 3 years ago
- A repo where I play with conditional flow approaches for learning time-varying vector-fields.☆18Updated 11 months ago
- [ICLR 2022] Path integral sampler☆46Updated last year
- ☆32Updated 11 months ago
- Improved sampling via learned diffusions (ICLR2024) and an optimal control perspective on diffusion-based generative modeling (TMLR2024)☆61Updated last month
- Neural Diffusion Processes☆77Updated 9 months ago
- NF-Layers for constructing neural functionals.☆84Updated last year
- Reward fine-tuning for Stable Diffusion models based on stochastic optimal control☆24Updated last week
- Code for "Optimizing DDPM Sampling with Shortcut Fine-Tuning" (https://arxiv.org/abs/2301.13362), ICML 2023☆30Updated last year
- NeurIPS'23: Energy Discrepancies: A Score-Independent Loss for Energy-Based Models☆16Updated 6 months ago
- ☆30Updated last month
- Official PyTorch implementation for the paper Minimizing Trajectory Curvature of ODE-based Generative Models, ICML 2023☆82Updated 2 months ago
- Second-Order Neural ODE Optimizer, NeurIPS 2021 spotlight☆52Updated 3 years ago
- Implementation of Action Matching for the Schrödinger equation☆24Updated last year
- Code for the paper: Rotating Features for Object Discovery☆51Updated 8 months ago
- Flow-matching algorithms in JAX☆90Updated 8 months ago
- ☆45Updated last year
- Likelihood Training of Schrödinger Bridge using FBSDEs Theory, ICLR 2022☆83Updated 3 years ago
- The official repository for the paper "Optimal Flow Matching: Learning Straight Trajectories in Just One Step" (NeurIPS 2024)☆75Updated 4 months ago
- [ICLR'23 Spotlight] gDDIM: analyze and accelerate general diffusion models, isotropic and non-isotropic☆49Updated last year
- Official implementation of Deep Momentum Schrödinger Bridge☆25Updated last year
- The Official PyTorch Implementation of "VAEBM: A Symbiosis between Variational Autoencoders and Energy-based Models" (ICLR 2021 spotlight…☆56Updated 2 years ago
- ☆26Updated last month
- PyTorch Package For Quasimetric Learning☆42Updated 6 months ago