Neural network configuration for wireless communication system assistance

An apparatus for wireless communications at a user equipment (UE), comprising:

• a processor,
• memory coupled with the processor;
• and
• instructions stored in the memory and executable by the processor to cause the apparatus to:
• transmit, to a base station, capability information indicating one or more neural network blocks supported by the UE;
• receive, from the base station, control signaling with one or more neural network block parameters based at least in part on the transmitting of the capability information;
• process one or more signals generated by the UE using a first neural network block of the one or more neural network blocks and the one or more neural network block parameters
• receive, from the base station, additional control signaling including one or more additional neural network block parameters;
• determine a priority status of the one or more additional neural network block parameters;
• and
• transmit, to the base station, a negative acknowledgement message in response to the one or more additional neural network block parameters failing to be successfully decoded and based at least in part on the priority status.