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:
- PyTorch re-implementation of parts of "Deep Sets" (NIPS 2017)☆72Updated 7 years ago
- ☆76Updated 8 years ago
- ☆124Updated last year
- [ICLR 2020] FSPool: Learning Set Representations with Featurewise Sort Pooling☆42Updated 2 years ago
- a python implementation of various versions of the information bottleneck, including automated parameter searching☆128Updated 5 years ago
- Reparameterize your PyTorch modules☆71Updated 4 years ago
- ☆124Updated 2 years ago
- Optimization with orthogonal constraints and on general manifolds☆130Updated 5 years ago
- Code for "A Meta Transfer Objective For Learning To Disentangle Causal Mechanisms"☆127Updated 6 years ago
- Source code for the paper "Hyperbolic Neural Networks", https://arxiv.org/abs/1805.09112☆179Updated 4 years ago
- Convolutional Neural Tangent Kernel☆112Updated 5 years ago
- ☆44Updated 5 years ago
- [AAAI 2020 Oral] Low-variance Black-box Gradient Estimates for the Plackett-Luce Distribution☆38Updated 4 years ago
- Collection of algorithms for approximating Fisher Information Matrix for Natural Gradient (and second order method in general)☆140Updated 6 years ago
- Hypergradient descent☆149Updated last year
- Python toolbox for sampling Determinantal Point Processes☆233Updated last year
- LEARNING LATENT PERMUTATIONS WITH GUMBEL-SINKHORN NETWORKS IMPLEMENTATION WITH PYTORCH☆79Updated 2 years ago
- Code for Self-Tuning Networks (ICLR 2019) https://arxiv.org/abs/1903.03088☆54Updated 6 years ago
- Python implementation of smooth optimal transport.☆60Updated 4 years ago
- Mixed-curvature Variational Autoencoders (ICLR 2020)☆65Updated 4 years ago
- A variational inference method with accurate uncertainty estimation. It uses a new semi-implicit variational family built on neural netwo…☆54Updated 11 months ago
- Pytorch Implemetation for our NAACL2019 Paper "Riemannian Normalizing Flow on Variational Wasserstein Autoencoder for Text Modeling" http…☆63Updated 5 years ago
- implements optimal transport algorithms in pytorch☆100Updated 3 years ago
- A manifold optimization library for deep learning☆249Updated 4 years ago
- [ICML 2020] Differentiating through the Fréchet Mean (https://arxiv.org/abs/2003.00335).☆59Updated 4 years ago
- A minimal pytorch implementation of VAE, IWAE, MIWAE☆48Updated 2 years ago
- [NeurIPS 2019] Deep Set Prediction Networks☆101Updated 5 years ago
- PyTorch implementation of Neural Processes☆89Updated 6 years ago
- Hyperbolic Neural Networks, pytorch☆87Updated 6 years ago
- ☆36Updated 4 years ago