feat: refactor init of dtensor policy v2#1709

terrykong merged 8 commits intomainNVIDIA-NeMo/RL:mainfrom

hemil/automodel-init-refactorNVIDIA-NeMo/RL:hemil/automodel-init-refactorCopy head branch name to clipboard