Implement Warmup-Stable-Decay (WSD) Learning Rate Schedule#2883

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

bzantium:feature/#2882bzantium/maxtext:feature/#2882Copy head branch name to clipboard