[ICLR 2025] Samba: Simple Hybrid State Space Models for Efficient Unlimited Context Language Modeling
☆971Nov 16, 2025Updated 8 months ago
Alternatives and similar repositories for Samba
Users that are interested in Samba are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Schedule-Free Optimization in PyTorch☆2,323Jul 28, 2026Updated last week
- PyTorch implementation of models from the Zamba2 series.☆194Jan 23, 2025Updated last year
- Implementation for MatMul-free LM.☆3,085Dec 2, 2025Updated 8 months ago
- HGRN2: Gated Linear RNNs with State Expansion☆58Aug 20, 2024Updated last year
- Mamba SSM architecture☆18,723Jul 22, 2026Updated 2 weeks ago
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- Code for exploring Based models from "Simple linear attention language models balance the recall-throughput tradeoff"☆256Jun 6, 2025Updated last year
- 🚀 Efficient implementations for emerging model architectures☆5,536Updated this week
- Annotated version of the Mamba paper☆502Feb 27, 2024Updated 2 years ago
- Implementation of Diffusion Transformer (DiT) in JAX☆322Jun 11, 2024Updated 2 years ago
- Efficient Triton Kernels for LLM Training☆6,562Updated this week
- A PyTorch native platform for training generative AI models☆5,610Updated this week
- Accelerated First Order Parallel Associative Scan☆198Jan 7, 2026Updated 7 months ago
- Q-GaLore: Quantized GaLore with INT4 Projection and Layer-Adaptive Low-Rank Gradients.☆206Jul 17, 2024Updated 2 years ago
- [ICLR 2025] Official PyTorch Implementation of Gated Delta Networks: Improving Mamba2 with Delta Rule☆643Mar 13, 2026Updated 4 months ago
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- [NeurIPS'24 Spotlight, ICLR'25, ICML'25] To speed up Long-context LLMs' inference, approximate and dynamic sparse calculate the attention…☆1,226Apr 8, 2026Updated 4 months ago
- Tile primitives for speedy kernels☆3,621Jul 13, 2026Updated 3 weeks ago
- Minimalistic large language model 3D-parallelism training☆2,780May 26, 2026Updated 2 months ago
- Official repository of paper "RNNs Are Not Transformers (Yet): The Key Bottleneck on In-context Retrieval"☆27Apr 17, 2024Updated 2 years ago
- Here we will test various linear attention designs.☆62Apr 25, 2024Updated 2 years ago
- Official implementation of "Hydra: Bidirectional State Space Models Through Generalized Matrix Mixers"☆176Jan 30, 2025Updated last year
- A subset of PyTorch's neural network modules, written in Python using OpenAI's Triton.☆606May 13, 2026Updated 2 months ago
- Understand and test language model architectures on synthetic tasks.☆281Mar 22, 2026Updated 4 months ago
- Some preliminary explorations of Mamba's context scaling.☆222Feb 8, 2024Updated 2 years ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- Mamba-Chat: A chat LLM based on the state-space model architecture 🐍☆942Mar 3, 2024Updated 2 years ago
- [NeurIPS 2024] Official Repository of The Mamba in the Llama: Distilling and Accelerating Hybrid Models☆243Oct 14, 2025Updated 9 months ago
- Simple, minimal implementation of the Mamba SSM in one file of PyTorch.☆2,965Mar 8, 2024Updated 2 years ago
- gpt-2 from scratch in mlx☆437Jun 12, 2024Updated 2 years ago
- Helpful tools and examples for working with flex-attention☆1,223Updated this week
- Code for BLT research paper☆2,059Nov 3, 2025Updated 9 months ago
- Simple & Scalable Pretraining for Neural Architecture Research☆341Mar 31, 2026Updated 4 months ago
- Repository for Meta Chameleon, a mixed-modal early-fusion foundation model from FAIR.☆2,104Jul 29, 2024Updated 2 years ago
- Tools for merging pretrained large language models.☆7,287Jun 17, 2026Updated last month
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- Fast and memory-efficient exact attention☆24,672Updated this week
- Open weights language model from Google DeepMind, based on Griffin.☆684Feb 6, 2026Updated 6 months ago
- ☆61Jul 9, 2024Updated 2 years ago
- Reference implementation of Megalodon 7B model☆526May 17, 2025Updated last year
- [ICML'24 Spotlight] LLM Maybe LongLM: Self-Extend LLM Context Window Without Tuning☆668Jun 1, 2024Updated 2 years ago
- ☆11Oct 11, 2023Updated 2 years ago
- A State-Space Model with Rational Transfer Function Representation.☆84May 17, 2024Updated 2 years ago