Add layer renormalization.
Fixed parameters, dropout, some other stuff.