GitLab

Fix dispatch where we have a parameter group with multiple
combinations of types
Optionally apply weight decay after momentum