sprillo / softsort
Code for "SoftSort: A Continuous Relaxation for the argsort Operator", ICML 2020.
☆46Updated last year
Alternatives and similar repositories for softsort:
Users that are interested in softsort are comparing it to the libraries listed below
- LEARNING LATENT PERMUTATIONS WITH GUMBEL-SINKHORN NETWORKS IMPLEMENTATION WITH PYTORCH☆79Updated last year
- Code for Reparameterizable Subset Sampling via Continuous Relaxations, IJCAI 2019.☆54Updated last year
- Backpropagation-Friendly-Eigendecomposition☆72Updated 5 years ago
- implements optimal transport algorithms in pytorch☆93Updated 2 years ago
- ☆45Updated 4 years ago
- Differentiable Optimizers with Perturbations in Pytorch☆66Updated 3 years ago
- Reparameterize your PyTorch modules☆70Updated 4 years ago
- ☆31Updated 4 years ago
- ☆39Updated 5 years ago
- Hyperspherical Prototype Networks☆66Updated 4 years ago
- Differentiable Top-k Classification Learning☆79Updated 2 years ago
- NeurIPS 2021, Code for Measuring Generalization with Optimal Transport☆28Updated 3 years ago
- [ICML 2020] Differentiating through the Fréchet Mean (https://arxiv.org/abs/2003.00335).☆54Updated 3 years ago
- Code for "Stochastic Optimization of Sorting Networks using Continuous Relaxations", ICLR 2019.☆138Updated last year
- Matrix square root with gradient support for PyTorch☆67Updated 2 years ago
- Efficient reservoir sampling implementation for PyTorch☆107Updated 3 years ago
- Mixed-curvature Variational Autoencoders (ICLR 2020)☆61Updated 4 years ago
- ☆122Updated 2 years ago
- ☆24Updated 3 years ago
- Official Implementation of Learning Diverse and Discriminative Representations via the Principle of Maximal Coding Rate Reduction (2020)☆198Updated 2 years ago
- ☆35Updated 4 years ago
- [ICLR 2020] FSPool: Learning Set Representations with Featurewise Sort Pooling☆42Updated last year
- A Pytorch implementation of the optimal transport kernel embedding☆114Updated 4 years ago
- [ICLR 2021] "GANs Can Play Lottery Too" by Xuxi Chen, Zhenyu Zhang, Yongduo Sui, Tianlong Chen☆26Updated 3 years ago
- ICLR22 "Fast Differentiable Matrix Square Root" and T-PAMI extension☆59Updated 3 months ago
- Gradients as Features for Deep Representation Learning☆43Updated 4 years ago
- Implementation of Inexact Proximal point method for Optimal Transport☆47Updated 3 years ago
- Code for the ICML 2021 paper "Bridging Multi-Task Learning and Meta-Learning: Towards Efficient Training and Effective Adaptation", Haoxi…☆67Updated 3 years ago
- Winning Solution of the NeurIPS 2020 Competition on Predicting Generalization in Deep Learning☆38Updated 3 years ago
- Code for "Online Learned Continual Compression with Adaptive Quantization Modules"☆27Updated 4 years ago