feat: custom callbacks for train() and return TrainOutput plus model_load_time#49

VassilisVassiliadis wants to merge 2 commits intofoundation-model-stack:mainfoundation-model-stack/fms-hf-tuning:mainfrom

VassilisVassiliadis:33_support_custom_callbacks_and_collect_metricsVassilisVassiliadis/fms-hf-tuning:33_support_custom_callbacks_and_collect_metricsCopy head branch name to clipboard