normalize loss better with hidden variables with MPI