Expand verify to support weight comparisons

[pglusacTT]

Currently, verify function takes in two models, calculates forward pass and then compares the outputs of those models.

For training purposes, it would be convenient to have a function that also supports weight comparisons so that we are certain that the model is training properly.

This way, we'd have an easy way to make sure that the optimizers work properly and are updating the weights in a manner that resembles the host optimizers.

[nvukobratTT]

I.e. something to replace this and similar logic through the code (if more appear in time being):

• https://github.com/tenstorrent/tt-forge-fe/pull/1053/files#diff-8521f66e3a604362c8e5e4d4266c91633a868df29492617b7e4b70343eb85b0dR50