Alternatives and similar repositories for nanoGPT-mup

Users that are interested in nanoGPT-mup are comparing it to the libraries listed below

• HazyResearch / zoology
  Understand and test language model architectures on synthetic tasks.
  ☆249Updated last week
• athms / mad-lab
  A MAD laboratory to improve AI architecture designs 🧪
  ☆136Updated last year
• cloneofsimo / min-fsdp
  ☆92Updated last year
• ethansmith2000 / fsdp_optimizers
  supporting pytorch FSDP for optimizers
  ☆84Updated last year
• epfml / llm-baselines
  nanoGPT-like codebase for LLM training
  ☆114Updated 2 months ago
• cloneofsimo / min-max-gpt
  Minimal (400 LOC) implementation Maximum (multi-node, FSDP) GPT training
  ☆132Updated last year
• HazyResearch / based
  Code for exploring Based models from "Simple linear attention language models balance the recall-throughput tradeoff"
  ☆244Updated 7 months ago
• berlino / seq_icl
  ☆53Updated last year
• epfml / schedules-and-scaling
  Code for NeurIPS 2024 Spotlight: "Scaling Laws and Compute-Optimal Training Beyond Fixed Training Durations"
  ☆87Updated last year
• NVIDIA / ngpt
  Normalized Transformer (nGPT)
  ☆196Updated last year
• mlfoundations / scaling
  Language models scale reliably with over-training and on downstream tasks
  ☆100Updated last year
• mcleish7 / arithmetic
  Code to reproduce "Transformers Can Do Arithmetic with the Right Embeddings", McLeish et al (NeurIPS 2024)
  ☆198Updated last year
• NVIDIA-NeMo / Emerging-Optimizers
  ☆119Updated 3 weeks ago
• lucidrains / coconut-pytorch
  Implementation of 🥥 Coconut, Chain of Continuous Thought, in Pytorch
  ☆181Updated 6 months ago
• martin-marek / batch-size
  📄Small Batch Size Training for Language Models
  ☆79Updated 3 months ago
• yixiaoer / tpux
  A set of Python scripts that makes your experience on TPU better
  ☆55Updated 4 months ago
• llm-random / llm-random
  ☆207Updated this week
• modula-systems / modula
  🧱 Modula software package
  ☆322Updated 5 months ago
• young-geng / mintext
  Minimal but scalable implementation of large language models in JAX
  ☆35Updated last month
• microsoft / mutransformers
  some common Huggingface transformers in maximal update parametrization (µP)
  ☆87Updated 3 years ago
• Zyphra / tree_attention
  Tree Attention: Topology-aware Decoding for Long-Context Attention on GPU clusters
  ☆131Updated last year
• google-deepmind / nanodo
  ☆287Updated last year
• Aleph-Alpha-Research / scaling
  Scaling is a distributed training library and installable dependency designed to scale up neural networks, with a dedicated module for tr…
  ☆66Updated 2 months ago
• RobertCsordas / moeut
  ☆91Updated last year
• samsja / muon_fsdp_2
  Muon fsdp 2
  ☆50Updated 5 months ago
• proger / accelerated-scan
  Accelerated First Order Parallel Associative Scan
  ☆193Updated last week
• MatX-inc / seqax
  seqax = sequence modeling + JAX
  ☆169Updated 5 months ago
• erfanzar / EasyDeL
  Accelerate, Optimize performance with streamlined training and serving options with JAX.
  ☆330Updated last week
• huggingface / picotron_tutorial
  ☆224Updated last month
• google-deepmind / mishax
  ☆151Updated 4 months ago