Various aspects of the present disclosure generally relate to wireless communication. A wireless communication device may have an apparatus that aligns the non-linearity between transmit chains of the wireless communication device that are driven by the same digital port. The apparatus may adjust an amplification power out or an amplification saturated power to adjust a ratio between the amplification saturated power and the amplification power out for one or more transmit chains of the wireless communication device. The apparatus may adjust the ratios of transmit chains to align the ratios of the transmit chains for more consistent management of non-linear characteristics of the chain components. Numerous other aspects are described.