Integrate new model builder interface into training loop

[maanug-nv]

Needs to be done for both Megatron Bridge and Megatron Training