feat(trainer): Add PyTorch Profiler integration to CustomTrainer#352

SoumyaRaikwar wants to merge 1 commit intokubeflow:mainkubeflow/sdk:mainfrom

SoumyaRaikwar:pytorch-profilerSoumyaRaikwar/sdk:pytorch-profilerCopy head branch name to clipboard