Main

Add more options to the `unshard_checkpoint` function to help scale #1049

Sign in to view logs

Run time

Learn about OS pricing on GitHub Actions

Job	Run time
Test (GPU)	3m 4s
Lint	38s
Test checkpoint (GPU)	2m 44s
Test MoE (GPU)	2m 23s
Test	1m 51s
Test checkpoint	1m 37s
Type check	54s
Build	39s
Style	37s
Docs	1m 2s
Lint (min Python)	1m 14s
Release	0s
	16m 43s