bojone / softtopk
differentiable top-k operator
☆21Updated 4 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☆71Updated last year
- IntLLaMA: A fast and light quantization solution for LLaMA☆18Updated last year
- Triton implement of bi-directional (non-causal) linear attention☆47Updated 3 months ago
- ☆30Updated 11 months ago
- [ICLR 2025] Official PyTorch implementation of "Forgetting Transformer: Softmax Attention with a Forget Gate"☆99Updated this week
- LLaVA combines with Magvit Image tokenizer, training MLLM without an Vision Encoder. Unifying image understanding and generation.☆38Updated 10 months ago
- Benchmarking Attention Mechanism in Vision Transformers.☆18Updated 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
- A tiny, didactical implementation of LLAMA 3☆40Updated 5 months ago
- A torch-based implementation of K-Means and K-Means++☆17Updated 4 years ago
- An object detection codebase based on MegEngine.☆28Updated 2 years ago
- A Tight-fisted Optimizer☆47Updated 2 years ago
- [EVA ICLR'23; LARA ICML'22] Efficient attention mechanisms via control variates, random features, and importance sampling☆85Updated 2 years ago
- Mixture of Attention Heads☆44Updated 2 years ago
- ☆17Updated 4 months ago
- Paper List for In-context Learning 🌷☆20Updated 2 years ago
- Document the demo and a series of documents for learning the diffusion model.☆39Updated last year
- A big_vision inspired repo that implements a generic Auto-Encoder class capable in representation learning and generative modeling.☆35Updated 10 months ago
- A small framework mimics PyTorch using CuPy or NumPy☆27Updated 3 years ago
- imagetokenizer is a python package, helps you encoder visuals and generate visuals token ids from codebook, supports both image and video…☆34Updated 10 months ago
- The open-source materials for paper "Sparsing Law: Towards Large Language Models with Greater Activation Sparsity".☆21Updated 6 months ago
- [EMNLP 2022] Official implementation of Transnormer in our EMNLP 2022 paper - The Devil in Linear Transformer☆60Updated last year
- This repository contains the code for the paper in Findings of EMNLP 2021: "EfficientBERT: Progressively Searching Multilayer Perceptron …☆32Updated last year
- An official pytorch implementation of AAAI 2024 paper "Latent Space Editing in Transformer-based Flow Matching"☆40Updated last year
- [CVPR 2024] DiffAgent: Fast and Accurate Text-to-Image API Selection with Large Language Model☆17Updated last year
- [ICLR 2024] This is the official PyTorch implementation of "QLLM: Accurate and Efficient Low-Bitwidth Quantization for Large Language Mod…☆27Updated last year
- [Preprint] Why is the State of Neural Network Pruning so Confusing? On the Fairness, Comparison Setup, and Trainability in Network Prunin…☆40Updated 2 years ago
- ☆22Updated last year
- [ICLR'25] "Understanding Bottlenecks of State Space Models through the Lens of Recency and Over-smoothing" by Peihao Wang, Ruisi Cai, Yue…☆11Updated last month
- The official implementation of the paper "Reducing Fine-Tuning Memory Overhead by Approximate and Memory-Sharing Backpropagation"☆20Updated 5 months ago