OswaldHe / HMT-pytorch
[NAACL 2025] Official Implementation of "HMT: Hierarchical Memory Transformer for Long Context Language Processing"
☆66Updated last week
Alternatives and similar repositories for HMT-pytorch:
Users that are interested in HMT-pytorch are comparing it to the libraries listed below
- Tree Attention: Topology-aware Decoding for Long-Context Attention on GPU clusters☆113Updated last month
- Token Omission Via Attention☆122Updated 3 months ago
- ☆43Updated 3 months ago
- Layer-Condensed KV cache w/ 10 times larger batch size, fewer params and less computation. Dramatic speed up with better task performance…☆147Updated last week
- The simplest implementation of recent Sparse Attention patterns for efficient LLM inference.☆56Updated this week
- ☆70Updated 5 months ago
- PB-LLM: Partially Binarized Large Language Models☆150Updated last year
- The source code of our work "Prepacking: A Simple Method for Fast Prefilling and Increased Throughput in Large Language Models"☆58Updated 3 months ago
- PyTorch implementation of models from the Zamba2 series.☆173Updated last week
- Fast and memory-efficient exact attention☆58Updated last month
- ☆80Updated 4 months ago
- A repository for research on medium sized language models.☆76Updated 8 months ago
- Using FlexAttention to compute attention with different masking patterns☆40Updated 4 months ago
- ☆125Updated last year
- Stick-breaking attention☆41Updated 2 weeks ago
- ShiftAddLLM: Accelerating Pretrained LLMs via Post-Training Multiplication-Less Reparameterization☆101Updated 3 months ago
- Code for exploring Based models from "Simple linear attention language models balance the recall-throughput tradeoff"☆219Updated last month
- [NeurIPS 2024] Official Repository of The Mamba in the Llama: Distilling and Accelerating Hybrid Models☆191Updated this week
- Repo for "LoLCATs: On Low-Rank Linearizing of Large Language Models"☆205Updated last month
- The code repository for the CURLoRA research paper. Stable LLM continual fine-tuning and catastrophic forgetting mitigation.☆41Updated 5 months ago
- ☆57Updated this week
- Code for PHATGOOSE introduced in "Learning to Route Among Specialized Experts for Zero-Shot Generalization"☆80Updated 11 months ago
- One Initialization to Rule them All: Fine-tuning via Explained Variance Adaptation☆36Updated 3 months ago
- Cascade Speculative Drafting☆28Updated 9 months ago
- RWKV-7: Surpassing GPT☆75Updated 2 months ago
- ☆136Updated last year
- Easy control for Key-Value Constrained Generative LLM Inference(https://arxiv.org/abs/2402.06262)☆60Updated 11 months ago
- Here we will test various linear attention designs.☆58Updated 9 months ago
- ☆79Updated 2 months ago
- Pytorch implementation of the PEER block from the paper, Mixture of A Million Experts, by Xu Owen He at Deepmind☆115Updated 5 months ago