[ICLR 2025] Samba: Simple Hybrid State Space Models for Efficient Unlimited Context Language Modeling
☆954Nov 16, 2025Updated 4 months ago
Alternatives and similar repositories for Samba
Users that are interested in Samba are comparing it to the libraries listed below
Sorting:
- Schedule-Free Optimization in PyTorch☆2,265May 21, 2025Updated 9 months ago
- Implementation for MatMul-free LM.☆3,060Dec 2, 2025Updated 3 months ago
- PyTorch implementation of models from the Zamba2 series.☆189Jan 23, 2025Updated last year
- HGRN2: Gated Linear RNNs with State Expansion☆56Aug 20, 2024Updated last year
- 🚀 Efficient implementations of state-of-the-art linear attention models☆4,630Updated this week
- Code for exploring Based models from "Simple linear attention language models balance the recall-throughput tradeoff"☆249Jun 6, 2025Updated 9 months ago
- Mamba SSM architecture☆17,524Updated this week
- Implementation of Diffusion Transformer (DiT) in JAX☆308Jun 11, 2024Updated last year
- Annotated version of the Mamba paper☆499Feb 27, 2024Updated 2 years ago
- Minimalistic large language model 3D-parallelism training☆2,617Feb 19, 2026Updated last month
- Efficient Triton Kernels for LLM Training☆6,216Updated this week
- A PyTorch native platform for training generative AI models☆5,162Updated this week
- Accelerated First Order Parallel Associative Scan☆196Jan 7, 2026Updated 2 months ago
- Tile primitives for speedy kernels☆3,232Updated this week
- Official repository of paper "RNNs Are Not Transformers (Yet): The Key Bottleneck on In-context Retrieval"☆27Apr 17, 2024Updated last year
- Q-GaLore: Quantized GaLore with INT4 Projection and Layer-Adaptive Low-Rank Gradients.☆203Jul 17, 2024Updated last year
- Helpful tools and examples for working with flex-attention☆1,157Feb 8, 2026Updated last month
- [ICLR 2025] Official PyTorch Implementation of Gated Delta Networks: Improving Mamba2 with Delta Rule☆506Mar 13, 2026Updated last week
- [NeurIPS'24 Spotlight, ICLR'25, ICML'25] To speed up Long-context LLMs' inference, approximate and dynamic sparse calculate the attention…☆1,196Mar 9, 2026Updated last week
- A subset of PyTorch's neural network modules, written in Python using OpenAI's Triton.☆598Aug 12, 2025Updated 7 months ago
- Here we will test various linear attention designs.☆62Apr 25, 2024Updated last year
- gpt-2 from scratch in mlx☆418Jun 12, 2024Updated last year
- Mamba-Chat: A chat LLM based on the state-space model architecture 🐍☆942Mar 3, 2024Updated 2 years ago
- Official implementation of "Hydra: Bidirectional State Space Models Through Generalized Matrix Mixers"☆170Jan 30, 2025Updated last year
- Understand and test language model architectures on synthetic tasks.☆262Updated this week
- Some preliminary explorations of Mamba's context scaling.☆218Feb 8, 2024Updated 2 years ago
- [NeurIPS 2024] Official Repository of The Mamba in the Llama: Distilling and Accelerating Hybrid Models☆238Oct 14, 2025Updated 5 months ago
- Simple, minimal implementation of the Mamba SSM in one file of PyTorch.☆2,929Mar 8, 2024Updated 2 years ago
- Code for BLT research paper☆2,030Nov 3, 2025Updated 4 months ago
- Repository for Meta Chameleon, a mixed-modal early-fusion foundation model from FAIR.☆2,085Jul 29, 2024Updated last year
- Open weights language model from Google DeepMind, based on Griffin.☆665Feb 6, 2026Updated last month
- ☆58Jul 9, 2024Updated last year
- Reference implementation of Megalodon 7B model☆528May 17, 2025Updated 10 months ago
- ☆19Dec 4, 2025Updated 3 months ago
- [ICML'24 Spotlight] LLM Maybe LongLM: Self-Extend LLM Context Window Without Tuning☆666Jun 1, 2024Updated last year
- Tools for merging pretrained large language models.☆6,867Updated this week
- Fast and memory-efficient exact attention☆22,832Updated this week
- 20+ high-performance LLMs with recipes to pretrain, finetune and deploy at scale.☆13,228Mar 6, 2026Updated 2 weeks ago
- The TinyLlama project is an open endeavor to pretrain a 1.1B Llama model on 3 trillion tokens.☆8,911May 3, 2024Updated last year