Alternatives and similar repositories for nGPT-pytorch

Users that are interested in nGPT-pytorch are comparing it to the libraries listed below

• NVIDIA / ngpt
  Normalized Transformer (nGPT)
  ☆192Updated 11 months ago
• bluorion-com / ZClip
  Official implementation of the paper: "ZClip: Adaptive Spike Mitigation for LLM Pre-Training".
  ☆135Updated last week
• jzhang38 / LongMamba
  Some preliminary explorations of Mamba's context scaling.
  ☆216Updated last year
• NVlabs / GatedDeltaNet
  [ICLR 2025] Official PyTorch Implementation of Gated Delta Networks: Improving Mamba2 with Delta Rule
  ☆331Updated last month
• lucidrains / PEER-pytorch
  Pytorch implementation of the PEER block from the paper, Mixture of A Million Experts, by Xu Owen He at Deepmind
  ☆129Updated last year
• apple / ml-sigmoid-attention
  ☆302Updated 6 months ago
• srush / annotated-mamba
  Annotated version of the Mamba paper
  ☆489Updated last year
• HanGuo97 / log-linear-attention
  ☆251Updated 4 months ago
• HazyResearch / lolcats
  Repo for "LoLCATs: On Low-Rank Linearizing of Large Language Models"
  ☆248Updated 8 months ago
• dvruette / gidd
  Code accompanying the paper "Generalized Interpolating Discrete Diffusion"
  ☆103Updated 4 months ago
• lucidrains / llama-qrlhf
  Implementation of the Llama architecture with RLHF + Q-learning
  ☆167Updated 8 months ago
• bloc97 / DeMo
  DeMo: Decoupled Momentum Optimization
  ☆194Updated 10 months ago
• HazyResearch / based
  Code for exploring Based models from "Simple linear attention language models balance the recall-throughput tradeoff"
  ☆241Updated 4 months ago
• nanowell / AdEMAMix-Optimizer-Pytorch
  The AdEMAMix Optimizer: Better, Faster, Older.
  ☆186Updated last year
• lucidrains / ring-attention-pytorch
  Implementation of 💍 Ring Attention, from Liu et al. at Berkeley AI, in Pytorch
  ☆542Updated 5 months ago
• amorehead / jvp_flash_attention
  Flash Attention Triton kernel with support for second-order derivatives
  ☆103Updated this week
• EleutherAI / nanoGPT-mup
  The simplest, fastest repository for training/finetuning medium-sized GPTs.
  ☆166Updated 3 months ago
• llm-random / llm-random
  ☆200Updated last month
• KindXiaoming / grow-crystals
  Getting crystal-like representations with harmonic loss
  ☆193Updated 6 months ago
• Haiyang-W / TokenFormer
  [ICLR2025 Spotlight🔥] Official Implementation of TokenFormer: Rethinking Transformer Scaling with Tokenized Model Parameters
  ☆574Updated 8 months ago
• HazyResearch / zoology
  Understand and test language model architectures on synthetic tasks.
  ☆233Updated last month
• KellerJordan / cifar10-airbench
  CIFAR-10 speedruns: 94% in 2.6 seconds and 96% in 27 seconds
  ☆313Updated 3 months ago
• zyushun / Adam-mini
  Code for Adam-mini: Use Fewer Learning Rates To Gain More https://arxiv.org/abs/2406.16793
  ☆439Updated 5 months ago
• lucidrains / hyper-connections
  Attempt to make multiple residual streams from Bytedance's Hyper-Connections paper accessible to the public
  ☆91Updated 4 months ago
• athms / mad-lab
  A MAD laboratory to improve AI architecture designs 🧪
  ☆131Updated 10 months ago
• lucidrains / adam-atan2-pytorch
  Implementation of the proposed Adam-atan2 from Google Deepmind in Pytorch
  ☆129Updated last week
• NVlabs / hymba
  ☆201Updated 10 months ago
• proger / accelerated-scan
  Accelerated First Order Parallel Associative Scan
  ☆189Updated last year
• nil0x9 / flash-muon
  Flash-Muon: An Efficient Implementation of Muon Optimizer
  ☆195Updated 4 months ago
• kvfrans / jax-diffusion-transformer
  Implementation of Diffusion Transformer (DiT) in JAX
  ☆295Updated last year