Open Source
Explore the latest AI open-source projects from GitHub and HuggingFace.
Explore the latest AI open-source projects from GitHub and HuggingFace.
HRM-Text is a 1B-parameter text generation model and full pretraining framework from Sapient Intelligence, built on the Hierarchical Reasoning Model (HRM) architecture. Its central claim is efficiency: the project positions foundation-model pretraining as accessible for roughly $1,000, reporting 130-600x less compute and 150-900x less data than conventional scaling-based training. Rather than shipping only weights, HRM-Text provides the end-to-end recipe — data pipeline, training code, evaluation, and checkpoint tooling — so researchers can pretrain a capable model from scratch.
HRM-Text is built on a hierarchical recurrent architecture strengthened by task completion and latent-space reasoning, a departure from the standard decoder-only transformer scaling recipe. The design emphasizes reasoning in latent space, and the framework integrates PrefixLM sequence packing, FlashAttention 3 kernels, and PyTorch FSDP2 for distributed training. Because the attention path depends on FlashAttention 3, Hopper-class GPUs (H100) are the expected training target.
The project publishes two reference configurations with reproducible cost and time estimates. The L size (0.6B parameters) trains on 8 H100s in a single node in about 50 hours ($800), while the XL size (1B parameters) uses 16 H100s across two nodes in about 46 hours ($1,472). These concrete recipes make the efficiency claims verifiable rather than aspirational, and they lower the barrier for academic labs and independent researchers who lack hyperscaler budgets.
The reference runs report strong results for models of this scale. The XL (1B) model reaches 84.7% on GSM8k, 56.5% on MATH, 82.3% on DROP, 60.7% on MMLU, 81.9% on ARC-C, 63.4% on HellaSwag, 72.4% on Winogrande, and 86.2% on BoolQ. The L (0.6B) model trails but remains competitive, for example 77.6% on GSM8k and 56.6% on MMLU. The emphasis on math and reasoning benchmarks aligns with the architecture's latent-reasoning focus.
HRM-Text is designed to be run, not just read. It pairs with a companion data_io pipeline for cleaning, tokenizing, and stratified-sampling the pretraining corpus, and it supports single-node and multi-node layouts with shared storage. A published Docker image bundles the full environment including PyTorch, CUDA, and FlashAttention 3, and training integrates Weights & Biases for metric tracking. The workflow covers NCCL communication checks, torchrun launch commands, and checkpoint conversion.
HRM-Text is a notable entry in efficient pretraining, offering a reproducible path to a reasoning-capable 1B model at a fraction of typical cost. Its main constraints are practical: it targets Hopper-class hardware due to FlashAttention 3, and reproducing the headline results still requires access to 8-16 H100s. The strong self-reported reasoning benchmarks warrant independent verification, but the open framework, transparent cost estimates, and complete tooling make it a valuable resource for researchers exploring alternatives to brute-force scaling.