🚀[FEA]: Generalize datapipe used in CoreDiff

[loliverhennigh]

Is this a new feature, an improvement, or a change to existing functionality?

New Feature

How would you describe the priority of this feature request

Critical (currently preventing usage)

Please provide a clear description of problem you would like to solve.

No response

Describe any alternatives you have considered

No response

[ram-cherukuri]

@loliverhennigh Is there an epic this task is associated with?