huginn-s-learned-entropy0p01
Table 2, S, Learned lambda_H=0.01 (Table 4 S teacher) of Towards Looped Models Done Right. Part II: Rethinking at Fixed Points.
[!IMPORTANT] Loading this checkpoint requires the xLLM code. The weights are stored in xLLM's native format (BF16 Safetensors with an xLLM
config.jsonandartifact_manifest.json). This is not a Hugging Facetransformerscheckpoint:AutoModel.from_pretrainedcannot load it. Get the code at https://github.com/ifm-ai/xllm-loop and load the directory withxllm.paper_part2.artifacts.load_artifact.
Model details
| Model | looped Transformer: 1 prelude, 2 recurrent and 1 coda block; width 1,536, 24 heads (6 KV heads), SwiGLU FFN 4,352 |
| Parameters | 303,579,648, including the embedding and output layers |
| Input injection | diagonal |
| Recurrent depth in training | depth drawn from a learned prior initialized at PLN-5, entropy coefficient lambda_H = 0.01 |
| Evaluation | R = 5 with terminal KV; the recurrent state is seeded by example identity (seed 42) |
| Training | recipe s_learned_entropy0p01: 5,120 updates of 512 x 8,192 tokens (21.47B), AdamW, warmup-stable-decay |
| Weights | BF16 Safetensors: the training checkpoint's FP32 weights rounded to BF16 |
| Tokenizer | tokenizer/, the Jais64k tokenizer of training |
Download
hf download IFM/LoopedLM-P2-huginn-s-learned-entropy0p01 --local-dir huginn-s-learned-entropy0p01
The xLLM loader checks the directory against artifact_manifest.json: it rejects symbolic links
and files the manifest does not list, apart from the .gitattributes file and the
.cache/huggingface/ folder that hf download --local-dir adds. Download into a directory as
above, not into the Hub cache (~/.cache/huggingface/hub), whose files are symbolic links.
Use
Evaluate with eval_paper_part2.py from the xLLM repository:
ENABLE_FLASH_ATTENTION_3=true python eval_paper_part2.py --artifact huginn-s-learned-entropy0p01 \
--data /path/to/eval-data/data.json --out out ppl
data.json and the evaluation inputs come from release/paper-part2/prepare-eval-data.py --output /path/to/eval-data.
xllm.paper_part2.artifacts.load_artifact("huginn-s-learned-entropy0p01") verifies the directory and returns the model
config fields and the BF16 state dict on CPU.
Distilled prefill students of this model (Table 4): distilled-s-random-init, distilled-s-teacher-init.
artifact_manifest.json records the size and SHA-256 of every file in this repository.
Paper and citation
Towards Looped Models Done Right. Part II: Rethinking at Fixed Points: https://arxiv.org/abs/2610.06833
@article{huang2026fixedpoints,
title = {Towards Looped Models Done Right, Part II: Rethinking at Fixed Points},
author = {Benhao Huang and Chufan Shi and Junlin Chen and Shicheng Wen and Zhengzhong Liu and Eric Xing and Xuezhe Ma},
journal = {arXiv preprint arXiv:2610.06833},
year = {2026}
}
License
The weights are released under the Apache License 2.0 (LICENSE); NOTICE records the
tokenizer's attribution and how the artifact was prepared. The xLLM code is distributed under its
own license.