bojone / softtopkLinks
differentiable top-k operator
☆22Updated last year
Alternatives and similar repositories for softtopk
Users that are interested in softtopk are comparing it to the libraries listed below
Sorting:
- Triton implement of bi-directional (non-causal) linear attention☆60Updated 11 months ago
- Keras implement of Finite Scalar Quantization☆83Updated 2 years ago
- flex-block-attn: an efficient block sparse attention computation library☆103Updated 2 weeks ago
- A torch-based implementation of K-Means and K-Means++☆17Updated 5 years ago
- ☆32Updated last year
- [ICLR 2025 & COLM 2025] Official PyTorch implementation of the Forgetting Transformer and Adaptive Computation Pruning☆134Updated 3 weeks ago
- ☆19Updated last year
- [EVA ICLR'23; LARA ICML'22] Efficient attention mechanisms via control variates, random features, and importance sampling☆87Updated 2 years ago
- [EMNLP 2022] Official implementation of Transnormer in our EMNLP 2022 paper - The Devil in Linear Transformer☆64Updated 2 years ago
- Xmixers: A collection of SOTA efficient token/channel mixers☆28Updated 4 months ago
- The official GitHub page for the survey paper "Discrete Tokenization for Multimodal LLMs: A Comprehensive Survey". And this paper is unde…☆76Updated 5 months ago
- Benchmarking Attention Mechanism in Vision Transformers.☆19Updated 3 years ago
- Document the demo and a series of documents for learning the diffusion model.☆42Updated 2 years ago
- A repository for DenseSSMs☆88Updated last year
- LLaVA combines with Magvit Image tokenizer, training MLLM without an Vision Encoder. Unifying image understanding and generation.☆39Updated last year
- ☆200Updated 2 years ago
- Self Reproduction Code of Paper "Reducing Transformer Key-Value Cache Size with Cross-Layer Attention (MIT CSAIL)☆18Updated last year
- 32 times longer context window than vanilla Transformers and up to 4 times longer than memory efficient Transformers.☆49Updated 2 years ago
- Official Code for "Rethinking Diffusion Model in High Dimension"☆24Updated 7 months ago
- ☆28Updated 3 months ago
- [ICML'24 Oral] The official code of "DiJiang: Efficient Large Language Models through Compact Kernelization", a novel DCT-based linear at…☆104Updated last year
- Patch convolution to avoid large GPU memory usage of Conv2D☆93Updated 11 months ago
- ☆46Updated 4 years ago
- Mixture of Attention Heads☆51Updated 3 years ago
- IntLLaMA: A fast and light quantization solution for LLaMA☆18Updated 2 years ago
- Tiny-FSDP, a minimalistic re-implementation of the PyTorch FSDP☆93Updated 4 months ago
- This is a PyTorch implementation of the paperViP A Differentially Private Foundation Model for Computer Vision☆36Updated 2 years ago
- imagetokenizer is a python package, helps you encoder visuals and generate visuals token ids from codebook, supports both image and video…☆37Updated last year
- WeGeFT: Weight‑Generative Fine‑Tuning for Multi‑Faceted Efficient Adaptation of Large Models☆22Updated 6 months ago
- ☆102Updated 10 months ago