Methods, systems, and devices for wireless communications are described. Generally, the described techniques provide for communicating capability information (e.g., regarding neural network blocks supported by a user equipment (UE) and a base station). A base station may configure one or more neural network block parameters, and may transmit the neural network block parameters to the UE. The UE may configure or reconfigure a neural network block according to the neural network block parameters, and may process one or more signals, e.g., baseband signals, generated by the UE using the neural network block and the neural network block parameters.