MaxText는 JAX로 작성된 단순하면서도 고성능·확장성을 갖춘 대규모 언어 모델 학습 프레임워크입니다. AI-Hypercomputer 프로젝트의 일부로 TPU와 GPU에서 대규모 분산 학습을 효율적으로 수행하도록 설계되었습니다. Llama, Gemma, Mistral, Mixtral, DeepSeek 등 다양한 모델 아키텍처의 사전학습과 미세조정(SFT)을 지원하며, 순수 Python과 JAX로 구현되어 코드 가독성과 수정 용이성이 높습니다. 높은 처리량과 명료한 코드베이스를 동시에 추구하는 연구·엔지니어링 워크플로우에 적합한 LLM 학습 도구입니다.