Felix-Petersen / difftopk
Differentiable Top-k Classification Learning
☆81Updated 2 years ago
Alternatives and similar repositories for difftopk:
Users that are interested in difftopk are comparing it to the libraries listed below
- PyTorch implementation of LEARNING LATENT PERMUTATIONS WITH GUMBEL-SINKHORN NETWORKS☆26Updated 5 years ago
- Differentiable Sorting Networks☆114Updated last year
- Code for "Can We Scale Transformers to Predict Parameters of Diverse ImageNet Models?" [ICML 2023]☆32Updated 7 months ago
- NeurIPS 2022: Tree Mover’s Distance: Bridging Graph Metrics and Stability of Graph Neural Networks☆36Updated last year
- Pytorch code for "Improving Self-Supervised Learning by Characterizing Idealized Representations"☆41Updated 2 years ago
- ☆51Updated 10 months ago
- PyTorch repository for ICLR 2022 paper (GSAM) which improves generalization (e.g. +3.8% top-1 accuracy on ImageNet with ViT-B/32)☆143Updated 2 years ago
- Official Pytorch's implementation of RotoGrad☆76Updated last year
- Reparameterize your PyTorch modules☆71Updated 4 years ago
- Code for "SAM as an Optimal Relaxation of Bayes", ICLR 2023.☆25Updated last year
- Model Zoos published at the NeurIPS 2022 Dataset & Benchmark track: "Model Zoos: A Dataset of Diverse Populations of Neural Network Model…☆55Updated last year
- ☆70Updated 2 years ago
- A Pytorch implementation of the optimal transport kernel embedding☆116Updated 4 years ago
- [NeurIPS 2021] Code for Unsupervised Learning of Compositional Energy Concepts☆61Updated 2 years ago
- The repository for Hyperbolic Representation Learning for Computer Vision, ECCV 2022☆63Updated 2 years ago
- ☆34Updated 3 years ago
- NF-Layers for constructing neural functionals.☆84Updated last year
- Framework code with wandb, checkpointing, logging, configs, experimental protocols. Useful for fine-tuning models or training from scratc…☆150Updated 2 years ago
- LEARNING LATENT PERMUTATIONS WITH GUMBEL-SINKHORN NETWORKS IMPLEMENTATION WITH PYTORCH☆79Updated last year
- Code used in "Understanding Dimensional Collapse in Contrastive Self-supervised Learning" paper.☆77Updated 2 years ago
- Transformers with doubly stochastic attention☆45Updated 2 years ago
- Code for Reparameterizable Subset Sampling via Continuous Relaxations, IJCAI 2019.☆55Updated last year
- This repository contains the code for our paper "Probabilistic Contrastive Learning Recovers the Correct Aleatoric Uncertainty of Ambiguo…☆40Updated 2 years ago
- Code for "SoftSort: A Continuous Relaxation for the argsort Operator", ICML 2020.☆50Updated 2 years ago
- Code for paper Fully Hyperbolic Neural Networks☆77Updated 2 years ago
- Code for "The Intrinsic Dimension of Images and Its Impact on Learning" - ICLR 2021 Spotlight https://openreview.net/forum?id=XJk19XzGq2J☆68Updated last year
- Whitening for Self-Supervised Representation Learning | Official repository☆130Updated 2 years ago
- Official PyTorch implementation of Hyperbolic Neural Networks++☆69Updated 4 years ago
- Learnable Fourier Features for Multi-Dimensional Spatial Positional Encoding☆47Updated 6 months ago
- Differentiable Optimizers with Perturbations in Pytorch☆68Updated 4 years ago