feat: Added additional events such as on_step_begin, on_optimizer_step, on_substep_end#293

ashokponkumar merged 1 commit intofoundation-model-stack:mainfoundation-model-stack/fms-hf-tuning:mainfrom

seshapad:additional_eventsseshapad/fms-hf-tuning:additional_eventsCopy head branch name to clipboard