ivan-chai / torch-linear-assignment
Batch computation of the linear assignment problem on GPU.
☆55Updated this week
Alternatives and similar repositories for torch-linear-assignment:
Users that are interested in torch-linear-assignment are comparing it to the libraries listed below
- [ECCV 2024] Official PyTorch implementation of RoPE-ViT "Rotary Position Embedding for Vision Transformer"☆270Updated last month
- Demystify RAM Usage in Multi-Process Data Loaders☆187Updated last year
- Neighborhood Attention Extension. Bringing attention to a neighborhood near you!☆396Updated this week
- A partial implementation of Generative Infinite Vocabulary Transformer (GIVT) from Google Deepmind, in PyTorch.☆17Updated 10 months ago
- A practical implementation of GradNorm, Gradient Normalization for Adaptive Loss Balancing, in Pytorch☆85Updated last year
- A Transformer made of Rotation-equivariant Attention using Vector Neurons☆84Updated last year
- Tiny PyTorch library for maintaining a moving average of a collection of parameters.☆417Updated 3 months ago
- Implementation of rectified flow and some of its followup research / improvements in Pytorch☆237Updated 2 weeks ago
- Implementation of Soft MoE, proposed by Brain's Vision team, in Pytorch☆258Updated 9 months ago
- A simple minimal implementation of Reversible Vision Transformers☆120Updated 10 months ago
- Implementation of Autoregressive Diffusion in Pytorch☆349Updated 2 months ago
- Implementation of fused cosine similarity attention in the same style as Flash Attention☆210Updated last year
- A library for calculating the FLOPs in the forward() process based on torch.fx☆95Updated 4 months ago
- A compilation of network architectures for vision and others without usage of self-attention mechanism☆77Updated 2 years ago
- Implementation of the proposed Adam-atan2 from Google Deepmind in Pytorch☆101Updated 2 months ago
- unofficial MaskGIT reproduction in PyTorch☆183Updated 11 months ago
- Implementation of Deformable Attention in Pytorch from the paper "Vision Transformer with Deformable Attention"☆306Updated 9 months ago
- PyTorch implementations of KMeans, Soft-KMeans and Constrained-KMeans which can be run on GPU and work on (mini-)batches of data.☆62Updated 2 years ago
- A simple way to keep track of an Exponential Moving Average (EMA) version of your Pytorch model☆552Updated last month
- Transformers w/o Attention, based fully on MLPs☆92Updated 9 months ago
- Educational implementation of the Discrete Flow Matching paper☆75Updated 5 months ago
- An efficient pytorch implementation of selective scan in one file, works with both cpu and gpu, with corresponding mathematical derivatio…☆78Updated 10 months ago
- ☆144Updated 2 years ago
- ☆67Updated 3 months ago
- a PyTorch re-implementation of ECCV 2022 paper based on Detectron2: k-means mask Transformer.☆71Updated last year
- A better PyTorch implementation of image local attention which reduces the GPU memory by an order of magnitude.☆136Updated 3 years ago
- [ICCV 2023] Online Clustered Codebook☆157Updated 4 months ago
- Official implementation of "Hydra: Bidirectional State Space Models Through Generalized Matrix Mixers"☆120Updated this week
- Modern Fixed Point Systems using Pytorch☆85Updated last year
- This is the official PyTorch implementation for "Mesa: A Memory-saving Training Framework for Transformers".☆120Updated 3 years ago