bojone / softtopk
differentiable top-k operator
☆21Updated 3 weeks ago
Alternatives and similar repositories for softtopk:
Users that are interested in softtopk are comparing it to the libraries listed below
- Triton implement of bi-directional (non-causal) linear attention☆35Updated last week
- A torch-based implementation of K-Means and K-Means++☆17Updated 4 years ago
- ☆31Updated 7 months ago
- IntLLaMA: A fast and light quantization solution for LLaMA☆18Updated last year
- Self Reproduction Code of Paper "Reducing Transformer Key-Value Cache Size with Cross-Layer Attention (MIT CSAIL)☆12Updated 7 months ago
- useful dotfiles included vim, zsh, tmux and vscode☆18Updated 3 weeks ago
- Benchmarking Attention Mechanism in Vision Transformers.☆17Updated 2 years ago
- Keras implement of Finite Scalar Quantization☆68Updated last year
- [EVA ICLR'23; LARA ICML'22] Efficient attention mechanisms via control variates, random features, and importance sampling☆80Updated last year
- A Tight-fisted Optimizer☆47Updated last year
- An object detection codebase based on MegEngine.☆28Updated 2 years ago
- A small framework mimics PyTorch using CuPy or NumPy☆27Updated 2 years ago
- # Unified Normalization (ACM MM'22) By Qiming Yang, Kai Zhang, Chaoxiang Lan, Zhi Yang, Zheyang Li, Wenming Tan, Jun Xiao, and Shiliang P…☆34Updated last year
- GIFT: Generative Interpretable Fine-Tuning☆19Updated 3 months ago
- Paper List for In-context Learning 🌷☆20Updated 2 years ago
- [CVPR2023] This is an official implementation of paper "DETRs with Hybrid Matching".☆14Updated 2 years ago
- ☆42Updated 3 years ago
- ☆52Updated last week
- [ICLR 2024] This is the official PyTorch implementation of "QLLM: Accurate and Efficient Low-Bitwidth Quantization for Large Language Mod…☆22Updated 10 months ago
- A big_vision inspired repo that implements a generic Auto-Encoder class capable in representation learning and generative modeling.☆34Updated 6 months ago
- [ECCV 2024] This is the official implementation of "Stitched ViTs are Flexible Vision Backbones".☆24Updated 11 months ago
- [ACL 2023] PuMer: Pruning and Merging Tokens for Efficient Vision Language Models☆28Updated 3 months ago
- ☆21Updated 3 months ago
- 32 times longer context window than vanilla Transformers and up to 4 times longer than memory efficient Transformers.☆44Updated last year
- [ICML 2024] When Linear Attention Meets Autoregressive Decoding: Towards More Effective and Efficient Linearized Large Language Models☆28Updated 7 months ago
- Mixture of Attention Heads☆41Updated 2 years ago
- ☆35Updated 7 months ago
- PyTorch code for Q-DiT: Accurate Post-Training Quantization for Diffusion Transformers☆36Updated 4 months ago
- A huge dataset for Document Visual Question Answering☆15Updated 5 months ago