Alternatives and similar repositories for gumbel-sinkhorn

Users that are interested in gumbel-sinkhorn are comparing it to the libraries listed below

• didriknielsen / argmax_flows
  Code for paper "Argmax Flows and Multinomial Diffusion: Learning Categorical Distributions"
  ☆90Updated 4 years ago
• ermongroup / GraphScoreMatching
  Official implementation for the paper: Permutation Invariant Graph Generation via Score-Based Generative Modeling
  ☆111Updated 2 years ago
• HeddaCohenIndelman / Learning-Gumbel-Sinkhorn-Permutations-w-Pytorch
  LEARNING LATENT PERMUTATIONS WITH GUMBEL-SINKHORN NETWORKS IMPLEMENTATION WITH PYTORCH
  ☆79Updated 2 years ago
• yilundu / improved_contrastive_divergence
  [ICML'21] Improved Contrastive Divergence Training of Energy Based Models
  ☆63Updated 3 years ago
• YannDubs / Invariant-Self-Supervised-Learning
  Pytorch code for "Improving Self-Supervised Learning by Characterizing Idealized Representations"
  ☆41Updated 2 years ago
• GFNOrg / EB_GFN
  Code for our paper "Generative Flow Networks for Discrete Probabilistic Modeling"
  ☆84Updated 2 years ago
• gnobitab / ConstrainedDiffusionBridge
  ☆13Updated 2 years ago
• xiaoiker / Awesome-Hyperbolic-NeuralNetworks
  Papers and Codes for the deep learning in hyperbolic space
  ☆180Updated 3 years ago
• mil-tokyo / hyperbolic_nn_plusplus
  Official PyTorch implementation of Hyperbolic Neural Networks++
  ☆73Updated 4 years ago
• Graph-COM / HEPT
  [ICML 2024 Oral] LSH-Based Efficient Point Transformer (HEPT)
  ☆21Updated 7 months ago
• SubsetSelection / EquiVSet
  NeurIPS'22 Oral: EquiVSet - Learning Neural Set Functions Under the Optimal Subset Oracle
  ☆20Updated 2 years ago
• Felix-Petersen / difftopk
  Differentiable Top-k Classification Learning
  ☆85Updated 2 years ago
• michaelsdr / sinkformers
  Transformers with doubly stochastic attention
  ☆47Updated 2 years ago
• bpucla / latent-space-EBM-prior
  ☆37Updated 4 years ago
• AvivNavon / DWSNets
  Official implementation for Equivariant Architectures for Learning in Deep Weight Spaces [ICML 2023]
  ☆89Updated 2 years ago
• volagold / nce
  Noise Contrastive Estimation (NCE) in PyTorch
  ☆32Updated 6 months ago
• chenweize1998 / fully-hyperbolic-nn
  Code for paper Fully Hyperbolic Neural Networks
  ☆81Updated 2 years ago
• xu-ji / information-bottleneck
  Deep Learning & Information Bottleneck
  ☆61Updated 2 years ago
• J-zin / energy-discrepancy
  NeurIPS'23: Energy Discrepancies: A Score-Independent Loss for Energy-Based Models
  ☆17Updated 10 months ago
• danielegrattarola / GINR
  Code for "Generalised Implicit Neural Representations" (NeurIPS 2022).
  ☆71Updated 2 years ago
• Johswald / awesome-hypernetworks
  ☆65Updated 3 years ago
• VinAIResearch / DSW
  Distributional Sliced-Wasserstein distance code
  ☆50Updated last year
• locuslab / deq-ddim
  ☆62Updated 2 years ago
• tung-nd / TNP-pytorch
  Official implementation of Transformer Neural Processes
  ☆78Updated 3 years ago
• kimiandj / gsw
  ☆41Updated 5 years ago
• wgrathwohl / GWG_release
  Official release of code for "Oops I Took A Gradient: Scalable Sampling for Discrete Distributions"
  ☆56Updated 2 years ago
• HyTruongSon / InvariantGraphNetworks-PyTorch
  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
• CogComp / TAWT
  Weighted Training for Cross-Task Learning
  ☆15Updated 2 years ago
• Noahs-ARK / RFA
  ☆33Updated 4 years ago
• saleml / continuous-gfn
  ☆26Updated 2 years ago