idoh / fast_mamba.np
A pure and fast NumPy implementation of Mamba with cache support.
☆17Updated 3 months ago
Related projects: ⓘ
- Token Omission Via Attention☆118Updated 7 months ago
- Testing LLM reasoning abilities with family relationship quizzes.☆40Updated 3 weeks ago
- ☆27Updated 2 months ago
- The simplest, fastest repository for training/finetuning medium-sized xLSTMs.☆38Updated 3 months ago
- Tree Attention: Topology-aware Decoding for Long-Context Attention on GPU clusters☆94Updated 2 weeks ago
- My Implementation of Q-Sparse: All Large Language Models can be Fully Sparsely-Activated☆27Updated last month
- RWKV, in easy to read code☆52Updated 6 months ago
- Fast modular code to create and train cutting edge LLMs☆63Updated 4 months ago
- Repository for the paper Stream of Search: Learning to Search in Language☆70Updated last month
- ☆55Updated 9 months ago
- NLP with Rust for Python 🦀🐍☆57Updated 3 months ago
- Implementation of the Quiet-STAR paper (https://arxiv.org/pdf/2403.09629.pdf)☆27Updated last month
- ☆53Updated 8 months ago
- Parameter-Efficient Sparsity Crafting From Dense to Mixture-of-Experts for Instruction Tuning on General Tasks☆31Updated 3 months ago
- ☆29Updated 3 weeks ago
- A byte-level decoder architecture that matches the performance of tokenized Transformers.☆57Updated 4 months ago
- A single repo with all scripts and utils to train / fine-tune the Mamba model with or without FIM☆46Updated 5 months ago
- Explorations into the proposal from the paper "Grokfast, Accelerated Grokking by Amplifying Slow Gradients"☆82Updated 3 weeks ago
- ☆15Updated 10 months ago
- ☆48Updated 6 months ago
- ☆75Updated 3 weeks ago
- ☆26Updated this week
- Deep learning library implemented from scratch in numpy. Mixtral, Mamba, LLaMA, GPT, ResNet, and other experiments.☆47Updated 5 months ago
- Code for the paper 'Grokked Transformers are Implicit Reasoners: A Mechanistic Journey to the Edge of Generalization'☆140Updated 3 months ago
- Gpu benchmark☆35Updated 2 weeks ago
- Collection of autoregressive model implementation☆62Updated 2 weeks ago
- Alice in Wonderland code base for experiments and raw experiments data☆96Updated last week
- ☆109Updated last month
- Comprehensive analysis of difference in performance of QLora, Lora, and Full Finetunes.☆81Updated last year
- Implementation of mamba with rust☆69Updated 6 months ago