Vidraft Lab has successfully pre-trained a 325 million parameter language model, VKORE-325M-NPU-Base, from scratch using the REBELLION ATOM-Max, a commercial inference-only NPU. This achievement challenges the notion that inference-focused NPUs cannot be used for training. The NPU's performance was comparable to that of an Nvidia B200 GPU when evaluated on unseen text data, with a loss of 0.7993 versus 0.7994 respectively. Vidraft has open-sourced the model weights and evaluation scripts, though their proprietary training framework and stabilization techniques, crucial for the successful training, remain under patent. AI
IMPACT Demonstrates that inference-focused NPUs can be viable for LLM pre-training, potentially reducing hardware costs and increasing accessibility.
RANK_REASON Research paper detailing successful LLM pre-training on an inference-only NPU. [lever_c_demoted from research: ic=1 ai=1.0]
- FineWeb-Edu
- LlamaForCausalLM
- Nvidia B200
- REBELLION ATOM-Max
- transformers
- Vidraft Lab
- VKORE-325M-NPU-Base
AI-generated summary · Google Gemini · from 1 sources. How we write summaries →