Alternatives and similar repositories for KAN-GPT-2

Users that are interested in KAN-GPT-2 are comparing it to the libraries listed below

• lucidrains / grokfast-pytorch
  Explorations into the proposal from the paper "Grokfast, Accelerated Grokking by Amplifying Slow Gradients"
  ☆103Updated last year
• LucasPrietoAl / grokking-at-the-edge-of-numerical-stability
  ☆109Updated 5 months ago
• Zyphra / Zamba2
  PyTorch implementation of models from the Zamba2 series.
  ☆186Updated 11 months ago
• tanaymeh / mamba-train
  A single repo with all scripts and utils to train / fine-tune the Mamba model with or without FIM
  ☆61Updated last year
• lucidrains / llama-qrlhf
  Implementation of the Llama architecture with RLHF + Q-learning
  ☆170Updated 11 months ago
• joey00072 / ohara
  Collection of autoregressive model implementation
  ☆85Updated last week
• lucidrains / PEER-pytorch
  Pytorch implementation of the PEER block from the paper, Mixture of A Million Experts, by Xu Owen He at Deepmind
  ☆132Updated 2 months ago
• jacobfa / fft
  ☆129Updated 5 months ago
• lucidrains / nGPT-pytorch
  Quick implementation of nGPT, learning entirely on the hypersphere, from NvidiaAI
  ☆293Updated 7 months ago
• kyegomez / swarms-pytorch
  Swarming algorithms like PSO, Ant Colony, Sakana, and more in PyTorch 😊
  ☆136Updated this week
• epfml / DenseFormer
  ☆82Updated last year
• ruke1ire / RTF
  A State-Space Model with Rational Transfer Function Representation.
  ☆83Updated last year
• jadechip / nanoXLSTM
  The simplest, fastest repository for training/finetuning medium-sized xLSTMs.
  ☆41Updated last year
• lucidrains / hyper-connections
  Attempt to make multiple residual streams from Bytedance's Hyper-Connections paper accessible to the public
  ☆148Updated this week
• KindXiaoming / grow-crystals
  Getting crystal-like representations with harmonic loss
  ☆195Updated 9 months ago
• NVIDIA / ngpt
  Normalized Transformer (nGPT)
  ☆196Updated last year
• apple / ml-pararnn
  ☆161Updated 2 months ago
• PeaBrane / mamba-tiny
  Simple, minimal implementation of the Mamba SSM in one pytorch file. Using logcumsumexp (Heisen sequence).
  ☆129Updated last year
• Zyphra / BlackMamba
  Code repository for Black Mamba
  ☆261Updated last year
• bloc97 / DeMo
  DeMo: Decoupled Momentum Optimization
  ☆198Updated last year
• martin-marek / batch-size
  📄Small Batch Size Training for Language Models
  ☆79Updated 3 months ago
• pbelcak / fastfeedforward
  A repository for log-time feedforward networks
  ☆224Updated last year
• johnma2006 / candle
  Deep learning library implemented from scratch in numpy. Mixtral, Mamba, LLaMA, GPT, ResNet, and other experiments.
  ☆53Updated last year
• nanowell / AdEMAMix-Optimizer-Pytorch
  The AdEMAMix Optimizer: Better, Faster, Older.
  ☆186Updated last year
• mcleish7 / arithmetic
  Code to reproduce "Transformers Can Do Arithmetic with the Right Embeddings", McLeish et al (NeurIPS 2024)
  ☆198Updated last year
• lucidrains / pytorch-custom-utils
  Just some miscellaneous utility functions / decorators / modules related to Pytorch and Accelerate to help speed up implementation of new…
  ☆126Updated last year
• andrewgcodes / xlstm
  my attempts at implementing various bits of Sepp Hochreiter's new xLSTM architecture
  ☆134Updated last year
• idiap / sigma-gpt
  σ-GPT: A New Approach to Autoregressive Models
  ☆70Updated last year
• kyegomez / Jamba
  PyTorch Implementation of Jamba: "Jamba: A Hybrid Transformer-Mamba Language Model"
  ☆203Updated 2 weeks ago
• NX-AI / mlstm_kernels
  Tiled Flash Linear Attention library for fast and efficient mLSTM Kernels.
  ☆82Updated last month