A single repo with all scripts and utils to train / fine-tune the Mamba model with or without FIM
☆62Apr 8, 2024Updated 2 years ago
Alternatives and similar repositories for mamba-train
Users that are interested in mamba-train are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Set of scripts to finetune LLMs☆38Mar 30, 2024Updated 2 years ago
- Modular task agnostic training pipeline using LFM2 from Liquid AI with unsloth.☆16Sep 13, 2025Updated 8 months ago
- A repository for research on medium sized language models.☆78May 23, 2024Updated 2 years ago
- Implementation of mamba with rust☆95Mar 9, 2024Updated 2 years ago
- Repo hosting codes and materials related to speeding LLMs' inference using token merging.☆37Oct 9, 2025Updated 7 months ago
- Wordpress hosting with auto-scaling - Free Trial Offer • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- A simpler Pytorch + Zeta Implementation of the paper: "SiMBA: Simplified Mamba-based Architecture for Vision and Multivariate Time series…☆29Nov 11, 2024Updated last year
- ☆14Dec 19, 2024Updated last year
- Official PyTorch Implementation of "The Hidden Attention of Mamba Models"☆232Oct 16, 2025Updated 7 months ago
- ☆15Apr 26, 2025Updated last year
- [NeurIPS 2022] Latency-aware Spatial-wise Dynamic Networks☆25Aug 21, 2023Updated 2 years ago
- Running LLMs against a sandbox airport to see if they can make the correct decisions in real time☆26Jul 22, 2025Updated 10 months ago
- 🐜🔧 A minimalistic tool to fine-tune your LLMs☆18Aug 17, 2023Updated 2 years ago
- Using multiple LLMs for ensemble Forecasting☆16Jan 17, 2024Updated 2 years ago
- ☆17Apr 29, 2024Updated 2 years 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.
- CODA: Repurposing Continuous VAEs for Discrete Tokenization☆36Jul 4, 2025Updated 10 months ago
- Group-relative Trajectory-based Policy Optimization: Increasing Quality and Training Stability☆40Feb 23, 2026Updated 3 months ago
- Accelerate your Hugging Face Transformers 7.6-9x. Native to Hugging Face and PyTorch.☆683Aug 22, 2024Updated last year
- ☆17Jan 30, 2024Updated 2 years ago
- Evaluating the Mamba architecture on the Othello game☆50Apr 25, 2024Updated 2 years ago
- ☆27Oct 6, 2022Updated 3 years ago
- ☆37Jan 18, 2023Updated 3 years ago
- ☆11Oct 11, 2023Updated 2 years ago
- Official implementation of the ICML 2024 paper RoSA (Robust Adaptation)☆45May 20, 2026Updated last week
- 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.
- Implementation of a Hierarchical Mamba as described in the paper: "Hierarchical State Space Models for Continuous Sequence-to-Sequence Mo…☆15Nov 11, 2024Updated last year
- Rust bindings for CTranslate2☆14Jun 21, 2023Updated 2 years ago
- Simple Model Similarities Analysis☆21Feb 3, 2024Updated 2 years ago
- ☆145Aug 20, 2025Updated 9 months ago
- Code for Blog Post: Can Better Cold-Start Strategies Improve RL Training for LLMs?☆20Mar 9, 2025Updated last year
- Implementation of BitNet-1.58 instruct tuning☆29Apr 14, 2024Updated 2 years ago
- [DEPRECEATED] Piano Transformer model trained on 2.6GB of MIDI piano music☆13Oct 10, 2022Updated 3 years ago
- Automated Identification of Redundant Layer Blocks for Pruning in Large Language Models☆267Apr 23, 2024Updated 2 years ago
- [ECCV 2024] AdaNAT: Exploring Adaptive Policy for Token-Based Image Generation☆35Sep 12, 2024Updated last year
- AI Agents on DigitalOcean Gradient AI Platform • AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- Toolkit for attaching, training, saving and loading of new heads for transformer models☆298Feb 12, 2026Updated 3 months ago
- AgentParse is a high-performance parsing library designed to map various structured data formats (such as Pydantic models, JSON, YAML, an…☆18Oct 13, 2025Updated 7 months ago
- Implementation of Gradient Agreement Filtering, from Chaubard et al. of Stanford, but for single machine microbatches, in Pytorch☆25Jan 21, 2025Updated last year
- [ECCV 2024] Efficient Diffusion Transformer with Step-wise Dynamic Attention Mediators☆47Sep 11, 2024Updated last year
- Repository of "Train Once, Get a Family: State-Adaptive Balances for Offline-to-Online Reinforcement Learning" (NeurIPS 2023 Spotlight)☆41Oct 30, 2023Updated 2 years ago
- Toy distributed PostgreSQL by implementing SQL over KV☆11Jan 14, 2026Updated 4 months ago
- A library for easily merging multiple LLM experts, and efficiently train the merged LLM.☆515Aug 26, 2024Updated last year