maxtext
MaxText is an open-source LLM that enables efficient training and inference on Google Cloud TPUs and GPUs. With support for models such as Llama2, Mistral, and Gemma, it provides robust scalability and high Model Flops Utilization through Jax and XLA compiler integration, making it adaptable for diverse LLM applications.