More improvements to FSDP, benchmark against DDP #49

Sign in to view logs

Run time

Learn about OS pricing on GitHub Actions

Job	Run time
Lint	47s
Test	1m 34s
Test checkpoint	2m 3s
Test FSDP	2m 58s
Type check	55s
Build	53s
Style	49s
Lint (min Python)	56s
	10m 55s