Upload folder using huggingface_hub
Browse files
README.md
ADDED
|
@@ -0,0 +1,60 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
license: mit
|
| 3 |
+
library_name: pytorch
|
| 4 |
+
tags:
|
| 5 |
+
- depth-recurrent
|
| 6 |
+
- weight-tied
|
| 7 |
+
- looped-transformer
|
| 8 |
+
---
|
| 9 |
+
|
| 10 |
+
# Depth-Recurrent Transformer (S1, step 49209)
|
| 11 |
+
|
| 12 |
+
Final recursive checkpoint for the paper *Per-Token Fixed-Point Convergence in
|
| 13 |
+
Depth-Recurrent Transformers, and Why Reading It Beats Learning It*.
|
| 14 |
+
|
| 15 |
+
A depth-recurrent (weight-tied, looped) transformer: prelude blocks, a
|
| 16 |
+
weight-tied core looped `r` times with the recursion count sampled per
|
| 17 |
+
optimization step during pretraining, then coda blocks. Trained on FineWeb-Edu
|
| 18 |
+
for 12.9B tokens.
|
| 19 |
+
|
| 20 |
+
- Parameters: 85.6M (state dict); ~57.3M active per forward at r=1
|
| 21 |
+
- Training tokens: 12,900,106,240
|
| 22 |
+
- Step: 49209
|
| 23 |
+
|
| 24 |
+
## Contents
|
| 25 |
+
|
| 26 |
+
This is an inference checkpoint (`config` + `state_dict` only; optimizer and RNG
|
| 27 |
+
state removed). It is a `torch.save` dict and requires the model code from the
|
| 28 |
+
[code release](https://github.com/jlognn/depth-recurrent-convergence) to load.
|
| 29 |
+
|
| 30 |
+
```python
|
| 31 |
+
import torch
|
| 32 |
+
from src.model import build_model
|
| 33 |
+
|
| 34 |
+
ck = torch.load("model.pt", map_location="cpu", weights_only=False)
|
| 35 |
+
model = build_model(ck["config"])
|
| 36 |
+
model.load_state_dict(ck["state_dict"])
|
| 37 |
+
model.eval()
|
| 38 |
+
|
| 39 |
+
# recursive forward takes a loops= kwarg
|
| 40 |
+
logits = model(input_ids, loops=8)
|
| 41 |
+
```
|
| 42 |
+
|
| 43 |
+
Or evaluate directly with the release harness:
|
| 44 |
+
|
| 45 |
+
```bash
|
| 46 |
+
python bench/benchmark.py --ckpt model.pt --tokens data/fwe-val.bin --loops 1 2 4 8 16 32
|
| 47 |
+
python bench/allocate.py --ckpt model.pt --tokens data/fwe-val.bin --diagnostic
|
| 48 |
+
```
|
| 49 |
+
|
| 50 |
+
## Citation
|
| 51 |
+
|
| 52 |
+
```
|
| 53 |
+
@misc{logan2026pertoken,
|
| 54 |
+
title = {Per-Token Fixed-Point Convergence in Depth-Recurrent Transformers,
|
| 55 |
+
and Why Reading It Beats Learning It},
|
| 56 |
+
author = {Logan, Joe},
|
| 57 |
+
year = {2026},
|
| 58 |
+
eprint = {arXiv:XXXX.XXXXX}
|
| 59 |
+
}
|
| 60 |
+
```
|
model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:571b8ed34c7ab9671985586158c6c13fc027eba542095dc41177ef5d1f9e2265
|
| 3 |
+
size 229230136
|