Main

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

Sign in to view logs

Run time

Learn about OS pricing on GitHub Actions

Job	Run time
Test (GPU)	59s
Lint	48s
Test checkpoint (GPU)	57s
Test	1m 14s
Test MoE (GPU)	58s
Test checkpoint	1m 13s
Type check	37s
Build	38s
Style	50s
Docs	1m 5s
Lint (min Python)	35s
Release	1s
	9m 55s