[RL] Changes to enable compilation for trainer#2568

Lucaskabela wants to merge 2 commits intopytorch:mainpytorch/torchtitan:mainfrom

Lucaskabela:lucaskabela/enable_trainer_compile_03_10Lucaskabela/torchtitan:lucaskabela/enable_trainer_compile_03_10Copy head branch name to clipboard