perrying / gumbel-sinkhorn
PyTorch implementation of LEARNING LATENT PERMUTATIONS WITH GUMBEL-SINKHORN NETWORKS
☆26Updated 5 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"☆85Updated 3 years ago
- ☆36Updated 4 years ago
- Pytorch code for "Improving Self-Supervised Learning by Characterizing Idealized Representations"☆40Updated 2 years ago
- ☆12Updated 2 years ago
- Generative Modeling with Optimal Transport Maps - ICLR 2022☆59Updated 2 years ago
- LEARNING LATENT PERMUTATIONS WITH GUMBEL-SINKHORN NETWORKS IMPLEMENTATION WITH PYTORCH☆79Updated last year
- Differentiable Top-k Classification Learning☆80Updated 2 years ago
- [ICML'21] Improved Contrastive Divergence Training of Energy Based Models☆62Updated 3 years ago
- ☆50Updated 4 years ago
- ☆61Updated 2 years ago
- NeurIPS 2021, Code for Measuring Generalization with Optimal Transport☆28Updated 3 years ago
- [ICML2023] InfoOT: Information Maximizing Optimal Transport☆40Updated last year
- ☆33Updated 4 years ago
- ☆63Updated 2 years ago
- Weighted Training for Cross-Task Learning☆15Updated 2 years ago
- ☆32Updated 2 years ago
- NeurIPS 2022: Tree Mover’s Distance: Bridging Graph Metrics and Stability of Graph Neural Networks☆36Updated last year
- Code for "Generalised Implicit Neural Representations" (NeurIPS 2022).☆70Updated 2 years ago
- ☆36Updated 4 years ago
- [ICLR'23 Spotlight] gDDIM: analyze and accelerate general diffusion models, isotropic and non-isotropic☆49Updated last year
- PyTorch implementation for our ICLR 2024 paper "Diffusion Generative Flow Samplers: Improving learning signals through partial trajectory…☆24Updated last year
- Information-Theoretic Diffusion: A brand new diffusion model / density estimator based on information theory.☆33Updated 9 months ago
- Learnable Fourier Features for Multi-Dimensional Spatial Positional Encoding☆46Updated 6 months ago
- Code for "SoftSort: A Continuous Relaxation for the argsort Operator", ICML 2020.☆50Updated 2 years ago
- Code repository for the paper "Group Equivariant Stand-Alone Self Attention For Vision" published at ICLR 2021. https://openreview.net/fo…☆29Updated 4 years ago
- implements optimal transport algorithms in pytorch☆96Updated 2 years ago
- ☆30Updated 4 years ago
- Deep Learning & Information Bottleneck☆60Updated last year
- Bayesian Attention Modules☆35Updated 4 years ago
- Code for Accelerated Linearized Laplace Approximation for Bayesian Deep Learning (ELLA, NeurIPS 22')☆16Updated 2 years ago