tanaymeh / mamba-train
A single repo with all scripts and utils to train / fine-tune the Mamba model with or without FIM
☆53Updated 11 months ago
Alternatives and similar repositories for mamba-train:
Users that are interested in mamba-train are comparing it to the libraries listed below
- Repo hosting codes and materials related to speeding LLMs' inference using token merging.☆35Updated 10 months ago
- Official repository for the paper "SwitchHead: Accelerating Transformers with Mixture-of-Experts Attention"☆97Updated 5 months ago
- RWKV-7: Surpassing GPT☆80Updated 3 months ago
- Collection of autoregressive model implementation☆83Updated 3 weeks ago
- Evaluating the Mamba architecture on the Othello game☆44Updated 10 months ago
- Tree Attention: Topology-aware Decoding for Long-Context Attention on GPU clusters☆123Updated 3 months ago
- A repository for research on medium sized language models.☆77Updated 9 months ago
- PB-LLM: Partially Binarized Large Language Models☆151Updated last year
- ☆73Updated 6 months ago
- working implimention of deepseek MLA☆38Updated 2 months ago
- Official repository for the paper "NeuZip: Memory-Efficient Training and Inference with Dynamic Compression of Neural Networks". This rep…☆54Updated 4 months ago
- Griffin MQA + Hawk Linear RNN Hybrid☆85Updated 10 months ago
- ☆181Updated this week
- This repo is based on https://github.com/jiaweizzhao/GaLore☆26Updated 5 months ago
- ☆49Updated 11 months ago
- Block Transformer: Global-to-Local Language Modeling for Fast Inference (NeurIPS 2024)☆149Updated 2 months ago
- ☆125Updated last year
- My fork os allen AI's OLMo for educational purposes.☆30Updated 3 months ago
- Deep learning library implemented from scratch in numpy. Mixtral, Mamba, LLaMA, GPT, ResNet, and other experiments.☆51Updated 11 months ago
- The simplest implementation of recent Sparse Attention patterns for efficient LLM inference.☆58Updated last month
- The simplest, fastest repository for training/finetuning medium-sized xLSTMs.☆41Updated 9 months ago
- ☆193Updated 3 months ago
- Fast and memory-efficient exact attention☆66Updated last week
- ☆49Updated 4 months ago
- Implementation of the Mamba SSM with hf_integration.☆56Updated 6 months ago
- ☆78Updated 10 months ago
- Layer-Condensed KV cache w/ 10 times larger batch size, fewer params and less computation. Dramatic speed up with better task performance…☆147Updated last month
- Efficient Infinite Context Transformers with Infini-attention Pytorch Implementation + QwenMoE Implementation + Training Script + 1M cont…☆79Updated 10 months ago