Felix-Petersen / difftopkLinks
Differentiable Top-k Classification Learning
☆89Updated 2 years ago
Alternatives and similar repositories for difftopk
Users that are interested in difftopk are comparing it to the libraries listed below
Sorting:
- Learnable Fourier Features for Multi-Dimensional Spatial Positional Encoding☆55Updated last year
- PyTorch implementation of LEARNING LATENT PERMUTATIONS WITH GUMBEL-SINKHORN NETWORKS☆31Updated 5 years ago
- Code for "SoftSort: A Continuous Relaxation for the argsort Operator", ICML 2020.☆53Updated 2 years ago
- Differentiable Sorting Networks☆125Updated 2 years ago
- NeurIPS 2022: Tree Mover’s Distance: Bridging Graph Metrics and Stability of Graph Neural Networks☆37Updated 2 years ago
- Pytorch code for "Improving Self-Supervised Learning by Characterizing Idealized Representations"☆41Updated 3 years ago
- Official implementation for Equivariant Architectures for Learning in Deep Weight Spaces [ICML 2023]☆90Updated 2 years ago
- Code for paper Fully Hyperbolic Neural Networks☆85Updated 2 years ago
- PyTorch repository for ICLR 2022 paper (GSAM) which improves generalization (e.g. +3.8% top-1 accuracy on ImageNet with ViT-B/32)☆144Updated 3 years ago
- [ICML2023] InfoOT: Information Maximizing Optimal Transport☆42Updated 2 years ago
- Meta Optimal Transport☆105Updated 2 years ago
- Papers and Codes for the deep learning in hyperbolic space☆185Updated 3 years ago
- Official Pytorch's implementation of RotoGrad☆80Updated 2 years ago
- Code repository for the paper "Group Equivariant Stand-Alone Self Attention For Vision" published at ICLR 2021. https://openreview.net/fo…☆29Updated 4 years ago
- Differentiable Optimizers with Perturbations in Pytorch☆69Updated 4 years ago
- LEARNING LATENT PERMUTATIONS WITH GUMBEL-SINKHORN NETWORKS IMPLEMENTATION WITH PYTORCH☆81Updated 2 years ago
- Framework code with wandb, checkpointing, logging, configs, experimental protocols. Useful for fine-tuning models or training from scratc…☆152Updated 2 years ago
- Official source code for "Graph Neural Networks for Learning Equivariant Representations of Neural Networks". In ICLR 2024 (oral).☆84Updated last year
- [ICML 2020] Differentiating through the Fréchet Mean (https://arxiv.org/abs/2003.00335).☆59Updated 4 years ago
- Official implementation of ORCA proposed in the paper "Cross-Modal Fine-Tuning: Align then Refine"☆74Updated last year
- Code for Neural Manifold Clustering and Embedding☆61Updated 3 years ago
- Code for "Generalised Implicit Neural Representations" (NeurIPS 2022).☆70Updated 2 years ago
- TLDR is an unsupervised dimensionality reduction method that combines neighborhood embedding learning with the simplicity and effectivene…☆126Updated 3 years ago
- The repository for Hyperbolic Representation Learning for Computer Vision, ECCV 2022☆66Updated 3 years ago
- Code used in "Understanding Dimensional Collapse in Contrastive Self-supervised Learning" paper.☆79Updated 3 years ago
- Implementation of the Gumbel-Sigmoid distribution in PyTorch.☆21Updated 3 years ago
- Code to reproduce the results for Compositional Attention☆59Updated 3 years ago
- ☆58Updated 2 years ago
- Deep Learning & Information Bottleneck☆62Updated 2 years ago
- A Pytorch implementation of the optimal transport kernel embedding☆118Updated 4 years ago