Normalizing X and scaling gradients
Normalizing X and scaling gradients