Papers
arxiv:2608.27370

Puro-2B: Poor Lab's Qwen2-1.5B Trained on RTX 5090 within $5090

Published on Aug 27
· Submitted by
Harry Chen
on Aug 31
Authors:
,
,
,
,
,
,

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

Paper author Paper submitter

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!

how many 5090s where used during training and how much time it took?

·

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

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

Sign up or log in to comment

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

Spaces citing this paper 0

No Space linking this paper

Cite arxiv.org/abs/2608.27370 in a Space README.md to link it from this page.

Collections including this paper 3