bkj / auction-lap
Auction algorithm for solving linear assignment problem (LAP)
☆70Updated 2 years ago
Related projects: ⓘ
- Repository for our paper: Deep Graph Matching via Blackbox Differentiation of Combinatorial Solvers☆86Updated 3 years ago
- Efficient reservoir sampling implementation for PyTorch☆104Updated 2 years ago
- Fast linear assignment problem (LAP) solvers for Python based on c-extensions☆156Updated 8 months ago
- Differentiable Optimizers with Perturbations in Pytorch☆64Updated 3 years ago
- [NeurIPS 2019] Deep Set Prediction Networks☆100Updated 4 years ago
- Differentiable bitonic sorting☆137Updated 4 years ago
- LEARNING LATENT PERMUTATIONS WITH GUMBEL-SINKHORN NETWORKS IMPLEMENTATION WITH PYTORCH☆76Updated last year
- Code for "Scale-Equivariant Steerable Networks"☆67Updated 4 years ago
- (Batched) advanced indexing for PyTorch.☆53Updated 8 months ago
- This repository is no longer maintained. Check☆82Updated 4 years ago
- A second-order optimizer for deep networks☆23Updated 4 years ago
- Code for "SoftSort: A Continuous Relaxation for the argsort Operator", ICML 2020.☆39Updated last year
- Matrix square root with gradient support for PyTorch☆63Updated 2 years ago
- Equivariant Transformer (ET) layers are image-to-image mappings that incorporate prior knowledge on invariances with respect to continuo…☆88Updated 5 years ago
- Differentiable Algorithms and Algorithmic Supervision.☆101Updated last year
- PyTorch Extension Library of Optimized Unique Operation☆37Updated 5 years ago
- PyTorch re-implementation of parts of "Deep Sets" (NIPS 2017)☆68Updated 6 years ago
- Implementation of Lie Transformer, Equivariant Self-Attention, in Pytorch☆87Updated 3 years ago
- Group Equivariant Convolutional Neural Networks☆96Updated 6 years ago
- ☆61Updated 4 years ago
- Torch modules that wrap blackbox combinatorial solvers according to the method presented in "Differentiating Blackbox Combinatorial Solve…☆334Updated 2 years ago
- Code for "Stochastic Optimization of Sorting Networks using Continuous Relaxations", ICLR 2019.☆134Updated last year
- implements optimal transport algorithms in pytorch☆91Updated 2 years ago
- Python pdb for multiple processes☆30Updated last year
- pytorch implementation of basic kmeans algorithm(lloyd method with forgy initialization) with gpu support☆93Updated 6 years ago
- Pytorch implementation of the hamburger module from the ICLR 2021 paper "Is Attention Better Than Matrix Decomposition"☆98Updated 3 years ago
- Backpropagation-Friendly-Eigendecomposition☆72Updated 4 years ago
- Official implementation of the paper "Topographic VAEs learn Equivariant Capsules"☆77Updated 2 years ago
- Demo source code for the paper "Esteves, C., Allen-Blanchette, C., Zhou, X. and Daniilidis, K, "Polar Transformer Networks", ICLR 2018.☆55Updated last year
- Torch implementation of orthoreg.☆15Updated 2 years ago