GitHub repository: mahdi-shafiei/maxtext
Primary language: Python
Description: A simple, performant and scalable Jax LLM!