Pre-training Decoder-based Tiny LLM with JAX and TPU