JAX와 TPU를 이용한 Decoder-based Tiny LLM 사전학습