Optimzing Our Jax LLM RL Pipeline
notes.kvfrans.com