[WIP][Dion Official Optimizer, Muon] Integrate official Dion, and high speed Muon, optimizer impl with TorchTitan and Optimizer component class

[lessw2020]

The Dion authors have released their official Dion implementation here.
Thus we can move from my prev unofficial impl here to their official implementation.

This PR:
integrates their three main optimizer files with Titan Optimizer class and Titan configs to make it available.
Dion optimizer files live under experiments/dion_optimizer.
directly added to build_optimizer for now, will look at the proper subclassing later.
adds parameterization file to classify lm head, embeddings and 2D matrix to route things appropriately with appropriate scaling factors.

• adds logging for located lm head and embeddings to make sure these can be checked by user:

Testing:
8B llama3 trains nicely - needs more debugging to verify head, embedding, etc. are all being properly found.