pytorch

2e761498 - Revert "`GradScaler` recomputes `optimizer_state["found_inf_per_device"]` before `optimizer.step` (#97415) (#97886)" (#98624)