Dropout dtype

In mlp.apply_dropout even though mask is respecting state.dtype, scale is not. This fixes that.