Add MuTransfer Capablity to RETRO model pretraining
# What does this PR do ?

It adds the muTransfer capablity to do RETRO model pretraining.
1. Included the mutransfer codes from MS github.
2. Add shape calculation script.
3. Add the mutransfer pretraining script.
4. Make sure it uses fused_softmax padding mode because of the customized masks
5. Include a CI test to make sure the model training performance matches. 

It also fix a bug that PTL grad sync is not called.

To use muTransfer for other models (GPT, T5, BERT etc), you can follow the same pattern.
1. register muadam or muadamw optimizer.
2. swap to use MuReadout to to convert model_d to logistics
3. Use the similar parameters init method in the mutransfer_pretrain file.
