Short Course Q&A   Build and Train an LLM with JAX