ExplainableML / uot-fm
Official Repository for "Unbalancedness in Neural Monge Maps Improves Unpaired Domain Translation" [ICLR 2024]
☆14Updated 11 months ago
Alternatives and similar repositories for uot-fm:
Users that are interested in uot-fm are comparing it to the libraries listed below
- ☆32Updated 11 months ago
- ☆17Updated last year
- Code release for "Stochastic Optimal Control Matching"☆33Updated 8 months ago
- NeurIPS23 "Flow Factorized Representation Learning"☆35Updated 6 months ago
- ☆13Updated last year
- PyTorch implementation for our ICLR 2024 paper "Diffusion Generative Flow Samplers: Improving learning signals through partial trajectory…☆24Updated last year
- A set of tests for evaluating large-scale algorithms for Wasserstein-1 transport computation (NeurIPS'22).☆21Updated 7 months ago
- A set of tests for evaluating large-scale algorithms for Wasserstein-2 transport maps computation (NeurIPS 2021)☆40Updated 2 years ago
- Transformers with doubly stochastic attention☆45Updated 2 years ago
- Implementation of Action Matching☆41Updated last year
- ☆30Updated last month
- Code for GFlowNet-EM, a novel algorithm for fitting latent variable models with compositional latents and an intractable true posterior.☆40Updated last year
- ☆45Updated last year
- Implementation of Action Matching for the Schrödinger equation☆24Updated last year
- code for "Generalized Schrödinger Bridge Matching" (ICLR 2024).☆68Updated last year
- Reward fine-tuning for Stable Diffusion models based on stochastic optimal control☆24Updated this week
- Official Implementation of the paper: A Complete Recipe for Diffusion Generative Models☆30Updated 6 months ago
- [NeurIPS'23 Spotlight] Learning Probabilistic Symmetrization for Architecture Agnostic Equivariance (LPS), in PyTorch☆29Updated last year
- Simple Guidance Mechanisms for Discrete Diffusion Models☆36Updated 4 months ago
- Official implementation of Deep Momentum Schrödinger Bridge☆25Updated last year
- Implementation of papers in 101 lines of code.☆18Updated last year
- Official Jax Implementation of MD4 Masked Diffusion Models☆78Updated 2 months ago
- ☆19Updated 3 months ago
- ☆25Updated last month
- [ICLR'23 Spotlight] gDDIM: analyze and accelerate general diffusion models, isotropic and non-isotropic☆49Updated last year
- Light and Optimal Schrödinger Bridge Matching (ICML 2024) official PyTorch implementation☆39Updated 8 months ago
- NeurIPS'23: Energy Discrepancies: A Score-Independent Loss for Energy-Based Models☆16Updated 6 months ago
- Entropic Optimal Transport Benchmark (NeurIPS 2023).☆23Updated last year
- ☆60Updated 2 years ago
- A demo shows how to combine Langevin dynamics with score matching for generative models.☆38Updated 4 years ago