More improvements to FSDP, benchmark against DDP #39

Sign in to view logs

Run time

Learn about OS pricing on GitHub Actions

Job	Run time
Lint	58s
Test	1m 40s
Test checkpoint	2m 0s
Test FSDP	2m 33s
Type check	46s
Build	57s
Style	48s
Lint (min Python)	51s
	10m 33s