Remove undocumented fp8 NaN sanitization from train_step#3721

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

fix/gate-fp8-nan-sanitizationAI-Hypercomputer/maxtext:fix/gate-fp8-nan-sanitizationCopy head branch name to clipboard