Options to scale gradients and to keep dtype in allreduce callback

[TakuyaNarihira]

Add scale_grad and keep_dtype option to allreduce callback to perform it for narrowed arrays.