perrying / gumbel-sinkhornLinks
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:
- ☆12Updated 2 years ago
- Code for paper "Argmax Flows and Multinomial Diffusion: Learning Categorical Distributions"☆86Updated 4 years ago
- Pytorch code for "Improving Self-Supervised Learning by Characterizing Idealized Representations"☆41Updated 2 years ago
- ☆36Updated 4 years ago
- Differentiable Top-k Classification Learning☆82Updated 2 years ago
- NeurIPS 2022: Tree Mover’s Distance: Bridging Graph Metrics and Stability of Graph Neural Networks☆36Updated last year
- [ICML2023] InfoOT: Information Maximizing Optimal Transport☆40Updated 2 years ago
- Code for "Generalised Implicit Neural Representations" (NeurIPS 2022).☆70Updated 2 years ago
- LEARNING LATENT PERMUTATIONS WITH GUMBEL-SINKHORN NETWORKS IMPLEMENTATION WITH PYTORCH☆79Updated last year
- [NeurIPS'23 Spotlight] Learning Probabilistic Symmetrization for Architecture Agnostic Equivariance (LPS), in PyTorch☆29Updated last year
- Official implementation for the paper: Permutation Invariant Graph Generation via Score-Based Generative Modeling☆110Updated last year
- ☆61Updated 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
- Transformers with doubly stochastic attention☆45Updated 2 years ago
- PyTorch implementation for our ICLR 2024 paper "Diffusion Generative Flow Samplers: Improving learning signals through partial trajectory…☆24Updated last year
- [ICML'21] Improved Contrastive Divergence Training of Energy Based Models☆63Updated 3 years ago
- Generative Modeling with Optimal Transport Maps - ICLR 2022☆59Updated 3 years ago
- ☆37Updated 10 months ago
- Pytorch implementation of "Entropic Neural Optimal Transport via Diffusion Processes" (NeurIPS 2023, oral).☆37Updated last year
- [NeurIPS'21] Higher-order Transformers for sets, graphs, and hypergraphs, in PyTorch☆68Updated 2 years ago
- Weighted Training for Cross-Task Learning☆15Updated 2 years ago
- Pytorch implementation of neural processes and variants☆29Updated 10 months ago
- Code for GFlowNet-EM, a novel algorithm for fitting latent variable models with compositional latents and an intractable true posterior.☆40Updated last year
- ☆51Updated 4 years ago
- ☆23Updated 2 years ago
- ☆22Updated 3 years ago
- NeurIPS 2021, Code for Measuring Generalization with Optimal Transport☆28Updated 3 years ago
- A PyTorch implementation of The ICLR 2019 paper "Invariant and Equivariant Graph Networks" by Haggai Maron, Heli Ben-Hamu, Nadav Shamir a…☆16Updated 3 years ago
- NeurIPS23 "Flow Factorized Representation Learning"☆35Updated 7 months ago
- Official Code: Estimating Model Uncertainty of Neural Networks in Sparse Information Form, ICML2020.☆30Updated 4 years ago