Building a Mini-GPT From Scratch: A JAX and Flax Guide to Small Language Models

Everyone's talking about small language models (SLMs) right now distilled, quantized, edge-deployable versions of the giants that still punch above their weight. But most tutorials on building one sto