A 125M custom GPT-X2 language model with RoPE, SwiGLU, grouped-query attention, and curriculum-trained code/math normalization.
Architecture graph for AxiomicLabs/GPT-S-5M.
Interactive architecture graph for AxiomicLabs/GPT-S-5M, visualized from Hugging Face model metadata.