Implement AdamW Optimizer StepDeep-ML | Problem