bkj / auction-lap
Auction algorithm for solving linear assignment problem (LAP)
☆70Updated 2 years ago
Alternatives and similar repositories for auction-lap:
Users that are interested in auction-lap are comparing it to the libraries listed below
- Efficient reservoir sampling implementation for PyTorch☆107Updated 3 years ago
- Repository for our paper: Deep Graph Matching via Blackbox Differentiation of Combinatorial Solvers☆88Updated 4 years ago
- Differentiable Optimizers with Perturbations in Pytorch☆66Updated 3 years ago
- Fast linear assignment problem (LAP) solvers for Python based on c-extensions☆158Updated 3 months ago
- Implementation of Lie Transformer, Equivariant Self-Attention, in Pytorch☆90Updated 4 years ago
- [NeurIPS 2019] Deep Set Prediction Networks☆100Updated 4 years ago
- Differentiable Algorithms and Algorithmic Supervision.☆114Updated last year
- Differentiable bitonic sorting☆139Updated 4 years ago
- LEARNING LATENT PERMUTATIONS WITH GUMBEL-SINKHORN NETWORKS IMPLEMENTATION WITH PYTORCH☆79Updated last year
- Drop-in replacement for any ResNet with a significantly reduced memory footprint and better representation capabilities☆209Updated 11 months ago
- Sinkhorn Transformer - Practical implementation of Sparse Sinkhorn Attention☆260Updated 3 years ago
- Pytorch Custom CUDA kernel for searchsorted☆137Updated last year
- [NeurIPS 2020 Oral] Is normalization indispensable for training deep neural networks?☆34Updated 2 years ago
- Torch modules that wrap blackbox combinatorial solvers according to the method presented in "Differentiating Blackbox Combinatorial Solve…☆343Updated 3 years ago
- Code for "Scale-Equivariant Steerable Networks"☆69Updated 4 years ago
- (Batched) advanced indexing for PyTorch.☆53Updated 3 months ago
- ☆30Updated 6 years ago
- Matrix square root with gradient support for PyTorch☆67Updated 2 years ago
- Pytorch implementation of the hamburger module from the ICLR 2021 paper "Is Attention Better Than Matrix Decomposition"☆98Updated 4 years ago
- Group Equivariant Convolutional Neural Networks☆100Updated 7 years ago
- Code for "Stochastic Optimization of Sorting Networks using Continuous Relaxations", ICLR 2019.☆139Updated 2 years ago
- PyTorch re-implementation of parts of "Deep Sets" (NIPS 2017)☆68Updated 6 years ago
- [NeurIPS'20] Multiscale Deep Equilibrium Models☆233Updated 3 years ago
- Code for the article "What if Neural Networks had SVDs?", to be presented as a spotlight paper at NeurIPS 2020.☆74Updated 8 months ago
- Masked Convolutional Flow☆59Updated 4 years ago
- Backpropagation-Friendly-Eigendecomposition☆72Updated 5 years ago
- Meta Optimal Transport☆101Updated last year
- Pytorch implementation of Variational Dropout Sparsifies Deep Neural Networks☆83Updated 3 years ago
- Official source code for "Roto-translated Local Coordinate Frames For Interacting Dynamical Systems". In NeurIPS 2021.☆21Updated 2 years ago
- A Pytorch implementation of the optimal transport kernel embedding☆114Updated 4 years ago