DeepSpeed

2e856639 - Replace get_accelerator().amp().autocast() with torch.amp.autocast(device_type=...)