CG80499 / KAN-GPT-2Links
Training small GPT-2 style models using Kolmogorov-Arnold networks.
☆122Updated 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"☆103Updated 11 months ago
- Pytorch implementation of the PEER block from the paper, Mixture of A Million Experts, by Xu Owen He at Deepmind☆132Updated last month
- ☆105Updated 4 months ago
- ☆82Updated last year
- Code repository for Black Mamba☆260Updated last year
- Collection of autoregressive model implementation☆85Updated 7 months ago
- Implementation of the Llama architecture with RLHF + Q-learning☆168Updated 10 months ago
- A single repo with all scripts and utils to train / fine-tune the Mamba model with or without FIM☆61Updated last year
- Deep learning library implemented from scratch in numpy. Mixtral, Mamba, LLaMA, GPT, ResNet, and other experiments.☆53Updated last year
- Quick implementation of nGPT, learning entirely on the hypersphere, from NvidiaAI☆294Updated 6 months ago
- Getting crystal-like representations with harmonic loss☆192Updated 8 months ago
- A State-Space Model with Rational Transfer Function Representation.☆83Updated last year
- Explorations into the recently proposed Taylor Series Linear Attention☆100Updated last year
- Code to reproduce "Transformers Can Do Arithmetic with the Right Embeddings", McLeish et al (NeurIPS 2024)☆195Updated last year
- ☆152Updated last month
- PyTorch Implementation of Jamba: "Jamba: A Hybrid Transformer-Mamba Language Model"☆198Updated last month
- DeMo: Decoupled Momentum Optimization☆197Updated last year
- Swarming algorithms like PSO, Ant Colony, Sakana, and more in PyTorch 😊☆134Updated last month
- ☆129Updated 4 months ago
- Simple, minimal implementation of the Mamba SSM in one pytorch file. Using logcumsumexp (Heisen sequence).☆128Updated last year
- Implementation of MambaByte in "MambaByte: Token-free Selective State Space Model" in Pytorch and Zeta☆126Updated last month
- Landing repository for the paper "Softpick: No Attention Sink, No Massive Activations with Rectified Softmax"☆85Updated 2 months ago
- Efficiently discovering algorithms via LLMs with evolutionary search and reinforcement learning.☆118Updated 3 weeks ago
- PyTorch implementation of models from the Zamba2 series.☆186Updated 10 months ago
- The AdEMAMix Optimizer: Better, Faster, Older.☆186Updated last year
- Mixture of A Million Experts☆50Updated last year
- An efficent implementation of the method proposed in "The Era of 1-bit LLMs"☆155Updated last year
- The simplest, fastest repository for training/finetuning medium-sized xLSTMs.☆41Updated last year
- ☆203Updated last year
- Kolmogorov-Arnold Networks (KAN) using Chebyshev polynomials instead of B-splines.☆395Updated last year