facebookresearch / memory
Memory layers use a trainable key-value lookup mechanism to add extra parameters to a model without increasing FLOPs. Conceptually, sparsely activated memory layers complement compute-heavy dense feed-forward layers, providing dedicated capacity to store and retrieve information cheaply.
☆322Updated 4 months ago
Alternatives and similar repositories for memory:
Users that are interested in memory are comparing it to the libraries listed below
- 🌾 OAT: A research-friendly framework for LLM online alignment, including preference learning, reinforcement learning, etc.☆338Updated this week
- A framework to study AI models in Reasoning, Alignment, and use of Memory (RAM).☆226Updated this week
- [ICML 2024] CLLMs: Consistency Large Language Models☆391Updated 5 months ago
- A project to improve skills of large language models☆354Updated this week
- Repo for "LoLCATs: On Low-Rank Linearizing of Large Language Models"☆231Updated 3 months ago
- ☆181Updated 2 months ago
- Benchmark and research code for the paper SWEET-RL Training Multi-Turn LLM Agents onCollaborative Reasoning Tasks☆186Updated 3 weeks ago
- PyTorch building blocks for the OLMo ecosystem☆205Updated this week
- OpenCoconut implements a latent reasoning paradigm where we generate thoughts before decoding.☆172Updated 3 months ago
- ☆176Updated 4 months ago
- LongRoPE is a novel method that can extends the context window of pre-trained LLMs to an impressive 2048k tokens.☆227Updated 8 months ago
- Code for "LayerSkip: Enabling Early Exit Inference and Self-Speculative Decoding", ACL 2024☆292Updated this week
- PyTorch implementation of models from the Zamba2 series.☆180Updated 3 months ago
- Q-GaLore: Quantized GaLore with INT4 Projection and Layer-Adaptive Low-Rank Gradients.☆199Updated 9 months ago
- An extension of the nanoGPT repository for training small MOE models.☆138Updated last month
- Code for the paper "Rethinking Benchmark and Contamination for Language Models with Rephrased Samples"☆301Updated last year
- A family of compressed models obtained via pruning and knowledge distillation☆336Updated 5 months ago
- Implementation of 🥥 Coconut, Chain of Continuous Thought, in Pytorch☆165Updated 4 months ago
- Tina: Tiny Reasoning Models via LoRA☆164Updated 2 weeks ago
- [NeurIPS 2024] Official Repository of The Mamba in the Llama: Distilling and Accelerating Hybrid Models☆215Updated this week
- Repo for Rho-1: Token-level Data Selection & Selective Pretraining of LLMs.☆409Updated last year
- ☆524Updated 3 weeks ago
- Block Transformer: Global-to-Local Language Modeling for Fast Inference (NeurIPS 2024)☆153Updated 3 weeks ago
- Efficient LLM Inference over Long Sequences☆372Updated last week
- Official PyTorch implementation for Hogwild! Inference: Parallel LLM Generation with a Concurrent Attention Cache☆98Updated 2 weeks ago
- ☆186Updated this week
- ☆287Updated last month
- Implementation of 💍 Ring Attention, from Liu et al. at Berkeley AI, in Pytorch☆511Updated 6 months ago
- Reproducible, flexible LLM evaluations☆198Updated last month
- Muon optimizer: +>30% sample efficiency with <3% wallclock overhead☆611Updated last month