Alternatives and similar repositories for ngpt:

Users that are interested in ngpt are comparing it to the libraries listed below

• HazyResearch / lolcats
  Repo for "LoLCATs: On Low-Rank Linearizing of Large Language Models"
  ☆231Updated 3 months ago
• jzhang38 / LongMamba
  Some preliminary explorations of Mamba's context scaling.
  ☆213Updated 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
  ☆123Updated 8 months ago
• HazyResearch / zoology
  Understand and test language model architectures on synthetic tasks.
  ☆194Updated last month
• HazyResearch / based
  Code for exploring Based models from "Simple linear attention language models balance the recall-throughput tradeoff"
  ☆232Updated 2 months ago
• lucidrains / nGPT-pytorch
  Quick implementation of nGPT, learning entirely on the hypersphere, from NvidiaAI
  ☆281Updated last month
• LucasPrietoAl / grokking-at-the-edge-of-numerical-stability
  ☆94Updated 3 months ago
• Zyphra / tree_attention
  Tree Attention: Topology-aware Decoding for Long-Context Attention on GPU clusters
  ☆126Updated 5 months ago
• EleutherAI / nanoGPT-mup
  The simplest, fastest repository for training/finetuning medium-sized GPTs.
  ☆105Updated this week
• athms / mad-lab
  A MAD laboratory to improve AI architecture designs 🧪
  ☆113Updated 4 months ago
• ethansmith2000 / fsdp_optimizers
  supporting pytorch FSDP for optimizers
  ☆80Updated 4 months ago
• proger / accelerated-scan
  Accelerated First Order Parallel Associative Scan
  ☆182Updated 8 months ago
• NVlabs / GatedDeltaNet
  [ICLR 2025] Official PyTorch Implementation of Gated Delta Networks: Improving Mamba2 with Delta Rule
  ☆159Updated last month
• jxiw / MambaInLlama
  [NeurIPS 2024] Official Repository of The Mamba in the Llama: Distilling and Accelerating Hybrid Models
  ☆215Updated this week
• bloc97 / DeMo
  DeMo: Decoupled Momentum Optimization
  ☆186Updated 5 months ago
• kyleliang919 / Super_Muon
  ☆54Updated last month
• OpenEvaByte / evabyte
  EvaByte: Efficient Byte-level Language Models at Scale
  ☆91Updated 2 weeks ago
• xiayuqing0622 / flex_head_fa
  Fast and memory-efficient exact attention
  ☆68Updated 2 months ago
• facebookresearch / memory
  Memory layers use a trainable key-value lookup mechanism to add extra parameters to a model without increasing FLOPs. Conceptually, spars…
  ☆322Updated 4 months ago
• shawntan / scattermoe
  Triton-based implementation of Sparse Mixture of Experts.
  ☆212Updated 5 months ago
• Zyphra / Zamba2
  PyTorch implementation of models from the Zamba2 series.
  ☆180Updated 3 months ago
• foundation-model-stack / fms-fsdp
  🚀 Efficiently (pre)training foundation models with native PyTorch features, including FSDP for training and SDPA implementation of Flash…
  ☆244Updated this week
• wolfecameron / nanoMoE
  An extension of the nanoGPT repository for training small MOE models.
  ☆138Updated last month
• PrimeIntellect-ai / prime-rl
  prime-rl is a codebase for decentralized RL training at scale
  ☆85Updated this week
• cloneofsimo / min-fsdp
  ☆78Updated 10 months ago
• fal-ai-community / NativeSparseAttention
  research impl of Native Sparse Attention (2502.11089)
  ☆53Updated 2 months ago
• cloneofsimo / min-max-gpt
  Minimal (400 LOC) implementation Maximum (multi-node, FSDP) GPT training
  ☆123Updated last year
• apple / ml-sigmoid-attention
  ☆278Updated last week
• lucidrains / coconut-pytorch
  Implementation of 🥥 Coconut, Chain of Continuous Thought, in Pytorch
  ☆165Updated 4 months ago
• nshepperd / flash_attn_jax
  JAX bindings for Flash Attention v2
  ☆89Updated 9 months ago