lucidrains / coconut-pytorchLinks
Implementation of π₯₯ Coconut, Chain of Continuous Thought, in Pytorch
β180Updated 5 months ago
Alternatives and similar repositories for coconut-pytorch
Users that are interested in coconut-pytorch are comparing it to the libraries listed below
Sorting:
- β109Updated last year
- [NeurIPS 2024] Official Repository of The Mamba in the Llama: Distilling and Accelerating Hybrid Modelsβ231Updated last month
- β199Updated 7 months ago
- Some preliminary explorations of Mamba's context scaling.β217Updated last year
- β89Updated last year
- Pytorch implementation of the PEER block from the paper, Mixture of A Million Experts, by Xu Owen He at Deepmindβ131Updated last month
- [COLM 2025] Code for Paper: Learning Adaptive Parallel Reasoning with Language Modelsβ134Updated 3 months ago
- β124Updated 9 months ago
- Block Transformer: Global-to-Local Language Modeling for Fast Inference (NeurIPS 2024)β162Updated 7 months ago
- Code for NeurIPS'24 paper 'Grokked Transformers are Implicit Reasoners: A Mechanistic Journey to the Edge of Generalization'β234Updated 4 months ago
- This is the official repository for Inheritune.β115Updated 9 months ago
- [ICLR2025] Codebase for "ReMoE: Fully Differentiable Mixture-of-Experts with ReLU Routing", built on Megatron-LM.β99Updated 11 months ago
- Code for ICLR 2025 Paper "What is Wrong with Perplexity for Long-context Language Modeling?"β105Updated last month
- β85Updated 3 weeks ago
- β75Updated last year
- A Large-Scale, High-Quality Math Dataset for Reinforcement Learning in Language Modelsβ68Updated 9 months ago
- Physics of Language Models, Part 4β260Updated 4 months ago
- Code for paper "Patch-Level Training for Large Language Models"β95Updated 3 weeks ago
- [NeurIPS 2024] Low rank memory efficient optimizer without SVDβ31Updated 5 months ago
- Code for exploring Based models from "Simple linear attention language models balance the recall-throughput tradeoff"β243Updated 5 months ago
- Unofficial implementation for the paper "Mixture-of-Depths: Dynamically allocating compute in transformer-based language models"β175Updated last year
- AnchorAttention: Improved attention for LLMs long-context trainingβ213Updated 10 months ago
- Implementation of CALM from the paper "LLM Augmented LLMs: Expanding Capabilities through Composition", out of Google Deepmindβ178Updated last year
- Parameter-Efficient Sparsity Crafting From Dense to Mixture-of-Experts for Instruction Tuning on General Tasks (EMNLP'24)β147Updated last year
- A framework to study AI models in Reasoning, Alignment, and use of Memory (RAM).β300Updated 3 weeks ago
- Replicating O1 inference-time scaling lawsβ90Updated last year
- OpenCoconut implements a latent reasoning paradigm where we generate thoughts before decoding.β173Updated 10 months ago
- Official implementation of Phi-Mamba. A MOHAWK-distilled model (Transformers to SSMs: Distilling Quadratic Knowledge to Subquadratic Modeβ¦β116Updated last year
- Repo for Rho-1: Token-level Data Selection & Selective Pretraining of LLMs.β447Updated last year
- Language models scale reliably with over-training and on downstream tasksβ100Updated last year