google / gumbel_sinkhorn
☆74Updated 7 years ago
Alternatives and similar repositories for gumbel_sinkhorn:
Users that are interested in gumbel_sinkhorn are comparing it to the libraries listed below
- [ICLR 2019] Learning Representations of Sets through Optimized Permutations☆36Updated 5 years ago
- [ICLR 2020] FSPool: Learning Set Representations with Featurewise Sort Pooling☆42Updated last year
- PyTorch re-implementation of parts of "Deep Sets" (NIPS 2017)☆68Updated 6 years ago
- [NeurIPS 2019] Deep Set Prediction Networks☆100Updated 4 years ago
- Code in PyTorch for the convex combination linear IAF and the Householder Flow, J.M. Tomczak & M. Welling☆91Updated 7 years ago
- LEARNING LATENT PERMUTATIONS WITH GUMBEL-SINKHORN NETWORKS IMPLEMENTATION WITH PYTORCH☆79Updated last year
- An implementation of a Variational-Autoencoder using the Gumbel-Softmax reparametrization trick in TensorFlow (tested on r1.5 CPU and GPU…☆73Updated 6 years ago
- [ICML 2020] Differentiating through the Fréchet Mean (https://arxiv.org/abs/2003.00335).☆54Updated 3 years ago
- Recurrent Back Propagation, Back Propagation Through Optimization, ICML 2018☆41Updated 6 years ago
- Mixed-curvature Variational Autoencoders (ICLR 2020)☆61Updated 4 years ago
- SparseMax activation function implementation (ICML 2016) (PyTorch)☆27Updated 7 years ago
- Estimating Gradients for Discrete Random Variables by Sampling without Replacement☆40Updated 5 years ago
- Code to build VAE models that are jointly conditioned.☆36Updated 7 years ago
- A variational inference method with accurate uncertainty estimation. It uses a new semi-implicit variational family built on neural netwo…☆53Updated 4 months ago
- Code for "Stochastic Optimization of Sorting Networks using Continuous Relaxations", ICLR 2019.☆138Updated last year
- Differentiable bitonic sorting☆138Updated 4 years ago
- implements optimal transport algorithms in pytorch☆93Updated 2 years ago
- code for "Continuous Hierarchical Representations with Poincaré Variational Auto-Encoders".☆123Updated last year
- Code for Self-Tuning Networks (ICLR 2019) https://arxiv.org/abs/1903.03088☆53Updated 5 years ago
- Implementation of the paper "Direct Optimization through argmax for discrete Variational Auto-Encoder"☆14Updated 4 years ago
- Equivalence Between Structural Representations and Positional Node Embeddings☆21Updated 5 years ago
- simple implementation of "Improved Variational Inference with Inverse Autoregressive Flow" paper with pytorch☆52Updated 7 years ago
- Scalable Training of Inference Networks for Gaussian-Process Models, ICML 2019☆41Updated 2 years ago
- ICML 2020 Paper: Latent Variable Modelling with Hyperbolic Normalizing Flows☆53Updated 2 years ago
- Notes from NeurIPS 2019☆29Updated 5 years ago
- Weekly reading group on Graphs at Mila☆31Updated 5 years ago
- An implementation of DIP-VAE from the paper "Variational Inference of Disentangled Latent Concepts from Unlabelled Observations" by Kumar…☆24Updated 6 years ago
- SparseMAP: differentiable sparse structure inference☆111Updated 6 years ago
- Implementation of iterative inference in deep latent variable models☆43Updated 5 years ago
- A public repository for our paper, Rao-Blackwellized Stochastic Gradients for Discrete Distributions☆22Updated 5 years ago