joelogan commited on
Commit
25c3a8b
·
verified ·
1 Parent(s): f624d77

Upload folder using huggingface_hub

Browse files
Files changed (2) hide show
  1. README.md +60 -0
  2. model.pt +3 -0
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