Скачать или смотреть MaxText: A High-Performance and Scalable Jax LLM for AI Enthusiasts

SUBSCRIBE CHANNEL: https://bit.ly/AIInsightNews
-----------------
MaxText is an open-source Large Language Model (LLM) written in Python/Jax, designed for high performance and scalability on Google Cloud TPUs and GPUs. It supports training and inference for models like Llama2, Mistral, and Gemma. Users can experiment with MaxText out of the box and customize it to meet their needs. Comments discuss comparisons with other models like EasyLM, Levanter, and T5X, highlighting issues with each. Some users find MaxText difficult to work with due to unnecessary layers of abstraction, while others appreciate its scalability and performance. There is also a discussion on the complexity of MaxText and the limitations of JAX in saving compiled functions.

🔗 https://github.com/google/maxtext

#AI #LLM