Embodiments of a User Equipment (UE), a generation Node-B (gNB) and methods for communication are generally described herein. The UE may the UE 102 may receive from a gNB, control signaling that indicates: a first pre-coding matrix indicator (PMI) that indicates a first pre-coder for a first sub-band; and a second PMI that indicates a second pre-coder for a second sub-band. The first and second pre-coders may be included in predetermined candidate pre-coders. The UE may encode a physical uplink shared channel (PUSCH) block for transmission. The UE may scale the PUSCH block in the first sub-band by the first pre-coder. The UE may scale the PUSCH block in the second sub-band by the second pre-coder.