Update optim.grad_scaler to use torch.amp#615

mryab wants to merge 1 commit intomasterlearning-at-home/hivemind:masterfrom

update_grad_scalerlearning-at-home/hivemind:update_grad_scalerCopy head branch name to clipboard