Daniil-Selikhanovych / ASBMLinks
PyTorch implementation of "Adversarial Schrödinger Bridge Matching" (NeurIPS 2024)
☆19Updated 2 months ago
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 month
- Pytorch implementation of "Light Schrödinger Bridge" (ICLR 2024)☆58Updated last year
- Light and Optimal Schrödinger Bridge Matching (ICML 2024) official PyTorch implementation☆42Updated last year
- [ICLR 2024] Official implementation of Bellman Optimal Stepsize Straightening of Flow-Matching Models☆36Updated last year
- ☆13Updated last year
- Discrete Flow Matching implemented in PyTorch☆17Updated 4 months ago
- Pytorch implementation of "Neural Optimal Transport with General Cost Functionals" (ICLR 2024)☆18Updated 11 months ago
- PyTorch implementation of "Kernel Neural Optimal Transport" (ICLR 2023)☆25Updated last year
- code for "Generalized Schrödinger Bridge Matching" (ICLR 2024).☆73Updated last year
- An official pytorch implementation of AAAI 2024 paper "Latent Space Editing in Transformer-based Flow Matching"☆42Updated last year
- PyTorch Implementation of Diffusion Schrodinger Bridge Matching☆170Updated 5 months ago
- PyTorch implementation of "Light Unbalanced Optimal Transport" (NeurIPS 2024)☆19Updated 7 months ago
- Official PyTorch implementation for "Unifying Bayesian Flow Networks and Diffusion Models through Stochastic Differential Equations"☆40Updated last year
- Official PyTorch implementation for the paper Minimizing Trajectory Curvature of ODE-based Generative Models, ICML 2023☆85Updated 5 months ago
- Pytorch implementation of "Entropic Neural Optimal Transport via Diffusion Processes" (NeurIPS 2023, oral).☆37Updated last year
- Official code for "Improved Techniques for Maximum Likelihood Estimation for Diffusion ODEs" (ICML 2023)☆18Updated 5 months ago
- ☆22Updated 2 years ago
- Educational implementation of the Discrete Flow Matching paper☆98Updated 11 months ago
- Entropic Optimal Transport Benchmark (NeurIPS 2023).☆24Updated last year
- The official codebase for Reflected Flow Matching (ICML 2024)☆19Updated last year
- Official Implementation of understanding the latent space of diffusion models through the lens of riemannian geometry (NeurIPS 2023)☆88Updated last year
- Simple Guidance Mechanisms for Discrete Diffusion Models☆47Updated 7 months ago
- Official implementation of the NeurIPS 24 paper of statistical flow matching (SFM) for discrete generation.☆31Updated 9 months ago
- Official codebase for "Diffusion Bridge Implicit Models" (ICLR 2025) and "Consistency Diffusion Bridge Models" (NeurIPS 2024)☆71Updated 4 months ago
- The codebase of our paper "Improving the Training of Rectified Flows", NeurIPS 2024☆119Updated 9 months ago
- Official implementation of Learning to Discretize Denoising Diffusion ODEs☆24Updated 2 months ago
- Reward fine-tuning for Stable Diffusion models based on stochastic optimal control, including Adjoint Matching☆40Updated 2 months ago
- Official implementation of our paper "Bidirectional Consistency Models"; and reproduced Improved Consistency Models (iCT).☆20Updated 2 months ago
- Official Implementation (Pytorch) of "Constant Acceleration Flow", NeurIPS 2024☆33Updated 6 months ago
- The official repository for the paper "Optimal Flow Matching: Learning Straight Trajectories in Just One Step" (NeurIPS 2024)☆81Updated 7 months ago