bojone / softtopkLinks
differentiable top-k operator
☆22Updated 8 months ago
Alternatives and similar repositories for softtopk
Users that are interested in softtopk are comparing it to the libraries listed below
Sorting:
- Keras implement of Finite Scalar Quantization☆80Updated last year
- Triton implement of bi-directional (non-causal) linear attention☆52Updated 6 months ago
- [EVA ICLR'23; LARA ICML'22] Efficient attention mechanisms via control variates, random features, and importance sampling☆86Updated 2 years ago
- A torch-based implementation of K-Means and K-Means++☆17Updated 4 years ago
- The official implementation of the paper "Reducing Fine-Tuning Memory Overhead by Approximate and Memory-Sharing Backpropagation"☆20Updated 8 months ago
- Self Reproduction Code of Paper "Reducing Transformer Key-Value Cache Size with Cross-Layer Attention (MIT CSAIL)☆17Updated last year
- ☆32Updated last year
- PyTorch implementation of "From Sparse to Soft Mixtures of Experts"☆61Updated 2 years ago
- A repository for DenseSSMs☆88Updated last year
- ☆18Updated 7 months ago
- [NeurIPS 2022 Spotlight] This is the official PyTorch implementation of "EcoFormer: Energy-Saving Attention with Linear Complexity"☆74Updated 2 years ago
- IntLLaMA: A fast and light quantization solution for LLaMA☆18Updated 2 years ago
- [ICLR 2025 & COLM 2025] Official PyTorch implementation of the Forgetting Transformer and Adaptive Computation Pruning☆124Updated 2 weeks ago
- A Tight-fisted Optimizer☆50Updated 2 years ago
- Document the demo and a series of documents for learning the diffusion model.☆40Updated 2 years ago
- Benchmarking Attention Mechanism in Vision Transformers.☆18Updated 2 years ago
- Mixture of Attention Heads☆49Updated 2 years ago
- ☆27Updated last month
- [EMNLP 2022] Official implementation of Transnormer in our EMNLP 2022 paper - The Devil in Linear Transformer☆62Updated 2 years ago
- Tiny-FSDP, a minimalistic re-implementation of the PyTorch FSDP☆73Updated last week
- This is a PyTorch implementation of the paperViP A Differentially Private Foundation Model for Computer Vision☆36Updated 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 2 years ago
- ☆45Updated 4 years ago
- ☆197Updated last year
- [ECCV 2024] This is the official implementation of "Stitched ViTs are Flexible Vision Backbones".☆28Updated last year
- Official code for "pi-Tuning: Transferring Multimodal Foundation Models with Optimal Multi-task Interpolation", ICML 2023.☆33Updated 2 years ago
- ☆106Updated last year
- WeGeFT: Weight‑Generative Fine‑Tuning for Multi‑Faceted Efficient Adaptation of Large Models☆21Updated last month
- Patch convolution to avoid large GPU memory usage of Conv2D☆92Updated 7 months ago
- [ICML'24 Oral] The official code of "DiJiang: Efficient Large Language Models through Compact Kernelization", a novel DCT-based linear at…☆102Updated last year