HazyResearch / basedLinks
Code for exploring Based models from "Simple linear attention language models balance the recall-throughput tradeoff"
β237Updated last month
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.β219Updated last month
- A MAD laboratory to improve AI architecture designs π§ͺβ123Updated 6 months ago
- Some preliminary explorations of Mamba's context scaling.β215Updated last year
- Repo for "LoLCATs: On Low-Rank Linearizing of Large Language Models"β243Updated 5 months ago
- β191Updated last week
- Tree Attention: Topology-aware Decoding for Long-Context Attention on GPU clustersβ128Updated 7 months ago
- β147Updated 2 years ago
- Triton-based implementation of Sparse Mixture of Experts.β224Updated 7 months ago
- The simplest, fastest repository for training/finetuning medium-sized GPTs.β147Updated 2 weeks ago
- Normalized Transformer (nGPT)β184Updated 7 months ago
- Layer-Condensed KV cache w/ 10 times larger batch size, fewer params and less computation. Dramatic speed up with better task performanceβ¦β150Updated 3 months ago
- β112Updated last year
- Multipack distributed sampler for fast padding-free training of LLMsβ194Updated 11 months ago
- Token Omission Via Attentionβ128Updated 9 months ago
- Pytorch implementation of the PEER block from the paper, Mixture of A Million Experts, by Xu Owen He at Deepmindβ127Updated 10 months ago
- β81Updated last year
- Accelerated First Order Parallel Associative Scanβ182Updated 10 months ago
- β53Updated last year
- Fast and memory-efficient exact attentionβ68Updated 4 months ago
- π₯ A minimal training framework for scaling FLA modelsβ188Updated last month
- Griffin MQA + Hawk Linear RNN Hybridβ87Updated last year
- β199Updated 7 months ago
- Implementation of π₯₯ Coconut, Chain of Continuous Thought, in Pytorchβ177Updated 3 weeks ago
- Code to reproduce "Transformers Can Do Arithmetic with the Right Embeddings", McLeish et al (NeurIPS 2024)β190Updated last year
- Language models scale reliably with over-training and on downstream tasksβ97Updated last year
- [NeurIPS 2024] Official Repository of The Mamba in the Llama: Distilling and Accelerating Hybrid Modelsβ222Updated 2 months ago
- The simplest implementation of recent Sparse Attention patterns for efficient LLM inference.β78Updated last month
- Code for studying the super weight in LLMβ113Updated 7 months ago
- Implementation of π Ring Attention, from Liu et al. at Berkeley AI, in Pytorchβ529Updated 2 months ago
- β127Updated last year