ExplainableML / uot-fm
Official Repository for "Unbalancedness in Neural Monge Maps Improves Unpaired Domain Translation" [ICLR 2024]
☆11Updated 5 months ago
Related projects ⓘ
Alternatives and complementary repositories for uot-fm
- ☆30Updated 5 months ago
- A set of tests for evaluating large-scale algorithms for Wasserstein-1 transport computation (NeurIPS'22).☆20Updated 2 months ago
- Implementation of Action Matching for the Schrödinger equation☆22Updated last year
- Code for GFlowNet-EM, a novel algorithm for fitting latent variable models with compositional latents and an intractable true posterior.☆38Updated 9 months ago
- Repository for the PopulAtion Parameter Averaging (PAPA) paper☆26Updated 7 months ago
- Transformers with doubly stochastic attention☆40Updated 2 years ago
- Code release for "Stochastic Optimal Control Matching"☆27Updated 2 months ago
- Implementation of Action Matching☆36Updated last year
- ☆33Updated last year
- PyTorch implementation for our ICLR 2024 paper "Diffusion Generative Flow Samplers: Improving learning signals through partial trajectory…☆21Updated 10 months ago
- ☆16Updated last year
- Official repository of Implicit Neural Convolutional Kernels for Steerable CNNs, Zhdanov et al.☆26Updated 7 months ago
- Neural Diffusion Processes☆72Updated 3 months ago
- GP Sinkhorn Implementation, paper: https://www.mdpi.com/1099-4300/23/9/1134☆22Updated 2 years ago
- Source code for Large-Scale Wasserstein Gradient Flows (NeurIPS 2021)☆29Updated 2 years ago
- Euclidean Wasserstein-2 optimal transportation☆44Updated last year
- ☆15Updated last year
- Official code for "Improved Techniques for Maximum Likelihood Estimation for Diffusion ODEs" (ICML 2023)☆14Updated 7 months ago
- ☆61Updated last year
- ☆49Updated 3 years ago
- PyTorch Implementation of Diffusion Schrodinger Bridge Matching☆119Updated last year
- NeurIPS'23: Energy Discrepancies: A Score-Independent Loss for Energy-Based Models☆11Updated 3 weeks ago
- Official code for "Accelerating Feedforward Computation via Parallel Nonlinear Equation Solving", ICML 2021☆26Updated 3 years ago
- [ICML2022] Variational Wasserstein gradient flow☆18Updated 2 years ago
- Code for the paper: Rotating Features for Object Discovery☆48Updated 3 months ago
- A set of tests for evaluating large-scale algorithms for Wasserstein-2 transport maps computation (NeurIPS 2021)☆35Updated 2 years ago
- Kernel Stein Discrepancy Descent : a method to sample from unnormalized densities☆21Updated 6 months ago
- The Energy Transformer block, in JAX☆50Updated 10 months ago
- NeurIPS23 "Flow Factorized Representation Learning"☆33Updated 2 weeks ago
- Improved sampling via learned diffusions (ICLR2024) and an optimal control perspective on diffusion-based generative modeling (TMLR2024)☆52Updated 2 months ago