looped-32L-d640-1e18

32-layer compute-optimal checkpoint for Sparse Layers are Critical to Scaling Looped Language Models (arXiv:2605.09165).

architecture looped
d_model 640
effective layers 32
compute budget 1e18 FLOPs
training steps 45,632
parameters (stored) 143,627,520
peak LR 0.005
batch size 16
muP width_ratio 2.5 (d_base=256)
final val loss n/a

This width is the architecture's measured minimum on a ten-rung isoFLOP sweep (d256-d1408) at 1e18 FLOPs.

Usage

from transformers import AutoModelForCausalLM, AutoTokenizer
m = AutoModelForCausalLM.from_pretrained(
    "ml-ryanlee/looped-32L-d640-1e18", trust_remote_code=True)
tok = AutoTokenizer.from_pretrained("gpt2")

Pass max_length=1024 when evaluating — the RoPE buffer is sized to the training context.

Downloads last month
23
Safetensors
Model size
0.1B params
Tensor type
F32
·
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Paper for ml-ryanlee/looped-16x2-d640-1e18