ermongroup / neuralsort
Code for "Stochastic Optimization of Sorting Networks using Continuous Relaxations", ICLR 2019.
☆137Updated last year
Alternatives and similar repositories for neuralsort:
Users that are interested in neuralsort are comparing it to the libraries listed below
- [NeurIPS 2019] Deep Set Prediction Networks☆100Updated 4 years ago
- Code for Self-Tuning Networks (ICLR 2019) https://arxiv.org/abs/1903.03088☆53Updated 5 years ago
- Optimization with orthogonal constraints and on general manifolds☆126Updated 4 years ago
- Mixed-curvature Variational Autoencoders (ICLR 2020)☆59Updated 3 years ago
- This repository is no longer maintained. Check☆81Updated 4 years ago
- LEARNING LATENT PERMUTATIONS WITH GUMBEL-SINKHORN NETWORKS IMPLEMENTATION WITH PYTORCH☆79Updated last year
- PyTorch re-implementation of parts of "Deep Sets" (NIPS 2017)☆68Updated 6 years ago
- code for "Continuous Hierarchical Representations with Poincaré Variational Auto-Encoders".☆123Updated last year
- ☆45Updated 4 years ago
- [AAAI 2020 Oral] Low-variance Black-box Gradient Estimates for the Plackett-Luce Distribution☆37Updated 4 years ago
- [ICLR 2020] FSPool: Learning Set Representations with Featurewise Sort Pooling☆42Updated last year
- Reparameterize your PyTorch modules☆70Updated 4 years ago
- ☆74Updated 7 years ago
- Experiments for Meta-Learning Symmetries by Reparameterization☆56Updated 3 years ago
- Differentiable bitonic sorting☆138Updated 4 years ago
- Pytorch Implemetation for our NAACL2019 Paper "Riemannian Normalizing Flow on Variational Wasserstein Autoencoder for Text Modeling" http…☆62Updated 4 years ago
- Implementation of the Functional Neural Process models☆43Updated 4 years ago
- Source code for the paper "Hyperbolic Neural Networks", https://arxiv.org/abs/1805.09112☆172Updated 4 years ago
- ☆65Updated 5 months ago
- Optimizing control variates for black-box gradient estimation☆162Updated 5 years ago
- [ICML 2020] Differentiating through the Fréchet Mean (https://arxiv.org/abs/2003.00335).☆54Updated 3 years ago
- Hypergradient descent☆143Updated 7 months ago
- Code for "A Spectral Approach to Gradient Estimation for Implicit Distributions" (ICML'18)☆32Updated last year
- implements optimal transport algorithms in pytorch☆92Updated 2 years ago
- pytorch implementation of VAE-Gumble-Softmax☆62Updated 4 years ago
- Autoregressive Energy Machines☆77Updated 2 years ago
- ☆120Updated 7 months ago
- PyTorch-SSO: Scalable Second-Order methods in PyTorch☆142Updated last year
- Discrete Normalizing Flows implemented in PyTorch☆108Updated 3 years ago
- Code for "SoftSort: A Continuous Relaxation for the argsort Operator", ICML 2020.☆43Updated last year