Refactor: extract shared post-training hooks and update SFT implementation#3862

copybara-service[bot] merged 1 commit into

mainAI-Hypercomputer/maxtext:mainfrom

igorts/post-train-hooks-refactorAI-Hypercomputer/maxtext:igorts/post-train-hooks-refactorCopy head branch name to clipboard