perrying / gumbel-sinkhornLinks
PyTorch implementation of LEARNING LATENT PERMUTATIONS WITH GUMBEL-SINKHORN NETWORKS
☆30Updated 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:
- Differentiable Top-k Classification Learning☆83Updated 2 years ago
- LEARNING LATENT PERMUTATIONS WITH GUMBEL-SINKHORN NETWORKS IMPLEMENTATION WITH PYTORCH☆79Updated 2 years ago
- Pytorch code for "Improving Self-Supervised Learning by Characterizing Idealized Representations"☆41Updated 2 years ago
- [ICML'21] Improved Contrastive Divergence Training of Energy Based Models☆63Updated 3 years ago
- Deep Learning & Information Bottleneck☆61Updated 2 years ago
- Code for paper "Argmax Flows and Multinomial Diffusion: Learning Categorical Distributions"☆90Updated 4 years ago
- Official implementation for the paper: Permutation Invariant Graph Generation via Score-Based Generative Modeling☆111Updated 2 years ago
- Official implementation for Equivariant Architectures for Learning in Deep Weight Spaces [ICML 2023]☆89Updated last year
- Papers and Codes for the deep learning in hyperbolic space☆178Updated 3 years ago
- Official PyTorch implementation of Hyperbolic Neural Networks++☆72Updated 4 years ago
- ☆13Updated 2 years ago
- Code for our paper "Generative Flow Networks for Discrete Probabilistic Modeling"☆85Updated 2 years ago
- Transformers with doubly stochastic attention☆46Updated 2 years ago
- ☆65Updated 3 years ago
- Generative Modeling with Optimal Transport Maps - ICLR 2022☆60Updated 3 years ago
- Code for "Generalised Implicit Neural Representations" (NeurIPS 2022).☆71Updated 2 years ago
- Official implementation of Transformer Neural Processes☆78Updated 2 years ago
- Package for working with hypernetworks in PyTorch.☆128Updated last year
- Code for GFlowNet-EM, a novel algorithm for fitting latent variable models with compositional latents and an intractable true posterior.☆41Updated last year
- NeurIPS 2022: Tree Mover’s Distance: Bridging Graph Metrics and Stability of Graph Neural Networks☆37Updated 2 years ago
- Code for paper Fully Hyperbolic Neural Networks☆79Updated 2 years ago
- PyTorch repository for ICLR 2022 paper (GSAM) which improves generalization (e.g. +3.8% top-1 accuracy on ImageNet with ViT-B/32)☆144Updated 2 years ago
- Noise Contrastive Estimation (NCE) in PyTorch☆32Updated 5 months ago
- Pytorch implementation of neural processes and variants☆29Updated last year
- Modern Fixed Point Systems using Pytorch☆103Updated last year
- A PyTorch implementation of The ICLR 2019 paper "Invariant and Equivariant Graph Networks" by Haggai Maron, Heli Ben-Hamu, Nadav Shamir a…☆17Updated 3 years ago
- Official source code for "Graph Neural Networks for Learning Equivariant Representations of Neural Networks". In ICLR 2024 (oral).☆81Updated last year
- Code for "The Intrinsic Dimension of Images and Its Impact on Learning" - ICLR 2021 Spotlight https://openreview.net/forum?id=XJk19XzGq2J☆71Updated last year
- ☆22Updated 3 years ago
- ☆24Updated 3 years ago