Building a Jax training loop for an LLM training run

1 points | by gpjt 6 hours ago