797GB of n-grams in read-only CPU RAM match 57% more compute at 1B-scale LM pretraining

anshulkundaje · x · 2026-10-16

The COUNT project trains LMs on top of empirical n-gram count tables precomputed into read-only CPU RAM — 797GB of n-grams. At 1B scale this matches 57% more compute, after a similar trick recently broke nanoGPT speedrun records. The tables are easy to precompute and store, significantly speed up pretraining, and can be hot-swapped at inference time for training-free domain adaptation.

Related event: 797GB n-gram Table in CPU Memory Boosts Pretraining Compute by 57%(2 posts)→

Original post →

More from Infra

Infra channel →