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

bzantium wants to merge 1 commit intoAI-Hypercomputer:mainAI-Hypercomputer/maxtext:mainfrom

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