cayley-10b-k16-3l-mlp_in

Architecture

• 205M-param GPT (12 layers, 8 heads, d=1024)
• CayleySAE inserted at mlp_in in every block (3-level hierarchy)

CayleySAE config

• Total features per layer: 74,752 (73x overcomplete)
• Active per token: 112
• per_parent_budget=True, score_standardize=True
• Location: mlp_in (RMSNorm → CayleySAE → MLP in each block)

Training config

optimizer: muon (lr=0.006), adamw (lr=0.006)
lr_schedule: linear_warmdown, warmdown_frac=0.2
batch_size: 80, seq_len: 1024, gradient_accumulation_steps: 1
max_iters: 16000