tanaymeh / mamba-train
A single repo with all scripts and utils to train / fine-tune the Mamba model with or without FIM
☆50Updated 7 months ago
Related projects ⓘ
Alternatives and complementary repositories for mamba-train
- Tree Attention: Topology-aware Decoding for Long-Context Attention on GPU clusters☆104Updated last month
- My fork os allen AI's OLMo for educational purposes.☆28Updated this week
- Official repository for the paper "SwitchHead: Accelerating Transformers with Mixture-of-Experts Attention"☆92Updated last month
- A repository for research on medium sized language models.☆74Updated 5 months ago
- ☆62Updated 3 months ago
- Collection of autoregressive model implementation☆67Updated this week
- [NeurIPS 2024] Official Repository of The Mamba in the Llama: Distilling and Accelerating Hybrid Models☆174Updated this week
- ☆49Updated 8 months ago
- ☆35Updated 3 weeks ago
- Parameter-Efficient Sparsity Crafting From Dense to Mixture-of-Experts for Instruction Tuning on General Tasks☆129Updated 2 months ago
- Token Omission Via Attention☆120Updated last month
- Official implementation of Phi-Mamba. A MOHAWK-distilled model (Transformers to SSMs: Distilling Quadratic Knowledge to Subquadratic Mode…☆78Updated 2 months ago
- PB-LLM: Partially Binarized Large Language Models☆148Updated last year
- A toolkit for fine-tuning, inferencing, and evaluating GreenBitAI's LLMs.☆74Updated last month
- Set of scripts to finetune LLMs☆36Updated 7 months ago
- ☆40Updated 2 weeks ago
- Repo hosting codes and materials related to speeding LLMs' inference using token merging.☆29Updated 6 months ago
- QuIP quantization☆46Updated 8 months ago
- ☆122Updated 9 months ago
- Griffin MQA + Hawk Linear RNN Hybrid☆85Updated 6 months ago
- ☆184Updated last month
- Pytorch implementation of the PEER block from the paper, Mixture of A Million Experts, by Xu Owen He at Deepmind☆112Updated 2 months ago
- code for training & evaluating Contextual Document Embedding models☆117Updated this week
- Block Transformer: Global-to-Local Language Modeling for Fast Inference (Official Code)☆135Updated last month
- Repo for "LoLCATs: On Low-Rank Linearizing of Large Language Models"☆177Updated last month
- Implementation of MoE Mamba from the paper: "MoE-Mamba: Efficient Selective State Space Models with Mixture of Experts" in Pytorch and Ze…☆84Updated last week
- ☆67Updated last week
- PyTorch implementation of models from the Zamba2 series.☆158Updated this week
- From GaLore to WeLore: How Low-Rank Weights Non-uniformly Emerge from Low-Rank Gradients. Ajay Jaiswal, Lu Yin, Zhenyu Zhang, Shiwei Liu,…☆43Updated 4 months ago
- Some preliminary explorations of Mamba's context scaling.☆191Updated 9 months ago