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)
  ☆195Updated last year
• bluorion-com / ZClip
  Official implementation of the paper: "ZClip: Adaptive Spike Mitigation for LLM Pre-Training".
  ☆141Updated last month
• apple / ml-sigmoid-attention
  ☆304Updated 8 months ago
• jzhang38 / LongMamba
  Some preliminary explorations of Mamba's context scaling.
  ☆217Updated last year
• lucidrains / hyper-connections
  Attempt to make multiple residual streams from Bytedance's Hyper-Connections paper accessible to the public
  ☆139Updated this week
• HanGuo97 / log-linear-attention
  ☆265Updated 7 months ago
• lucidrains / llama-qrlhf
  Implementation of the Llama architecture with RLHF + Q-learning
  ☆170Updated 11 months ago
• bloc97 / DeMo
  DeMo: Decoupled Momentum Optimization
  ☆198Updated last year
• 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
• srush / annotated-mamba
  Annotated version of the Mamba paper
  ☆494Updated last year
• HazyResearch / lolcats
  Repo for "LoLCATs: On Low-Rank Linearizing of Large Language Models"
  ☆249Updated 11 months ago
• NVlabs / GatedDeltaNet
  [ICLR 2025] Official PyTorch Implementation of Gated Delta Networks: Improving Mamba2 with Delta Rule
  ☆421Updated 4 months ago
• dvruette / gidd
  Code accompanying the paper "Generalized Interpolating Discrete Diffusion"
  ☆112Updated 7 months ago
• HazyResearch / zoology
  Understand and test language model architectures on synthetic tasks.
  ☆249Updated this week
• nanowell / AdEMAMix-Optimizer-Pytorch
  The AdEMAMix Optimizer: Better, Faster, Older.
  ☆186Updated last year
• KellerJordan / cifar10-airbench
  CIFAR-10 speedruns: 94% in 2.6 seconds and 96% in 27 seconds
  ☆340Updated 2 months ago
• ethansmith2000 / fsdp_optimizers
  supporting pytorch FSDP for optimizers
  ☆84Updated last year
• lucidrains / ring-attention-pytorch
  Implementation of 💍 Ring Attention, from Liu et al. at Berkeley AI, in Pytorch
  ☆549Updated 8 months ago
• KindXiaoming / grow-crystals
  Getting crystal-like representations with harmonic loss
  ☆195Updated 9 months ago
• Haiyang-W / TokenFormer
  [ICLR2025 Spotlight🔥] Official Implementation of TokenFormer: Rethinking Transformer Scaling with Tokenized Model Parameters
  ☆581Updated 11 months ago
• llm-random / llm-random
  ☆207Updated last month
• NX-AI / flashrnn
  FlashRNN - Fast RNN Kernels with I/O Awareness
  ☆174Updated 2 months ago
• HomebrewML / HeavyBall
  Efficient optimizers
  ☆280Updated 3 weeks ago
• NVlabs / hymba
  ☆204Updated last year
• proger / accelerated-scan
  Accelerated First Order Parallel Associative Scan
  ☆193Updated last week
• zyushun / Adam-mini
  Code for Adam-mini: Use Fewer Learning Rates To Gain More https://arxiv.org/abs/2406.16793
  ☆449Updated 8 months ago
• EleutherAI / nanoGPT-mup
  The simplest, fastest repository for training/finetuning medium-sized GPTs.
  ☆185Updated 6 months ago
• athms / mad-lab
  A MAD laboratory to improve AI architecture designs 🧪
  ☆136Updated last year
• cloneofsimo / min-max-gpt
  Minimal (400 LOC) implementation Maximum (multi-node, FSDP) GPT training
  ☆132Updated last year
• kvfrans / jax-diffusion-transformer
  Implementation of Diffusion Transformer (DiT) in JAX
  ☆304Updated last year