Felix-Petersen / difftopk
Differentiable Top-k Classification Learning
☆80Updated 2 years ago
Alternatives and similar repositories for difftopk:
Users that are interested in difftopk are comparing it to the libraries listed below
- Pytorch code for "Improving Self-Supervised Learning by Characterizing Idealized Representations"☆40Updated 2 years ago
- Official Pytorch's implementation of RotoGrad☆72Updated last year
- PyTorch implementation of LEARNING LATENT PERMUTATIONS WITH GUMBEL-SINKHORN NETWORKS☆26Updated 5 years ago
- NeurIPS 2022: Tree Mover’s Distance: Bridging Graph Metrics and Stability of Graph Neural Networks☆36Updated last year
- Differentiable Sorting Networks☆112Updated last year
- LEARNING LATENT PERMUTATIONS WITH GUMBEL-SINKHORN NETWORKS IMPLEMENTATION WITH PYTORCH☆79Updated last year
- Official implementation for Equivariant Architectures for Learning in Deep Weight Spaces [ICML 2023]☆86Updated last year
- A Pytorch implementation of the optimal transport kernel embedding☆114Updated 4 years ago
- ☆57Updated 2 years ago
- [ICML2023] InfoOT: Information Maximizing Optimal Transport☆40Updated last year
- ☆47Updated 2 years ago
- Meta Optimal Transport☆100Updated last year
- NF-Layers for constructing neural functionals.☆82Updated last year
- PyTorch repository for ICLR 2022 paper (GSAM) which improves generalization (e.g. +3.8% top-1 accuracy on ImageNet with ViT-B/32)☆140Updated 2 years ago
- ☆70Updated 2 years ago
- This repository contains the code for our paper "Probabilistic Contrastive Learning Recovers the Correct Aleatoric Uncertainty of Ambiguo…☆40Updated last year
- ☆51Updated 9 months ago
- Code to reproduce the results for Compositional Attention☆60Updated 2 years ago
- ☆58Updated 3 years ago
- Code for "Generalised Implicit Neural Representations" (NeurIPS 2022).☆70Updated 2 years ago
- Whitening for Self-Supervised Representation Learning | Official repository☆129Updated 2 years ago
- implements optimal transport algorithms in pytorch☆93Updated 2 years ago
- Code for Neural Manifold Clustering and Embedding☆59Updated 3 years ago
- Implementation of the Gumbel-Sigmoid distribution in PyTorch.☆19Updated 2 years ago
- [NeurIPS 2022] Your Transformer May Not be as Powerful as You Expect (official implementation)☆34Updated last year
- Sequence Modeling with Multiresolution Convolutional Memory (ICML 2023)☆122Updated last year
- Code for "Can We Scale Transformers to Predict Parameters of Diverse ImageNet Models?" [ICML 2023]☆31Updated 6 months ago
- Code for paper Fully Hyperbolic Neural Networks☆77Updated 2 years ago
- Beyond Straight-Through☆94Updated last year
- NeurIPS 2021, Code for Measuring Generalization with Optimal Transport☆28Updated 3 years ago