After https://github.com/NVIDIA-NeMo/RL/pull/1143 is merged we will have this knob available for SFT, we should similarly support it on other algorithms since data loading may be exposed