sprillo / softsort
Code for "SoftSort: A Continuous Relaxation for the argsort Operator", ICML 2020.
☆47Updated 2 years ago
Alternatives and similar repositories for softsort:
Users that are interested in softsort are comparing it to the libraries listed below
- Code for "Stochastic Optimization of Sorting Networks using Continuous Relaxations", ICLR 2019.☆140Updated 2 years ago
- LEARNING LATENT PERMUTATIONS WITH GUMBEL-SINKHORN NETWORKS IMPLEMENTATION WITH PYTORCH☆79Updated last year
- implements optimal transport algorithms in pytorch☆94Updated 2 years ago
- Backpropagation-Friendly-Eigendecomposition☆72Updated 5 years ago
- [ICLR 2020] FSPool: Learning Set Representations with Featurewise Sort Pooling☆42Updated last year
- ☆31Updated 4 years ago
- Efficient reservoir sampling implementation for PyTorch☆107Updated 3 years ago
- Reparameterize your PyTorch modules☆70Updated 4 years ago
- PyTorch implementation of NICE☆33Updated 6 years ago
- ☆36Updated 4 years ago
- ☆122Updated 2 years ago
- Code for Reparameterizable Subset Sampling via Continuous Relaxations, IJCAI 2019.☆54Updated last year
- Differentiable Optimizers with Perturbations in Pytorch☆67Updated 4 years ago
- Mixed-curvature Variational Autoencoders (ICLR 2020)☆61Updated 4 years ago
- Low-variance, efficient and unbiased gradient estimation for optimizing models with binary latent variables. (ICLR 2019)☆28Updated 6 years ago
- A Pytorch implementation of the optimal transport kernel embedding☆116Updated 4 years ago
- ☆32Updated 5 years ago
- ☆24Updated 3 years ago
- Gradients as Features for Deep Representation Learning☆43Updated 5 years ago
- Learning To Stop While Learning To Predict☆34Updated 2 years ago
- ☆89Updated 3 years ago
- Official Implementation of Learning Diverse and Discriminative Representations via the Principle of Maximal Coding Rate Reduction (2020)☆198Updated 2 years ago
- ☆45Updated 4 years ago
- Official Pytorch's implementation of RotoGrad☆72Updated last year
- NeurIPS 2021, Code for Measuring Generalization with Optimal Transport☆28Updated 3 years ago
- Implementation of the Sliced Wasserstein Autoencoder using PyTorch☆101Updated 6 years ago
- PyTorch implementation of LEARNING LATENT PERMUTATIONS WITH GUMBEL-SINKHORN NETWORKS☆26Updated 5 years ago
- Codes for 'Deep Deterministic Information Bottleneck with Matrix-based entropy functional' in ICASSP 2021☆13Updated 2 years ago
- ☆44Updated 4 years ago
- [ICML 2020] Differentiating through the Fréchet Mean (https://arxiv.org/abs/2003.00335).☆55Updated 3 years ago