[ICLR 2025] Samba: Simple Hybrid State Space Models for Efficient Unlimited Context Language Modeling
☆979Nov 16, 2025Updated 10 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 month
- PyTorch implementation of models from the Zamba2 series.☆196Jan 23, 2025Updated last year
- Implementation for MatMul-free LM.☆3,093Sep 6, 2026Updated 3 weeks ago
- HGRN2: Gated Linear RNNs with State Expansion☆59Aug 20, 2024Updated 2 years ago
- Mamba SSM architecture☆18,869Jul 22, 2026Updated 2 months ago
- Bare Metal GPUs on DigitalOcean Gradient AI • AdPurpose-built for serious AI teams training foundational models, running large-scale inference, and pushing the boundaries of what's possible.
- Code for exploring Based models from "Simple linear attention language models balance the recall-throughput tradeoff"☆258Jun 6, 2025Updated last year
- 🚀 Efficient implementations for emerging model architectures☆5,792Updated this week
- Annotated version of the Mamba paper☆504Feb 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,634Updated this week
- A PyTorch native platform for training generative AI models☆5,767Updated this week
- Accelerated First Order Parallel Associative Scan☆199Jan 7, 2026Updated 8 months ago
- Q-GaLore: Quantized GaLore with INT4 Projection and Layer-Adaptive Low-Rank Gradients.☆206Jul 17, 2024Updated 2 years ago
- Tile primitives for speedy kernels☆3,729Sep 12, 2026Updated 2 weeks ago
- GPU virtual machines on DigitalOcean Gradient AI • AdGet to production fast with high-performance AMD and NVIDIA GPUs you can spin up in seconds. The definition of operational simplicity.
- [ICLR 2025] Official PyTorch Implementation of Gated Delta Networks: Improving Mamba2 with Delta Rule☆676Mar 13, 2026Updated 6 months ago
- [NeurIPS'24 Spotlight, ICLR'25, ICML'25] To speed up Long-context LLMs' inference, approximate and dynamic sparse calculate the attention…☆1,230Sep 10, 2026Updated 2 weeks ago
- Minimalistic large language model 3D-parallelism training☆2,827Updated this week
- 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
- A subset of PyTorch's neural network modules, written in Python using OpenAI's Triton.☆604Aug 14, 2026Updated last month
- Official implementation of "Hydra: Bidirectional State Space Models Through Generalized Matrix Mixers"☆177Jan 30, 2025Updated last year
- Understand and test language model architectures on synthetic tasks.☆288Mar 22, 2026Updated 6 months ago
- Some preliminary explorations of Mamba's context scaling.☆222Feb 8, 2024Updated 2 years ago
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- 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 11 months ago
- Simple, minimal implementation of the Mamba SSM in one file of PyTorch.☆2,970Mar 8, 2024Updated 2 years ago
- gpt-2 from scratch in mlx☆439Jun 12, 2024Updated 2 years ago
- Helpful tools and examples for working with flex-attention☆1,253Updated this week
- Code for BLT research paper☆2,064Nov 3, 2025Updated 10 months ago
- Simple & Scalable Pretraining for Neural Architecture Research☆347Mar 31, 2026Updated 5 months ago
- Repository for Meta Chameleon, a mixed-modal early-fusion foundation model from FAIR.☆2,102Jul 29, 2024Updated 2 years ago
- Tools for merging pretrained large language models.☆7,374Sep 12, 2026Updated 2 weeks ago
- Proton VPN Special Offer - Get 70% off • AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- Open weights language model from Google DeepMind, based on Griffin.☆685Feb 6, 2026Updated 7 months ago
- Fast and memory-efficient exact attention☆25,019Updated this week
- Reference implementation of Megalodon 7B model☆527May 17, 2025Updated last year
- [ICML'24 Spotlight] LLM Maybe LongLM: Self-Extend LLM Context Window Without Tuning☆668Jun 1, 2024Updated 2 years ago
- ☆63Jul 9, 2024Updated 2 years ago
- ☆11Oct 11, 2023Updated 2 years ago
- A State-Space Model with Rational Transfer Function Representation.☆85May 17, 2024Updated 2 years ago