Skip to content

Support loading Tunix post-train checkpoints in MaxText pre-training - #4945

Draft
hsuan-lun-chiang wants to merge 35 commits into
mainfrom
feat/post-train-checkpoint-layout
Draft

Support loading Tunix post-train checkpoints in MaxText pre-training#4945
hsuan-lun-chiang wants to merge 35 commits into
mainfrom
feat/post-train-checkpoint-layout

Move Tunix checkpoint layout conversion to load time in pre-training

dc198a4
Select commit
Loading
Failed to load commit list.
Google CLA / cla/google succeeded Aug 19, 2026 in 20s

✅ All contributors are covered under a CLA with Google

See https://cla.developers.google.com/ for more info about Google's Contributor License Agreement (CLA).

ℹ️ Googlers: Go here to view more details and manage scans for this pull request.

Details

The following contributors were found for this pull request:

dc198a4 Author: @hsuan-lun-chiang <hsua********iang​@cienet.com>
9d84c37 Author: @ecnal-cienet <lan*****ng​@cienet.com>
e5b6f91 Author: @mesakhcienet <mesa********tian​@cienet.com>

(Only the first commit for a unique contributor is listed.)