perrying / gumbel-sinkhorn
PyTorch implementation of LEARNING LATENT PERMUTATIONS WITH GUMBEL-SINKHORN NETWORKS
☆29Updated 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
Sorting:
- LEARNING LATENT PERMUTATIONS WITH GUMBEL-SINKHORN NETWORKS IMPLEMENTATION WITH PYTORCH☆79Updated last year
- ☆36Updated 4 years ago
- Differentiable Top-k Classification Learning☆81Updated 2 years ago
- Pytorch code for "Improving Self-Supervised Learning by Characterizing Idealized Representations"☆41Updated 2 years ago
- Official implementation for the paper: Permutation Invariant Graph Generation via Score-Based Generative Modeling☆110Updated last year
- ☆12Updated 2 years ago
- Code for paper "Argmax Flows and Multinomial Diffusion: Learning Categorical Distributions"☆86Updated 3 years ago
- Weighted Training for Cross-Task Learning☆15Updated 2 years ago
- ☆60Updated 2 years ago
- Code for GFlowNet-EM, a novel algorithm for fitting latent variable models with compositional latents and an intractable true posterior.☆40Updated last year
- [ICML'21] Improved Contrastive Divergence Training of Energy Based Models☆63Updated 3 years ago
- ☆33Updated 4 years ago
- Generative Modeling with Optimal Transport Maps - ICLR 2022☆59Updated 3 years ago
- Multiscale de-noising score matching☆20Updated 4 years ago
- NeurIPS 2022: Tree Mover’s Distance: Bridging Graph Metrics and Stability of Graph Neural Networks☆36Updated last year
- Official Pytorch's implementation of RotoGrad☆76Updated last year
- NeurIPS 2021, Code for Measuring Generalization with Optimal Transport☆28Updated 3 years ago
- [NeurIPS'23 Spotlight] Learning Probabilistic Symmetrization for Architecture Agnostic Equivariance (LPS), in PyTorch☆29Updated last year
- ☆62Updated 3 years ago
- ☆23Updated 2 years ago
- Code for the paper https://arxiv.org/abs/2205.14987v2☆49Updated last year
- NeurIPS'22 Oral: EquiVSet - Learning Neural Set Functions Under the Optimal Subset Oracle☆19Updated 2 years ago
- ☆30Updated 4 years ago
- [ICML 2024 Oral] LSH-Based Efficient Point Transformer (HEPT)☆19Updated 3 months ago
- [ICML2023] InfoOT: Information Maximizing Optimal Transport☆40Updated 2 years ago
- Deep Learning & Information Bottleneck☆60Updated last year
- ☆50Updated 4 years ago
- Efficient Riemannian Optimization on Stiefel Manifold via Cayley Transform☆40Updated 6 years ago
- Score-based generative models for compact manifolds☆110Updated last year
- Information-Theoretic Diffusion: A brand new diffusion model / density estimator based on information theory.☆33Updated 10 months ago