ermongroup / neuralsortLinks
Code for "Stochastic Optimization of Sorting Networks using Continuous Relaxations", ICLR 2019.
☆146Updated 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 7 years ago
- Reparameterize your PyTorch modules☆71Updated 4 years ago
- PyTorch re-implementation of parts of "Deep Sets" (NIPS 2017)☆72Updated 7 years ago
- ☆124Updated last year
- Hypergradient descent☆149Updated last year
- [ICLR 2020] FSPool: Learning Set Representations with Featurewise Sort Pooling☆42Updated last year
- Mixed-curvature Variational Autoencoders (ICLR 2020)☆65Updated 4 years ago
- LEARNING LATENT PERMUTATIONS WITH GUMBEL-SINKHORN NETWORKS IMPLEMENTATION WITH PYTORCH☆79Updated 2 years ago
- [AAAI 2020 Oral] Low-variance Black-box Gradient Estimates for the Plackett-Luce Distribution☆38Updated 4 years ago
- ☆44Updated 5 years ago
- implements optimal transport algorithms in pytorch☆100Updated 3 years ago
- Differentiable bitonic sorting☆141Updated 5 years ago
- Efficient reservoir sampling implementation for PyTorch☆106Updated 3 years ago
- Python toolbox for sampling Determinantal Point Processes☆231Updated last year
- Optimization with orthogonal constraints and on general manifolds☆130Updated 5 years ago
- Memory efficient MAML using gradient checkpointing☆86Updated 5 years ago
- a python implementation of various versions of the information bottleneck, including automated parameter searching☆128Updated 5 years ago
- Pytorch Implemetation for our NAACL2019 Paper "Riemannian Normalizing Flow on Variational Wasserstein Autoencoder for Text Modeling" http…☆63Updated 5 years ago
- [ICML 2020] Differentiating through the Fréchet Mean (https://arxiv.org/abs/2003.00335).☆58Updated 3 years ago
- [NeurIPS 2019] Deep Set Prediction Networks☆100Updated 4 years ago
- Code for "A Meta Transfer Objective For Learning To Disentangle Causal Mechanisms"☆127Updated 6 years ago
- Hyperbolic Neural Networks, pytorch☆87Updated 6 years ago
- pytorch implementation of VAE-Gumble-Softmax☆63Updated 5 years ago
- Python implementation of smooth optimal transport.☆60Updated 4 years ago
- ☆123Updated 2 years ago
- Convolutional Neural Tangent Kernel☆113Updated 5 years ago
- Source code for the paper "Hyperbolic Neural Networks", https://arxiv.org/abs/1805.09112☆179Updated 4 years ago
- Collection of algorithms for approximating Fisher Information Matrix for Natural Gradient (and second order method in general)☆140Updated 6 years ago
- Code for Reparameterizable Subset Sampling via Continuous Relaxations, IJCAI 2019.☆57Updated last year
- Optimizing control variates for black-box gradient estimation☆163Updated 6 years ago