ermongroup / neuralsort
Code for "Stochastic Optimization of Sorting Networks using Continuous Relaxations", ICLR 2019.
☆144Updated 2 years ago
Alternatives and similar repositories for neuralsort:
Users that are interested in neuralsort are comparing it to the libraries listed below
- [ICLR 2020] FSPool: Learning Set Representations with Featurewise Sort Pooling☆42Updated last year
- Code for Self-Tuning Networks (ICLR 2019) https://arxiv.org/abs/1903.03088☆53Updated 5 years ago
- pytorch implementation of VAE-Gumble-Softmax☆63Updated 4 years ago
- ☆75Updated 7 years ago
- ☆45Updated 4 years ago
- Differentiable bitonic sorting☆140Updated 4 years ago
- [AAAI 2020 Oral] Low-variance Black-box Gradient Estimates for the Plackett-Luce Distribution☆37Updated 4 years ago
- Autoregressive Energy Machines☆77Updated 2 years ago
- ☆122Updated 2 years ago
- Code for "SoftSort: A Continuous Relaxation for the argsort Operator", ICML 2020.☆51Updated 2 years ago
- LEARNING LATENT PERMUTATIONS WITH GUMBEL-SINKHORN NETWORKS IMPLEMENTATION WITH PYTORCH☆79Updated last year
- Memory efficient MAML using gradient checkpointing☆84Updated 5 years ago
- Low-variance and unbiased gradient for backpropagation through categorical random variables, with application in variational auto-encoder…☆17Updated 4 years ago
- Hypergradient descent☆146Updated 11 months ago
- This repository is no longer maintained. Check☆81Updated 5 years ago
- Efficient reservoir sampling implementation for PyTorch☆106Updated 3 years ago
- Reparameterize your PyTorch modules☆71Updated 4 years ago
- PyTorch re-implementation of parts of "Deep Sets" (NIPS 2017)☆69Updated 7 years ago
- [NeurIPS 2019] Deep Set Prediction Networks☆100Updated 4 years ago
- implements optimal transport algorithms in pytorch☆97Updated 2 years ago
- Mixed-curvature Variational Autoencoders (ICLR 2020)☆62Updated 4 years ago
- Optimization with orthogonal constraints and on general manifolds☆128Updated 4 years ago
- The Limited Multi-Label Projection Layer☆58Updated 9 months ago
- code for "Continuous Hierarchical Representations with Poincaré Variational Auto-Encoders".☆124Updated last year
- Testing Nerual Tangent Kernel (NTK) on small UCI datasets☆83Updated 5 years ago
- Low-variance, efficient and unbiased gradient estimation for optimizing models with binary latent variables. (ICLR 2019)☆28Updated 6 years ago
- Differentiable Sorting Networks☆114Updated last year
- [ICML 2020] Differentiating through the Fréchet Mean (https://arxiv.org/abs/2003.00335).☆55Updated 3 years ago
- An implementation of DIP-VAE from the paper "Variational Inference of Disentangled Latent Concepts from Unlabelled Observations" by Kumar…☆25Updated 7 years ago
- A variational inference method with accurate uncertainty estimation. It uses a new semi-implicit variational family built on neural netwo…☆54Updated 6 months ago