Fix mlp weight init formula. Reported by Alex Rothberg