Add param group weight decay

[3outeille]

• Can now apply weight decay to different params
• Make it compatible with MuTransfer
• Add support for SGD optimizer which make it easier to test params groups