ivan-chai / torch-linear-assignment
Batch computation of the linear assignment problem on GPU.
☆65Updated last month
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"☆283Updated 2 months ago
- A partial implementation of Generative Infinite Vocabulary Transformer (GIVT) from Google Deepmind, in PyTorch.☆17Updated 11 months ago
- N-dimensional Rotary Position Embeddings for PyTorch☆40Updated last year
- Transformers w/o Attention, based fully on MLPs☆93Updated 10 months ago
- Implementation of rectified flow and some of its followup research / improvements in Pytorch☆253Updated 3 weeks ago
- Implementation of Soft MoE, proposed by Brain's Vision team, in Pytorch☆264Updated 10 months ago
- A Transformer made of Rotation-equivariant Attention using Vector Neurons☆87Updated last year
- Implementation of Autoregressive Diffusion in Pytorch☆357Updated 4 months ago
- ☆75Updated 4 months ago
- Implementation of fused cosine similarity attention in the same style as Flash Attention☆210Updated 2 years ago
- An efficient pytorch implementation of selective scan in one file, works with both cpu and gpu, with corresponding mathematical derivatio…☆79Updated 11 months ago
- Demystify RAM Usage in Multi-Process Data Loaders☆189Updated last year
- A simple minimal implementation of Reversible Vision Transformers☆121Updated 11 months ago
- Halton Scheduler for Masked Generative Image Transformer☆190Updated this week
- A practical implementation of GradNorm, Gradient Normalization for Adaptive Loss Balancing, in Pytorch☆86Updated last year
- A library for calculating the FLOPs in the forward() process based on torch.fx☆97Updated 5 months ago
- The official implementation of "[MASK] is All You Need"☆111Updated this week
- When it comes to optimizers, it's always better to be safe than sorry☆180Updated last week
- Implementation of the proposed Adam-atan2 from Google Deepmind in Pytorch☆101Updated 3 months ago
- Official PyTorch implementation of the paper: Flow Matching in Latent Space☆253Updated last month
- Implementation of a multimodal diffusion transformer in Pytorch☆100Updated 8 months ago
- A simple way to keep track of an Exponential Moving Average (EMA) version of your Pytorch model☆563Updated 3 months ago
- Official Implementation for "Consistency Flow Matching: Defining Straight Flows with Velocity Consistency"☆193Updated last month
- Official implementation of "Hydra: Bidirectional State Space Models Through Generalized Matrix Mixers"☆123Updated last month
- Implementation of "Analyzing and Improving the Training Dynamics of Diffusion Models"☆92Updated last year
- Implementation of a U-net complete with efficient attention as well as the latest research findings☆274Updated 10 months ago
- Educational implementation of the Discrete Flow Matching paper☆76Updated 6 months ago
- 1.5−3.0× lossless training or pre-training speedup. An off-the-shelf, easy-to-implement algorithm for the efficient training of foundatio…☆218Updated 6 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