Abstract
A cost-efficient open-source pretraining recipe trains 2B-parameter models on consumer GPUs for under $7K, yielding performance near larger baselines while deriving cost scaling laws and studying data curricula.
Language model pretraining has become almost synonymous with prohibitive cost, placing it out of reach for much of the academic and open-source communities. Although strong open-source efforts already exist, including open-weight models and open-source training recipes, a cost-efficient, hardware-accessible, and open-source pretraining recipe has long been missing. Even at a small scale, training Llama-3.2-3B costs over \1.5M, and reproducing SmolLM3-3B needs over 700K. In this report, we present an open pretraining recipe designed to lower this barrier. Using this recipe, we train a collection of Puro-2B models from scratch on up to 1.4 trillion tokens with FP8 precision on consumer-grade RTX 5090 GPUs. The models in the collection differ in token budgets and selected recipe variants. Our best model is trained at a compute cost of less than \6.9K and approaches Qwen2.5-1.5B performance under our evaluation protocol. This cost efficiency is enabled by a combination of approaches, including hardware selection, low-precision training, hyperball optimization, curriculum model averaging, and the data recipe. Beyond the recipe itself, we provide two additional results. First, across the Puro-2B collection, we derive a Puro Cost Scaling Law that relates training cost to average model performance; the fitted law suggests that about 4.4K, less than \$5,090, is sufficient to reach the performance of Qwen2-1.5B. Second, as an end-to-end case study, we examine how pretraining data curricula shape downstream performance after post-training. Such controlled studies are enabled by having access to the full pretraining pipeline rather than model weights alone. We release the full training recipe for Puro-2B, including data, code, and model weights under Apache 2.0 at https://huggingface.co/collections/thu-pacman/puro-2b.
Community
How much would it cost to pretrain a 2B LLM from scratch? $1M? $100K?
Puro-2B matches Qwen2-1.5B for under $5,090, trained on RTX 5090s — with a fully open training recipe!
24 RTX5090s in Phase 1 and 96 in Phase 2. See Table 1 in our report for details!
on FP8: the 419 TFLOPS you quote is the FP32-accumulate number, which NVIDIA runs at half rate on “consumer blackwell”/sm120.
Did you look at the FP16-accumulate path at all? I have poked at this myself with hadamard rotations etc and never got it stable, so I am curious whether you tried and hit the same wall or just left it alone. Seems like it would move the effective peak and you care about cost here
Yes, we did try the FP16-accumulation path. In our experiments, however, FP16 accumulation noticeably hurt model performance, so we ultimately stayed with FP32 accumulation.
We agree that if the FP16-accumulation path could be made stable without sacrificing quality, it could further improve the effective compute/$ — definitely an interesting direction to explore.
This is an automated message from the Librarian Bot. I found the following papers similar to this paper.
The following papers were recommended by the Semantic Scholar API
- Memory Decoder at Scale: A Pretrained, Parametric Long-Term Memory (2026)
- Synthetic Persona Pretraining: Alignment from Token Zero (2026)
- Hidden Decoding at Scale: Latent Computation Scaling for Large Language Models (2026)
- A Sovereign, Open-Source Foundation Model for German and English (2026)
- Full-Stack FP4: Stable LLM Pretraining with Quantized Projections, Optimizers, and Attention (2026)
- Efficient Knowledge Distillation for LLMs: Offline Top-K Logits and a Fused Chunked KL Loss (2026)
- Matryoshka Language Model Suites (2026)
Please give a thumbs up to this comment if you found it helpful!
If you want recommendations for any Paper on Hugging Face checkout this Space
You can directly ask Librarian Bot for paper recommendations by tagging it in a comment: @librarian-bot recommend
Get this paper in your agent:
hf papers read 2608.27370 Don't have the latest CLI?
curl -LsSf https://hf.co/cli/install.sh | bash Models citing this paper 1
Datasets citing this paper 1
thu-pacman/Puro-2B
Spaces citing this paper 0
No Space linking this paper