Add layer normalization

[aromanro]

See for example https://arxiv.org/abs/1607.06450