GitHub repository: 0x7o/maxtext
Description: A simple, performant and scalable Jax LLM!