[PyTorch] Use same API in optimizer zero_grad as PyTorch optimizers#1466

timmoon10 merged 2 commits into

NVIDIA:mainNVIDIA/TransformerEngine:mainfrom

timmoon10:optim-set-to-nonetimmoon10/TransformerEngine:optim-set-to-noneCopy head branch name to clipboard