G github.com Maxtext: A simple, performant and scalable Jax LLM 115 points zerojames 2 years ago 7 comments