AI-Guru / helibrunna
A HuggingFace compatible Small Language Model trainer.
☆74Updated last month
Alternatives and similar repositories for helibrunna:
Users that are interested in helibrunna are comparing it to the libraries listed below
- Tiled Flash Linear Attention library for fast and efficient mLSTM Kernels.☆47Updated this week
- Explorations into the proposal from the paper "Grokfast, Accelerated Grokking by Amplifying Slow Gradients"☆98Updated 3 months ago
- SaLSa Optimizer implementation (No learning rates needed)☆29Updated this week
- ☆43Updated last month
- A State-Space Model with Rational Transfer Function Representation.☆78Updated 10 months ago
- ☆120Updated 4 months ago
- A byte-level decoder architecture that matches the performance of tokenized Transformers.☆63Updated 11 months ago
- Implementation of a Light Recurrent Unit in Pytorch☆47Updated 5 months ago
- Implementation of the proposed Adam-atan2 from Google Deepmind in Pytorch☆102Updated 4 months ago
- ☆42Updated 2 months ago
- Collection of autoregressive model implementation☆83Updated last month
- ☆79Updated 11 months ago
- A single repo with all scripts and utils to train / fine-tune the Mamba model with or without FIM☆54Updated 11 months ago
- Efficient encoder-decoder architecture for small language models (≤1B parameters) with cross-architecture knowledge distillation and visi…☆23Updated last month
- The AdEMAMix Optimizer: Better, Faster, Older.☆179Updated 6 months ago
- ☆91Updated 2 months ago
- ☆52Updated 6 months ago
- Attempt to make multiple residual streams from Bytedance's Hyper-Connections paper accessible to the public☆78Updated last month
- Set of scripts to finetune LLMs☆37Updated 11 months ago
- Implementation of GateLoop Transformer in Pytorch and Jax☆87Updated 9 months ago
- ☆59Updated 4 months ago
- SlamKit is an open source tool kit for efficient training of SpeechLMs. It was used for "Slamming: Training a Speech Language Model on On…☆186Updated this week
- Implementation of Agent Attention in Pytorch☆90Updated 8 months ago
- Training small GPT-2 style models using Kolmogorov-Arnold networks.☆115Updated 10 months ago
- The simplest, fastest repository for training/finetuning medium-sized xLSTMs.☆42Updated 10 months ago
- https://x.com/BlinkDL_AI/status/1884768989743882276☆27Updated last month
- MLX implementation of xLSTM model by Beck et al. (2024)☆27Updated 9 months ago
- Code and pretrained models for the paper: "MatMamba: A Matryoshka State Space Model"☆59Updated 4 months ago
- This is the code that went into our practical dive using mamba as information extraction☆53Updated last year
- Pytorch implementation of the PEER block from the paper, Mixture of A Million Experts, by Xu Owen He at Deepmind☆121Updated 7 months ago