A dynamic UE capability message specifying a number of beams to be evaluated within different groups of UE TX beam directions associated with different antenna arrays of a UE may be communicated to a transmit receive point (TRP). Each group of UE TX beam directions may include beam directions that are candidates for uplink data transmission. Based on information in the dynamic UE capability message, the TRP can assign SRS resources (for uplink beam management or uplink channel sounding) and/or schedule a multi-layer MIMO uplink transmission in accordance with a UE capability constraint. This may ensure that the SRS resources assignment and/or scheduled multi-layer MIMO uplink transmission is consistent with the UE's beamforming capabilities.