| I trained a small transformer in 1.5hrs and it beats many LLMs(mvakde.github.io) | |
| 651 points by porridgeraisin 9 days ago | 162 comments | |
tl;dr: A researcher trained a small transformer from scratch on a 5090 GPU in 1.5 hours for $0.67, scoring 45% on ARC-1 and 7% on ARC-2, matching TRM/HRM without recursion. Key improvements over the prior version include modern architecture (SwiGLU, RMSNorm), NorMuon optimizer, 3D RoPE with per-task embeddings, and supervised training on output tokens only. The author argues recursion isn't necessary, criticizes reliance on synthetic data and augmentations in other ARC approaches, and believes 65% is achievable within the transformer framework. | |
HN Discussion:
| |