sprillo / softsortLinks
Code for "SoftSort: A Continuous Relaxation for the argsort Operator", ICML 2020.
☆52Updated 2 years ago
Alternatives and similar repositories for softsort
Users that are interested in softsort are comparing it to the libraries listed below
Sorting:
- implements optimal transport algorithms in pytorch☆97Updated 3 years ago
- Differentiable Top-k Classification Learning☆82Updated 2 years ago
- LEARNING LATENT PERMUTATIONS WITH GUMBEL-SINKHORN NETWORKS IMPLEMENTATION WITH PYTORCH☆79Updated last year
- [ICML 2020] Differentiating through the Fréchet Mean (https://arxiv.org/abs/2003.00335).☆56Updated 3 years ago
- Code for "Stochastic Optimization of Sorting Networks using Continuous Relaxations", ICLR 2019.☆145Updated 2 years ago
- Differentiable Optimizers with Perturbations in Pytorch☆68Updated 4 years ago
- Mixed-curvature Variational Autoencoders (ICLR 2020)☆62Updated 4 years ago
- Backpropagation-Friendly-Eigendecomposition☆72Updated 5 years ago
- Official Pytorch's implementation of RotoGrad☆78Updated last year
- ☆122Updated 2 years ago
- Reparameterize your PyTorch modules☆71Updated 4 years ago
- Code for Reparameterizable Subset Sampling via Continuous Relaxations, IJCAI 2019.☆56Updated last year
- ☆41Updated 2 years ago
- A Pytorch implementation of the optimal transport kernel embedding☆116Updated 4 years ago
- NeurIPS 2021, Code for Measuring Generalization with Optimal Transport☆28Updated 3 years ago
- Efficient reservoir sampling implementation for PyTorch☆106Updated 3 years ago
- [ICLR 2020] FSPool: Learning Set Representations with Featurewise Sort Pooling☆42Updated last year
- Hyperbolic Neural Networks, pytorch☆86Updated 5 years ago
- ☆36Updated 4 years ago
- [ICML 2020] code for the flooding regularizer proposed in "Do We Need Zero Training Loss After Achieving Zero Training Error?"☆92Updated 2 years ago
- ☆45Updated 4 years ago
- ☆32Updated 6 years ago
- Meta Optimal Transport☆102Updated 2 years ago
- ☆30Updated 4 years ago
- A neural network architecture for prediction on sets☆22Updated 3 years ago
- Low-variance and unbiased gradient for backpropagation through categorical random variables, with application in variational auto-encoder…☆17Updated 4 years ago
- This repository is no longer maintained. Check☆81Updated 5 years ago
- Estimating Gradients for Discrete Random Variables by Sampling without Replacement☆40Updated 5 years ago
- Hyperspherical Prototype Networks☆67Updated 5 years ago
- ☆46Updated 2 years ago