Methods and devices for providing a feedback network in a multi-stage power amplifier are described. According to one aspect, a final amplifier of the multi-stage power amplifier is a cascode amplifier. The feedback network is placed between an output of the final amplifier and an output of a driver amplifier. The feedback network can decrease a mismatch between the output impedance of the final amplifier and a load presented to the final amplifier. In addition, the feedback network can change a load presented to the driver amplifier and thereby allow the transfer functions of each stage to be tuned so that the overall transfer function of the multi-stage amplifier becomes more linear.