perrying / gumbel-sinkhornLinks
PyTorch implementation of LEARNING LATENT PERMUTATIONS WITH GUMBEL-SINKHORN NETWORKS
☆31Updated 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:
- Code for paper "Argmax Flows and Multinomial Diffusion: Learning Categorical Distributions"☆91Updated 4 years ago
- Differentiable Top-k Classification Learning☆91Updated 3 years ago
- Pytorch code for "Improving Self-Supervised Learning by Characterizing Idealized Representations"☆41Updated 3 years ago
- [ICML'21] Improved Contrastive Divergence Training of Energy Based Models☆69Updated 3 years ago
- LEARNING LATENT PERMUTATIONS WITH GUMBEL-SINKHORN NETWORKS IMPLEMENTATION WITH PYTORCH☆81Updated 2 years ago
- Official implementation for the paper: Permutation Invariant Graph Generation via Score-Based Generative Modeling☆115Updated 2 years ago
- ☆14Updated 2 years ago
- NeurIPS 2022: Tree Mover’s Distance: Bridging Graph Metrics and Stability of Graph Neural Networks☆37Updated 2 years ago
- Transformers with doubly stochastic attention☆51Updated 3 years ago
- Deep Learning & Information Bottleneck☆63Updated 2 years ago
- Papers and Codes for the deep learning in hyperbolic space☆186Updated 3 years ago
- Code for our paper "Generative Flow Networks for Discrete Probabilistic Modeling"☆87Updated 2 years ago
- ☆24Updated 3 years ago
- [ICML 2024 Oral] LSH-Based Efficient Point Transformer (HEPT)☆24Updated 11 months ago
- PyTorch implementation for our ICLR 2024 paper "Diffusion Generative Flow Samplers: Improving learning signals through partial trajectory…☆26Updated 2 years ago
- Code for "SoftSort: A Continuous Relaxation for the argsort Operator", ICML 2020.☆53Updated 2 years ago
- The official PyTorch implementation of the paper: Xili Dai, Shengbang Tong, et al. "Closed-Loop Data Transcription to an LDR via Minimaxi…☆64Updated 3 years ago
- NeurIPS 2021, Code for Measuring Generalization with Optimal Transport☆28Updated 4 years ago
- Official PyTorch implementation of Hyperbolic Neural Networks++☆79Updated 4 years ago
- NeurIPS'22 Oral: EquiVSet - Learning Neural Set Functions Under the Optimal Subset Oracle☆21Updated 3 years ago
- Official implementation for Equivariant Architectures for Learning in Deep Weight Spaces [ICML 2023]☆90Updated 2 years ago
- The repository includes evidence that a published paper in TKDE shares surprisingly high similarity to our paper.☆32Updated 3 years ago
- Generative Modeling with Optimal Transport Maps - ICLR 2022☆61Updated 3 years ago
- Code for GFlowNet-EM, a novel algorithm for fitting latent variable models with compositional latents and an intractable true posterior.☆41Updated last year
- Efficient Conditionally Invariant Representation Learning (ICLR 2023, Oral)☆21Updated 3 years ago
- ☆33Updated 4 years ago
- Code for paper Fully Hyperbolic Neural Networks☆86Updated 3 years ago
- Code for the paper https://arxiv.org/abs/2205.14987v2☆57Updated last year
- ☆66Updated 4 years ago
- A PyTorch Implementation of Density Estimation Using Real NVP☆82Updated 4 years ago