MaxText

MaxText는 TPU에서 학습과 추론을 수행할 수 있도록 JAX 기반 코드를 구성하는 고성능 프레임워크입니다. MaxCode는 PyTorch 모델과 코드를 MaxText와 호환되는 JAX 코드로 변환해 TPU 실행 환경으로의 이전을 돕습니다.