DDP refactoring: Extract parameter layout computation into optimizer classmethod#3812

deepakn94 wants to merge 1 commit intoNVIDIA:mainNVIDIA/Megatron-LM:mainfrom

deepakn94:dnarayanan/refactor_param_mappingdeepakn94/Megatron-LM:dnarayanan/refactor_param_mappingCopy head branch name to clipboard