Paz, Octavio, Escritor, México