Daniil-Selikhanovych / ASBMLinks
PyTorch implementation of "Adversarial Schrödinger Bridge Matching" (NeurIPS 2024)
☆19Updated last month
Alternatives and similar repositories for ASBM
Users that are interested in ASBM are comparing it to the libraries listed below
Sorting:
- PyTorch implementation of "Robust Barycenter Estimation using Semi-unbalanced Neural Optimal Transport" (ICLR 2025)☆11Updated last week
- Pytorch implementation of "Light Schrödinger Bridge" (ICLR 2024)☆58Updated 11 months ago
- Light and Optimal Schrödinger Bridge Matching (ICML 2024) official PyTorch implementation☆41Updated 11 months ago
- Pytorch implementation of "Neural Optimal Transport with General Cost Functionals" (ICLR 2024)☆18Updated 10 months ago
- ☆13Updated last year
- Official implementation of Learning to Discretize Denoising Diffusion ODEs☆23Updated last month
- [ICLR 2024] Official implementation of Bellman Optimal Stepsize Straightening of Flow-Matching Models☆35Updated last year
- PyTorch implementation of "Kernel Neural Optimal Transport" (ICLR 2023)☆25Updated last year
- Pytorch implementation of "Entropic Neural Optimal Transport via Diffusion Processes" (NeurIPS 2023, oral).☆37Updated last year
- Official PyTorch implementation for the paper Minimizing Trajectory Curvature of ODE-based Generative Models, ICML 2023☆85Updated 5 months ago
- code for "Generalized Schrödinger Bridge Matching" (ICLR 2024).☆72Updated last year
- Official code for "Improved Techniques for Maximum Likelihood Estimation for Diffusion ODEs" (ICML 2023)☆18Updated 4 months ago
- Entropic Optimal Transport Benchmark (NeurIPS 2023).☆24Updated last year
- Simple Guidance Mechanisms for Discrete Diffusion Models☆46Updated 7 months ago
- PyTorch Implementation of Diffusion Schrodinger Bridge Matching☆166Updated 4 months ago
- Official PyTorch implementation for "Unifying Bayesian Flow Networks and Diffusion Models through Stochastic Differential Equations"☆40Updated last year
- PyTorch implementation of "Light Unbalanced Optimal Transport" (NeurIPS 2024)☆19Updated 6 months ago
- Reward fine-tuning for Stable Diffusion models based on stochastic optimal control, including Adjoint Matching☆39Updated last month
- ☆22Updated last year
- Official Implementation of understanding the latent space of diffusion models through the lens of riemannian geometry (NeurIPS 2023)☆88Updated last year
- An official pytorch implementation of AAAI 2024 paper "Latent Space Editing in Transformer-based Flow Matching"☆42Updated last year
- [NeurIPS 2023] Formulating Discrete Probability Flow Through Optimal Transport☆20Updated last year
- PyTorch implementation of "Extremal Domain Translation with Neural Optimal Transport" (NeurIPS 2023)☆20Updated 6 months ago
- Code for TFG: Unified Training-Free Guidance for Diffusion Models☆60Updated 2 months ago
- Educational implementation of the Discrete Flow Matching paper☆95Updated 10 months ago
- ☆221Updated last year
- [ICML 2025] GitHub repository for LDMol, a latent text-to-molecule diffusion model.☆15Updated 3 weeks ago
- Official codebase for "Diffusion Bridge Implicit Models" (ICLR 2025) and "Consistency Diffusion Bridge Models" (NeurIPS 2024)☆68Updated 4 months ago
- ICLR25 | Official code base for Heavy-Tailed Diffusion with Denoising Levy Probabilistic Models (DLPM)☆16Updated 5 months ago
- The codebase of our paper "Improving the Training of Rectified Flows", NeurIPS 2024☆117Updated 8 months ago