🚀[FEA]: Modularize CorrDiff inference and move into the main package

[akshaysubr]

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.

The CorrDiff example provides a generate.py script for inference. It would be very useful to have the core logic of this moved into the main modulus package so it can be reused across applications.

Describe any alternatives you have considered

Duplicate the inference script in other applications which is undesirable