A method of beam reciprocity state reporting and uplink beam management in wireless communication systems with beamforming is proposed. In one novel aspect, a BS configures one or more resource sets to a UE for uplink beam management. The one or more resource sets are allocated for UE to transmit UL reference signals using a number of UE beams. The number of UE beams to be trained is reported by the UE, e.g., via a “UE beam reciprocity state (update)” message. The BS also indicates whether a fixed UE TX beam or which UE TX beam is used for transmission, or indicates whether different UE TX beams are used for transmission of different resources in a resource set. The BS then feedback measurement results for UE to choose a proper TX beam.