Integrate Multi-Token Prediction (MTP) Training objective#1837

copybara-service[bot] merged 1 commit intomainAI-Hypercomputer/maxtext:mainfrom

parambole/maxtext_mtp_training_obectiveAI-Hypercomputer/maxtext:parambole/maxtext_mtp_training_obectiveCopy head branch name to clipboard