Add Muon post-training support

[ashors1]

What does this PR do ?

NOTE: blocked by #1787

W&B report with latest experiments (all using adam-pretrained base models): https://wandb.ai/nvidia/ashors-muon/reports/Muon-for-Post-Training--VmlldzoxNTAzMzcwMA

Issues

List issues that this PR closes (syntax):

Usage

• You can potentially add a usage example below

# Add a code snippet demonstrating how to use this

Before your PR is "Ready for review"

Pre checks:

• Make sure you read and followed Contributor guidelines
• Did you write any new necessary tests?
• Did you run the unit tests and functional tests locally? Visit our Testing Guide for how to run tests
• Did you add or update any necessary documentation? Visit our Document Development Guide for how to write, build and test the docs.

Additional Information

• ...