machinestein / GNOTLinks
Pytorch implementation of "Neural Optimal Transport with General Cost Functionals" (ICLR 2024)
☆18Updated last year
Alternatives and similar repositories for GNOT
Users that are interested in GNOT are comparing it to the libraries listed below
Sorting:
- Pytorch implementation of "Light Schrödinger Bridge" (ICLR 2024)☆61Updated last year
- Light and Optimal Schrödinger Bridge Matching (ICML 2024) official PyTorch implementation☆50Updated last year
- Reward fine-tuning for Stable Diffusion models based on stochastic optimal control, including Adjoint Matching☆46Updated 4 months ago
- PyTorch implementation of "Robust Barycenter Estimation using Semi-unbalanced Neural Optimal Transport" (ICLR 2025)☆12Updated 3 months ago
- A repo where I play with conditional flow approaches for learning time-varying vector-fields.☆23Updated last year
- ☆33Updated 7 months ago
- Entropic Optimal Transport Benchmark (NeurIPS 2023).☆27Updated last year
- Neural Diffusion Processes☆81Updated last year
- Code release for "Stochastic Optimal Control Matching"☆37Updated last year
- NeurIPS23 "Flow Factorized Representation Learning"☆38Updated last year
- ☆37Updated last year
- PyTorch implementation for our ICLR 2024 paper "Diffusion Generative Flow Samplers: Improving learning signals through partial trajectory…☆24Updated last year
- PyTorch implementation of "Kernel Neural Optimal Transport" (ICLR 2023)☆27Updated 2 years ago
- A set of tests for evaluating large-scale algorithms for Wasserstein-1 transport computation (NeurIPS'22).☆21Updated last year
- The official implementation of "Smoothie: Smoothing Diffusion on Token Embeddings for Text Generation"☆22Updated last month
- ☆34Updated 2 months ago
- NF-Layers for constructing neural functionals.☆90Updated last year
- Improved sampling via learned diffusions (ICLR2024) and an optimal control perspective on diffusion-based generative modeling (TMLR2024)☆69Updated 7 months ago
- ☆31Updated last year
- Official Jax Implementation of MD4 Masked Diffusion Models☆138Updated 8 months ago
- Code for paper "Compositional Sculpting of Iterative Generative Processes"☆25Updated 2 years ago
- ☆51Updated last year
- PyTorch implementation of "Neural Optimal Transport" (ICLR 2023 Spotlight)☆210Updated last year
- Inference-Time Alignment in Protein Diffusion Models☆44Updated 9 months ago
- Pytorch implementation of "Entropic Neural Optimal Transport via Diffusion Processes" (NeurIPS 2023, oral).☆39Updated last year
- Code for the paper: "Fine-Tuning Discrete Diffusion Models with Policy Gradient Methods"☆27Updated 5 months ago
- Generative Equilibrium Transformer☆25Updated last year
- Code for ICLR 2023 Paper, "Stable Target Field for Reduced Variance Score Estimation in Diffusion Models”☆76Updated 2 years ago
- PyTorch implementation of "Light Unbalanced Optimal Transport" (NeurIPS 2024)☆19Updated 10 months ago
- Simple Guidance Mechanisms for Discrete Diffusion Models☆57Updated 10 months ago