Methods, systems, and apparatus, including computer programs encoded on computer storage media, for correcting distortion of radio signals A transmit radio signal corresponding to an output of a transmitting radio signal processing system is obtained. A pre-distorted radio signal is then generated by processing the transmit radio signal using a nonlinear pre-distortion machine learning model. The nonlinear pre-distortion machine learning model includes model parameters and at least one nonlinear function to correct radio signal distortion or interference. A transmit output radio signal is obtained by processing the pre-distorted radio signal through the transmitting radio signal processing system. The transmit output radio signal is then transmitted to one or more radio receivers.