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
- Code to reproduce the results for Compositional Attention☆60Updated 2 years ago
- Differentiable Sorting Networks☆110Updated last year
- PyTorch implementation of LEARNING LATENT PERMUTATIONS WITH GUMBEL-SINKHORN NETWORKS☆26Updated 4 years ago
- Whitening for Self-Supervised Representation Learning | Official repository☆129Updated 2 years ago
- A Pytorch implementation of the optimal transport kernel embedding☆114Updated 4 years ago
- Learnable Fourier Features for Multi-Dimensional Spatial Positional Encoding☆45Updated 4 months ago
- Code for paper Fully Hyperbolic Neural Networks☆77Updated 2 years ago
- ☆45Updated 2 years ago
- ☆57Updated 2 years ago
- NeurIPS 2022: Tree Mover’s Distance: Bridging Graph Metrics and Stability of Graph Neural Networks☆36Updated last year
- Code for "The Intrinsic Dimension of Images and Its Impact on Learning" - ICLR 2021 Spotlight https://openreview.net/forum?id=XJk19XzGq2J☆65Updated 10 months ago
- ☆58Updated 3 years ago
- [NeurIPS 2022] Your Transformer May Not be as Powerful as You Expect (official implementation)☆34Updated last year
- Code for "SAM as an Optimal Relaxation of Bayes", ICLR 2023.☆25Updated last year
- This repository contains the code for our paper "Probabilistic Contrastive Learning Recovers the Correct Aleatoric Uncertainty of Ambiguo…☆39Updated last year
- Transformers with doubly stochastic attention☆45Updated 2 years ago
- Code for Reparameterizable Subset Sampling via Continuous Relaxations, IJCAI 2019.☆54Updated 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)☆139Updated 2 years ago
- ☆63Updated 2 years ago
- Framework code with wandb, checkpointing, logging, configs, experimental protocols. Useful for fine-tuning models or training from scratc…☆148Updated 2 years ago
- Official implementation for Equivariant Architectures for Learning in Deep Weight Spaces [ICML 2023]☆86Updated last year
- Reparameterize your PyTorch modules☆70Updated 4 years ago
- ☆51Updated 8 months 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
- Code for "Can We Scale Transformers to Predict Parameters of Diverse ImageNet Models?" [ICML 2023]☆31Updated 5 months ago
- Official code for the ICML 2024 paper "The Entropy Enigma: Success and Failure of Entropy Minimization"☆48Updated 8 months ago
- [NeurIPS'20] Code for the Paper Compositional Visual Generation and Inference with Energy Based Models☆44Updated last year
- [ICML2023] InfoOT: Information Maximizing Optimal Transport☆40Updated last year
- Code for Neural Manifold Clustering and Embedding☆57Updated 2 years ago