Skip to content

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

Draft
hsuan-lun-chiang wants to merge 32 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 32 commits into
mainfrom
feat/post-train-checkpoint-layout

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

d54d6fa
Select commit
Loading
Failed to load commit list.