[rl] Update callsite to init_batch_invariance to pass attention backend.#2176

zhxchen17 wants to merge 1 commit intomainpytorch/torchtitan:mainfrom

zhxchen17/init_batch_invariancepytorch/torchtitan:zhxchen17/init_batch_invarianceCopy head branch name to clipboard