ermongroup / neuralsortLinks
Code for "Stochastic Optimization of Sorting Networks using Continuous Relaxations", ICLR 2019.
☆148Updated 2 years ago
Alternatives and similar repositories for neuralsort
Users that are interested in neuralsort are comparing it to the libraries listed below
Sorting:
- ☆76Updated 8 years ago
- PyTorch re-implementation of parts of "Deep Sets" (NIPS 2017)☆73Updated 7 years ago
- [ICLR 2020] FSPool: Learning Set Representations with Featurewise Sort Pooling☆41Updated 2 years ago
- ☆126Updated last year
- ☆125Updated 2 years ago
- Mixed-curvature Variational Autoencoders (ICLR 2020)☆65Updated 4 years ago
- implements optimal transport algorithms in pytorch☆102Updated 3 years ago
- Sample code for running deterministic variational inference to train Bayesian neural networks☆102Updated 7 years ago
- Reparameterize your PyTorch modules☆71Updated 4 years ago
- A variational inference method with accurate uncertainty estimation. It uses a new semi-implicit variational family built on neural netwo…☆54Updated last year
- Hypergradient descent☆147Updated last year
- [ICML 2020] Differentiating through the Fréchet Mean (https://arxiv.org/abs/2003.00335).☆59Updated 4 years ago
- Pytorch Implemetation for our NAACL2019 Paper "Riemannian Normalizing Flow on Variational Wasserstein Autoencoder for Text Modeling" http…☆63Updated 5 years ago
- [AAAI 2020 Oral] Low-variance Black-box Gradient Estimates for the Plackett-Luce Distribution☆39Updated 4 years ago
- [NeurIPS 2019] Deep Set Prediction Networks☆101Updated 5 years ago
- a python implementation of various versions of the information bottleneck, including automated parameter searching☆130Updated 5 years ago
- Code for Self-Tuning Networks (ICLR 2019) https://arxiv.org/abs/1903.03088☆60Updated 6 years ago
- LEARNING LATENT PERMUTATIONS WITH GUMBEL-SINKHORN NETWORKS IMPLEMENTATION WITH PYTORCH☆81Updated 2 years ago
- A minimal pytorch implementation of VAE, IWAE, MIWAE☆47Updated 3 years ago
- Python implementation of smooth optimal transport.☆61Updated 4 years ago
- Code for "A Meta Transfer Objective For Learning To Disentangle Causal Mechanisms"☆127Updated 6 years ago
- code for "Continuous Hierarchical Representations with Poincaré Variational Auto-Encoders".☆129Updated 2 years ago
- Optimization with orthogonal constraints and on general manifolds☆130Updated 5 years ago
- Optimizing control variates for black-box gradient estimation☆163Updated 6 years ago
- Source code for the paper "Hyperbolic Neural Networks", https://arxiv.org/abs/1805.09112☆182Updated 5 years ago
- ☆44Updated 5 years ago
- Pytorch implementation of Hyperspherical Variational Auto-Encoders☆380Updated 5 years ago
- Code for Reparameterizable Subset Sampling via Continuous Relaxations, IJCAI 2019.☆57Updated 2 years ago
- Deep Generative Models with Stick-Breaking Priors☆96Updated 9 years ago
- PyTorch-SSO: Scalable Second-Order methods in PyTorch☆148Updated 2 years ago