sprillo / softsort
Code for "SoftSort: A Continuous Relaxation for the argsort Operator", ICML 2020.
☆44Updated 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
- implements optimal transport algorithms in pytorch☆92Updated 2 years ago
- Code for Reparameterizable Subset Sampling via Continuous Relaxations, IJCAI 2019.☆54Updated last year
- [ICML 2020] Differentiating through the Fréchet Mean (https://arxiv.org/abs/2003.00335).☆54Updated 3 years ago
- ☆31Updated 4 years ago
- ☆35Updated 4 years ago
- Implementation of Inexact Proximal point method for Optimal Transport☆47Updated 3 years ago
- ☆39Updated 4 years ago
- Code for "Online Learned Continual Compression with Adaptive Quantization Modules"☆27Updated 4 years ago
- ☆87Updated 3 years ago
- Reparameterize your PyTorch modules☆70Updated 4 years ago
- Mixed-curvature Variational Autoencoders (ICLR 2020)☆60Updated 3 years ago
- Code for Sliced Gromov-Wasserstein☆66Updated 5 years ago
- Official Pytorch's implementation of RotoGrad☆72Updated last year
- ICLR22 "Fast Differentiable Matrix Square Root" and T-PAMI extension☆59Updated 2 months ago
- Source code for the ICML2019 paper "Subspace Robust Wasserstein Distances"☆28Updated 5 years ago
- A Pytorch implementation of the optimal transport kernel embedding☆114Updated 3 years ago
- Backpropagation-Friendly-Eigendecomposition☆71Updated 5 years ago
- ICLR 2021, Fair Mixup: Fairness via Interpolation☆55Updated 3 years ago
- Official implementation of paper Gradient Matching for Domain Generalization☆117Updated 3 years ago
- Codes for CyGen, the novel generative modeling framework proposed in "On the Generative Utility of Cyclic Conditionals" (NeurIPS-21)☆45Updated 2 years ago
- Robust Optimal Transport code☆40Updated 2 years ago
- [ICLR 2020] FSPool: Learning Set Representations with Featurewise Sort Pooling☆42Updated last year
- Code for "Stochastic Optimization of Sorting Networks using Continuous Relaxations", ICLR 2019.☆136Updated last year
- Code for the ICML 2021 paper "Bridging Multi-Task Learning and Meta-Learning: Towards Efficient Training and Effective Adaptation", Haoxi…☆67Updated 3 years ago
- NeurIPS 2021, Code for Measuring Generalization with Optimal Transport☆28Updated 3 years ago
- ☆32Updated 5 years ago
- Low-variance, efficient and unbiased gradient estimation for optimizing models with binary latent variables. (ICLR 2019)☆28Updated 5 years ago
- ☆24Updated 3 years ago
- ☆44Updated 4 years ago