CG80499 / KAN-GPT-2Links
Training small GPT-2 style models using Kolmogorov-Arnold networks.
☆121Updated last year
Alternatives and similar repositories for KAN-GPT-2
Users that are interested in KAN-GPT-2 are comparing it to the libraries listed below
Sorting:
- Explorations into the proposal from the paper "Grokfast, Accelerated Grokking by Amplifying Slow Gradients"☆101Updated 7 months ago
- Quick implementation of nGPT, learning entirely on the hypersphere, from NvidiaAI☆289Updated 2 months ago
- Code repository for Black Mamba☆252Updated last year
- Implementation of the Llama architecture with RLHF + Q-learning☆166Updated 6 months ago
- Pytorch implementation of the PEER block from the paper, Mixture of A Million Experts, by Xu Owen He at Deepmind☆127Updated 11 months ago
- Collection of autoregressive model implementation☆86Updated 3 months ago
- A single repo with all scripts and utils to train / fine-tune the Mamba model with or without FIM☆56Updated last year
- Explorations into the recently proposed Taylor Series Linear Attention☆100Updated 11 months ago
- Deep learning library implemented from scratch in numpy. Mixtral, Mamba, LLaMA, GPT, ResNet, and other experiments.☆50Updated last year
- ☆100Updated 2 weeks ago
- Implementations of attention with the softpick function, naive and FlashAttention-2☆81Updated 3 months ago
- PyTorch implementation of models from the Zamba2 series.☆184Updated 6 months ago
- Simple, minimal implementation of the Mamba SSM in one pytorch file. Using logcumsumexp (Heisen sequence).☆120Updated 9 months ago
- ☆81Updated last year
- Efficiently discovering algorithms via LLMs with evolutionary search and reinforcement learning.☆105Updated last week
- A State-Space Model with Rational Transfer Function Representation.☆79Updated last year
- PyTorch Implementation of Jamba: "Jamba: A Hybrid Transformer-Mamba Language Model"☆182Updated last week
- Swarming algorithms like PSO, Ant Colony, Sakana, and more in PyTorch 😊☆129Updated last week
- Normalized Transformer (nGPT)☆186Updated 8 months ago
- Getting crystal-like representations with harmonic loss☆193Updated 4 months ago
- The simplest, fastest repository for training/finetuning medium-sized xLSTMs.☆41Updated last year
- ☆126Updated last week
- ☆190Updated 8 months ago
- Tiled Flash Linear Attention library for fast and efficient mLSTM Kernels.☆67Updated last week
- Kolmogorov-Arnold Networks (KAN) using Chebyshev polynomials instead of B-splines.☆383Updated last year
- Integrating Mamba/SSMs with Transformer for Enhanced Long Context and High-Quality Sequence Modeling☆201Updated 3 weeks ago
- Griffin MQA + Hawk Linear RNN Hybrid☆88Updated last year
- ☆48Updated 9 months ago
- DeMo: Decoupled Momentum Optimization☆190Updated 8 months ago
- The AdEMAMix Optimizer: Better, Faster, Older.☆184Updated 10 months ago