perrying / gumbel-sinkhorn
PyTorch implementation of LEARNING LATENT PERMUTATIONS WITH GUMBEL-SINKHORN NETWORKS
☆26Updated 4 years ago
Alternatives and similar repositories for gumbel-sinkhorn:
Users that are interested in gumbel-sinkhorn are comparing it to the libraries listed below
- Code for paper "Argmax Flows and Multinomial Diffusion: Learning Categorical Distributions"☆84Updated 3 years ago
- ☆12Updated last year
- ☆35Updated 4 years ago
- ☆49Updated 3 years ago
- [ICML'21] Improved Contrastive Divergence Training of Energy Based Models☆61Updated 2 years ago
- LEARNING LATENT PERMUTATIONS WITH GUMBEL-SINKHORN NETWORKS IMPLEMENTATION WITH PYTORCH☆79Updated last year
- ☆61Updated last year
- NeurIPS 2021, Code for Measuring Generalization with Optimal Transport☆28Updated 3 years ago
- Code for "Generalised Implicit Neural Representations" (NeurIPS 2022).☆68Updated last year
- Pytorch implementation of "Entropic Neural Optimal Transport via Diffusion Processes" (NeurIPS 2023, oral).☆35Updated 10 months ago
- NeurIPS23 "Flow Factorized Representation Learning"☆35Updated 3 months ago
- The repository contains reproducible PyTorch source code of our paper Generative Modeling with Optimal Transport Maps, ICLR 2022.☆57Updated 2 years ago
- Pytorch code for "Improving Self-Supervised Learning by Characterizing Idealized Representations"☆40Updated 2 years ago
- NeurIPS 2022: Tree Mover’s Distance: Bridging Graph Metrics and Stability of Graph Neural Networks☆36Updated last year
- PyTorch implementation for our ICLR 2024 paper "Diffusion Generative Flow Samplers: Improving learning signals through partial trajectory…☆23Updated last year
- A PyTorch implementation of The ICLR 2019 paper "Invariant and Equivariant Graph Networks" by Haggai Maron, Heli Ben-Hamu, Nadav Shamir a…☆15Updated 2 years ago
- A new mini-batch framework for optimal transport in deep generative models, deep domain adaptation, approximate Bayesian computation, col…☆37Updated last year
- Differentiable Top-k Classification Learning☆79Updated 2 years ago
- [ICML2023] InfoOT: Information Maximizing Optimal Transport☆40Updated last year
- ☆212Updated last year
- Code for "SoftSort: A Continuous Relaxation for the argsort Operator", ICML 2020.☆44Updated last year
- ☆22Updated 2 years ago
- Code for "Implicit Normalizing Flows" (ICLR 2021 spotlight)☆34Updated 3 years ago
- Official code for "Maximum Likelihood Training of Score-Based Diffusion Models", NeurIPS 2021 (spotlight)☆134Updated 3 years ago
- Multiscale de-noising score matching☆19Updated 3 years ago
- [NeurIPS'23 Spotlight] Learning Probabilistic Symmetrization for Architecture Agnostic Equivariance (LPS), in PyTorch☆29Updated 9 months ago
- Transformers with doubly stochastic attention☆44Updated 2 years ago
- Code for the paper https://arxiv.org/abs/2205.14987v2☆47Updated 9 months ago
- Weighted Training for Cross-Task Learning☆15Updated last year
- ☆31Updated 4 years ago