Main

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

Sign in to view logs

Run time

Learn about OS pricing on GitHub Actions

Job	Run time
Test (GPU)	45s
Lint	34s
Test checkpoint (GPU)	43s
Test	49s
Test MoE (GPU)	45s
Test checkpoint	49s
Type check	50s
Build	39s
Style	53s
Docs	57s
Lint (min Python)	47s
Release	1s
	8m 32s