Also while were here, update to some more modern TFF coding practices:
-  Use `tff.learning.framework.ModelWeights` instead of a custom type.
-  Delete `from_tff_result` conversion helper, it is no longer needed.
-  Remove unused test helper methods.
-  Replace numpy dependency with TF ops.

PiperOrigin-RevId: 322359152