fix weight initialization in nv_mlp