Menu

1 / 7

The Inference Hardware You Already Own Might Train Models Too: A 325M Pretraining Run on an NPU

DEV Community: machinelearning·GINIGEN AI·about 21 hours ago
#KSAd5pOj
Reading 0:00
15s threshold

Disclosure: the work described here is by VIDRAFT, an affiliate of Ginigen AI.

TL;DR

Edge and on-premise deployments usually split hardware into two worlds: GPUs to train, cheaper accelerators to serve. A new result from VIDRAFT narrows that split. They pretrained a 325M parameter language model from random initialization on the Rebellions ATOM-Max , a commercial inference NPU released in 2025, and trained the same model on a B200 GPU for comparison. On text neither model saw in training, the losses are 0.7993 (NPU) and 0.7994 (GPU) .

Why should edge and on-prem teams care?

Most organizations running AI on premise already have inference accelerators in the rack. They buy them because they are cheaper per query and draw less power than training GPUs. When a model needs adapting to local data, the usual answer is to rent GPU time or ship data out.

If the inference hardware can train, that changes the economics and the data story.…

Continue reading — create a free account

Join HashtagPLUS to read full articles, follow hashtags, vote, and join the conversation.