Allow FSDP to use with torch.autocast for bfloat16 mixed precision
#3416
| Job | Run time |
|---|---|
| 2m 20s | |
| 2m 20s |
torch.autocast for bfloat16 mixed precision
#3416
| Job | Run time |
|---|---|
| 2m 20s | |
| 2m 20s |