LLM pretraining's next step: bringing data into architecture-optimizer co-design

akbirthko · x · 2026-11-04

The author frames industrial LLM pretraining around three interlocking circles: architecture, data, and optimization. Co-design of architecture and optimization has advanced via custom optimizers for embeddings, LM heads, modules like Over-Encoding and Engram, and Muon for selected Transformer weight matrices. The next step is folding data into the co-design — e.g., jointly designing data schedules with learning-rate schedules — noting n-gram modules' training dynamics already depend on data.

Original post →

More from Research

Research channel →