perrying / gumbel-sinkhorn
PyTorch implementation of LEARNING LATENT PERMUTATIONS WITH GUMBEL-SINKHORN NETWORKS
☆25Updated 4 years ago
Related projects ⓘ
Alternatives and complementary repositories for gumbel-sinkhorn
- Code for paper "Argmax Flows and Multinomial Diffusion: Learning Categorical Distributions"☆83Updated 3 years ago
- NeurIPS 2021, Code for Measuring Generalization with Optimal Transport☆28Updated 3 years ago
- ☆49Updated 3 years ago
- Transformers with doubly stochastic attention☆40Updated 2 years ago
- [ICML'21] Improved Contrastive Divergence Training of Energy Based Models☆62Updated 2 years ago
- ☆33Updated 4 years ago
- LEARNING LATENT PERMUTATIONS WITH GUMBEL-SINKHORN NETWORKS IMPLEMENTATION WITH PYTORCH☆76Updated last year
- Pytorch code for "Improving Self-Supervised Learning by Characterizing Idealized Representations"☆40Updated last year
- Code for GFlowNet-EM, a novel algorithm for fitting latent variable models with compositional latents and an intractable true posterior.☆38Updated 9 months ago
- A PyTorch implementation of The ICLR 2019 paper "Invariant and Equivariant Graph Networks" by Haggai Maron, Heli Ben-Hamu, Nadav Shamir a…☆15Updated 2 years ago
- Code repository for the paper "Group Equivariant Stand-Alone Self Attention For Vision" published at ICLR 2021. https://openreview.net/fo…☆28Updated 3 years ago
- ☆205Updated last year
- Official repository for "Categorical Normalizing Flows via Continuous Transformations"☆55Updated 3 years ago
- Code for the paper https://arxiv.org/abs/2205.14987v2☆43Updated 6 months ago
- PyTorch implementation for our ICLR 2024 paper "Diffusion Generative Flow Samplers: Improving learning signals through partial trajectory…☆21Updated 10 months ago
- Code for "Generalised Implicit Neural Representations" (NeurIPS 2022).☆68Updated last year
- Official code for "Maximum Likelihood Training of Score-Based Diffusion Models", NeurIPS 2021 (spotlight)☆134Updated 2 years ago
- ☆31Updated 4 years ago
- ☆61Updated last year
- Code for "Implicit Normalizing Flows" (ICLR 2021 spotlight)☆34Updated 3 years ago
- PyTorch implementation of the paper "Discovering and Explaining the Representation Bottleneck of DNNs" (ICLR 2022)☆37Updated last week
- Weighted Training for Cross-Task Learning☆15Updated last year
- Official implementation for the paper: Permutation Invariant Graph Generation via Score-Based Generative Modeling☆106Updated last year
- Differentiable Top-k Classification Learning☆72Updated last year
- ☆16Updated last year
- NeurIPS'23: Energy Discrepancies: A Score-Independent Loss for Energy-Based Models☆11Updated 2 weeks ago
- Noise Contrastive Estimation (NCE) in PyTorch☆31Updated last year
- ☆18Updated last year
- ☆15Updated 2 years ago
- ☆13Updated 7 months ago