Oxen-AI / mamba-diveLinks
This is the code that went into our practical dive using mamba as information extraction
☆53Updated last year
Alternatives and similar repositories for mamba-dive
Users that are interested in mamba-dive are comparing it to the libraries listed below
Sorting:
- A single repo with all scripts and utils to train / fine-tune the Mamba model with or without FIM☆54Updated last year
- ☆95Updated 4 months ago
- Collection of autoregressive model implementation☆85Updated last month
- Set of scripts to finetune LLMs☆37Updated last year
- Explorations into the proposal from the paper "Grokfast, Accelerated Grokking by Amplifying Slow Gradients"☆100Updated 5 months ago
- The simplest, fastest repository for training/finetuning medium-sized xLSTMs.☆41Updated last year
- ☆80Updated last year
- Official repository for the paper "Approximating Two-Layer Feedforward Networks for Efficient Transformers"☆37Updated last year
- Implementation of the Mamba SSM with hf_integration.☆56Updated 9 months ago
- Repo hosting codes and materials related to speeding LLMs' inference using token merging.☆36Updated last year
- A repository for research on medium sized language models.☆76Updated last year
- ☆53Updated last year
- Prune transformer layers☆69Updated last year
- Deep learning library implemented from scratch in numpy. Mixtral, Mamba, LLaMA, GPT, ResNet, and other experiments.☆51Updated last year
- Pytorch implementation of the PEER block from the paper, Mixture of A Million Experts, by Xu Owen He at Deepmind☆126Updated 9 months ago
- Small and Efficient Mathematical Reasoning LLMs☆71Updated last year
- Training small GPT-2 style models using Kolmogorov-Arnold networks.☆117Updated last year
- ☆31Updated last year
- Efficient encoder-decoder architecture for small language models (≤1B parameters) with cross-architecture knowledge distillation and visi…☆27Updated 3 months ago
- A MAD laboratory to improve AI architecture designs 🧪☆116Updated 5 months ago
- ☆79Updated 9 months ago
- Implementation of GateLoop Transformer in Pytorch and Jax☆88Updated 11 months ago
- GoldFinch and other hybrid transformer components☆45Updated 10 months ago
- ☆49Updated last year
- ☆47Updated 9 months ago
- A byte-level decoder architecture that matches the performance of tokenized Transformers.☆63Updated last year
- ☆50Updated 7 months ago
- Simple GRPO scripts and configurations.☆58Updated 4 months ago
- Spherical Merge Pytorch/HF format Language Models with minimal feature loss.☆124Updated last year
- An unofficial pytorch implementation of 'Efficient Infinite Context Transformers with Infini-attention'☆52Updated 9 months ago