Shypd.aiThe largest open index of AI toolsMaxText is an open-source library for building high-performance and scalable Large Language Models (LLMs) using JAX. It provides reference implementations for various LLMs and supports pre-training and post-training on Google Cloud TPUs and GPUs.
Trending
Also listed in
About
MaxText is a high-performance, highly scalable, open-source library for Large Language Models (LLMs), implemented in pure Python/JAX. It is designed to run efficiently on Google Cloud TPUs and GPUs, supporting both pre-training and scalable post-training with techniques like Supervised Fine-Tuning (SFT) and Reinforcement Learning (GRPO, GSPO). MaxText achieves high Model FLOPs Utilization (MFU) and tokens/second across various cluster sizes, leveraging the power of JAX and the XLA compiler. It offers a library of high-performance models including Gemma, Llama, DeepSeek, Qwen, and Mistral, and serves as a launching point for ambitious LLM projects in research and production. Users can experiment with MaxText out of the box or fork and modify it to meet specific needs, with support for multi-modal training.
Capabilities
Pricing & Plans
Open Source
Free
FAQs
Trending
