Main

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

Sign in to view logs

Run time

Learn about OS pricing on GitHub Actions

Job	Run time
Test (GPU)	2m 36s
Test checkpoint (GPU)	3m 56s
Lint	57s
Test MoE (GPU)	1m 7s
Test	1m 46s
Test checkpoint	1m 57s
Type check	51s
Build	52s
Style	58s
Docs	59s
Lint (min Python)	37s
Release	0s
	16m 36s