Methods, systems, and devices for wireless communications are described. A base station may explicitly or implicitly identify one or more parameters associated with a user equipment (UE), and may classify the UE into a cluster of UEs based on identifying the one or more parameters. The base station may determine a customized function block based on classifying the UE into the cluster of UEs. For example, the base station may determine the customized function block based on determining whether the cluster of UEs is associated with a pre-trained customized function block. The base station may then transmit the customized function block to the UE. The UE may receive the customized function block and may process one or more signals at the UE based on the received customized function block.