AIAny
AI Train2024
Icon for item

modded-nanogpt

A community speedrun to train a 124M GPT as fast as possible on 8 H100s, all chasing a fixed 3.28 FineWeb loss. Successive records cut the run from llm.c's 45 minutes to under 1.4, mostly via the new Muon optimizer rather than more hardware.

Introduction

Most "train GPT-2 on a budget" projects ship one clever recipe and call it done. This one turned the recipe into a leaderboard. By freezing the hardware (8 NVIDIA H100s), the dataset (FineWeb), and the finish line (3.28 cross-entropy validation loss), it leaves wall-clock time as the only variable — so every contributor's trick is directly comparable, and the record only moves when something is genuinely faster.

What Sets It Apart
  • It is where the Muon optimizer first proved itself: roughly 1.5x more sample-efficient than Adam, lower memory than Adam, and under 2% wallclock overhead — gains that showed up as record drops, not benchmark claims.
  • The numbers are stark: the baseline (Karpathy's llm.c, May 2024) needed 45 minutes and ~10B tokens; the current record reaches the same loss in about 1.3 minutes on under 400M tokens.
  • Records are open to anyone, by any method, and every entry ships runnable code — so the leaderboard doubles as a curated archive of training optimizations that actually work.
  • A second track trains GPT-2 Medium to 2.92 loss, extending the same discipline to a larger model.
Who It's For

Great fit if you want a battle-tested, end-to-end reference for modern LLM training tricks — Muon, architectural tweaks, and data/throughput tuning — or a fair playground to pit your own optimization against a real record. Look elsewhere if you need a general-purpose, multi-node training framework: this is deliberately a single-node, 124M-parameter sprint, and its code is tuned for the leaderboard's exact constraints rather than flexibility or scale.

Information

  • Websitegithub.com
  • OrganizationsNanoGPT Speedrun Community
  • AuthorsKeller Jordan, Jeremy Bernstein, Brendan Rappazzo, Community contributors
  • Published date2024/06/01

Categories

More Items

GitHub
AI Train2019

Train and experiment with multi-billion to trillion-parameter transformer models on large GPU clusters using GPU-optimized building blocks and reference training scripts; offers advanced parallelism and mixed-precision support for research teams and ML engineers.

GitHub
AI Train2026

Provides a one-command CLI to fine-tune and post-train LLMs, with layer streaming that lets an 8B model be fine-tuned on a 4 GB laptop GPU. Auto-configures quantization, LoRA adapters, batching and evaluation gates, and supports export and serving workflows.

GitHub
AI Train2025

A PyTorch DTensor-native SPMD library for training and fine-tuning LLMs, VLMs, diffusion and retrieval models. Integrates with Hugging Face for day-0 model support, provides YAML-driven recipes, DTensor/FSDP2 parallelism and NVIDIA-optimized kernels (Transformer Engine, DeepEP, FlexAttn).