HazyResearch / based
Code for exploring Based models from "Simple linear attention language models balance the recall-throughput tradeoff"
β232Updated 2 months ago
Alternatives and similar repositories for based
Users that are interested in based are comparing it to the libraries listed below
Sorting:
- Understand and test language model architectures on synthetic tasks.β195Updated 2 months ago
- A MAD laboratory to improve AI architecture designs π§ͺβ115Updated 4 months ago
- Some preliminary explorations of Mamba's context scaling.β213Updated last year
- Tree Attention: Topology-aware Decoding for Long-Context Attention on GPU clustersβ126Updated 5 months ago
- Repo for "LoLCATs: On Low-Rank Linearizing of Large Language Models"β231Updated 3 months ago
- β146Updated last year
- Triton-based implementation of Sparse Mixture of Experts.β214Updated 5 months ago
- β189Updated this week
- Layer-Condensed KV cache w/ 10 times larger batch size, fewer params and less computation. Dramatic speed up with better task performanceβ¦β149Updated last month
- Multipack distributed sampler for fast padding-free training of LLMsβ189Updated 9 months ago
- Pytorch implementation of the PEER block from the paper, Mixture of A Million Experts, by Xu Owen He at Deepmindβ124Updated 8 months ago
- π₯ A minimal training framework for scaling FLA modelsβ128Updated last week
- Fast and memory-efficient exact attentionβ68Updated 2 months ago
- The simplest, fastest repository for training/finetuning medium-sized GPTs.β120Updated this week
- Token Omission Via Attentionβ126Updated 7 months ago
- [NeurIPS 2024] Official Repository of The Mamba in the Llama: Distilling and Accelerating Hybrid Modelsβ216Updated last week
- Cold Compress is a hackable, lightweight, and open-source toolkit for creating and benchmarking cache compression methods built on top ofβ¦β131Updated 9 months ago
- Accelerated First Order Parallel Associative Scanβ182Updated 8 months ago
- β198Updated 5 months ago
- Language models scale reliably with over-training and on downstream tasksβ97Updated last year
- Griffin MQA + Hawk Linear RNN Hybridβ86Updated last year
- Normalized Transformer (nGPT)β176Updated 5 months ago
- β106Updated 11 months ago
- Official implementation of Phi-Mamba. A MOHAWK-distilled model (Transformers to SSMs: Distilling Quadratic Knowledge to Subquadratic Modeβ¦β106Updated 8 months ago
- β52Updated 11 months ago
- β78Updated 8 months ago
- Implementation of π Ring Attention, from Liu et al. at Berkeley AI, in Pytorchβ512Updated 6 months ago
- Efficient triton implementation of Native Sparse Attention.β147Updated this week
- Positional Skip-wise Training for Efficient Context Window Extension of LLMs to Extremely Length (ICLR 2024)β205Updated 11 months ago
- [ICLR 2025] Official PyTorch Implementation of Gated Delta Networks: Improving Mamba2 with Delta Ruleβ161Updated last month