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
• apple / ml-sigmoid-attention
  ☆302Updated 6 months ago
• NVlabs / GatedDeltaNet
  [ICLR 2025] Official PyTorch Implementation of Gated Delta Networks: Improving Mamba2 with Delta Rule
  ☆367Updated last month
• jzhang38 / LongMamba
  Some preliminary explorations of Mamba's context scaling.
  ☆216Updated last year
• HanGuo97 / log-linear-attention
  ☆253Updated 5 months ago
• lucidrains / PEER-pytorch
  Pytorch implementation of the PEER block from the paper, Mixture of A Million Experts, by Xu Owen He at Deepmind
  ☆131Updated last week
• Haiyang-W / TokenFormer
  [ICLR2025 Spotlight🔥] Official Implementation of TokenFormer: Rethinking Transformer Scaling with Tokenized Model Parameters
  ☆576Updated 9 months ago
• dvruette / gidd
  Code accompanying the paper "Generalized Interpolating Discrete Diffusion"
  ☆108Updated 5 months ago
• bluorion-com / ZClip
  Official implementation of the paper: "ZClip: Adaptive Spike Mitigation for LLM Pre-Training".
  ☆138Updated 3 weeks ago
• srush / annotated-mamba
  Annotated version of the Mamba paper
  ☆490Updated last year
• bloc97 / DeMo
  DeMo: Decoupled Momentum Optimization
  ☆196Updated 11 months ago
• HazyResearch / lolcats
  Repo for "LoLCATs: On Low-Rank Linearizing of Large Language Models"
  ☆249Updated 9 months ago
• zyushun / Adam-mini
  Code for Adam-mini: Use Fewer Learning Rates To Gain More https://arxiv.org/abs/2406.16793
  ☆440Updated 6 months ago
• KellerJordan / cifar10-airbench
  CIFAR-10 speedruns: 94% in 2.6 seconds and 96% in 27 seconds
  ☆322Updated 3 months ago
• nanowell / AdEMAMix-Optimizer-Pytorch
  The AdEMAMix Optimizer: Better, Faster, Older.
  ☆186Updated last year
• lucidrains / llama-qrlhf
  Implementation of the Llama architecture with RLHF + Q-learning
  ☆167Updated 9 months ago
• lucidrains / ring-attention-pytorch
  Implementation of 💍 Ring Attention, from Liu et al. at Berkeley AI, in Pytorch
  ☆543Updated 5 months ago
• llm-random / llm-random
  ☆204Updated 3 weeks ago
• KindXiaoming / grow-crystals
  Getting crystal-like representations with harmonic loss
  ☆192Updated 7 months ago
• fal-ai / diffusion-speedrun
  Focused on fast experimentation and simplicity
  ☆75Updated 10 months ago
• HomebrewML / HeavyBall
  Efficient optimizers
  ☆276Updated 3 weeks ago
• amorehead / jvp_flash_attention
  Flash Attention Triton kernel with support for second-order derivatives
  ☆111Updated 3 weeks ago
• lucidrains / hyper-connections
  Attempt to make multiple residual streams from Bytedance's Hyper-Connections paper accessible to the public
  ☆91Updated 4 months ago
• dingo-actual / infini-transformer
  PyTorch implementation of Infini-Transformer from "Leave No Context Behind: Efficient Infinite Context Transformers with Infini-attention…
  ☆292Updated last year
• athms / mad-lab
  A MAD laboratory to improve AI architecture designs 🧪
  ☆132Updated 10 months ago
• fal-ai-community / nano-mdm
  Tiny re-implementation of MDM in style of LLaDA and nano-gpt speedrun
  ☆57Updated 8 months ago
• Zyphra / Zamba2
  PyTorch implementation of models from the Zamba2 series.
  ☆185Updated 9 months ago
• HazyResearch / based
  Code for exploring Based models from "Simple linear attention language models balance the recall-throughput tradeoff"
  ☆241Updated 5 months ago
• lucidrains / titok-pytorch
  Implementation of TiTok, proposed by Bytedance in "An Image is Worth 32 Tokens for Reconstruction and Generation"
  ☆181Updated last year
• kvfrans / jax-diffusion-transformer
  Implementation of Diffusion Transformer (DiT) in JAX
  ☆294Updated last year