Building a Jax training loop for an LLM training run #1 Post by gpjt » Tue, Jun 30, 2026, 5:59 PM UTC Building a Jax training loop for an LLM training rungilesthomas.com