jax-with-mha-bias-larger-chinchilla-1 — LLM | Genaihub