feat: refactor common data utilities of dtensor policy v2#1710

yuki-97 merged 7 commits intomainNVIDIA-NeMo/RL:mainfrom

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