Capability indication for a multi-block machine learning model

A method for wireless communications at a network entity, comprising:

• receiving, from a user equipment (UE), an indication identifying a first UE capability for a backbone block of a multi-block machine learning application and a second UE capability for a task-specific block of the multi-block machine learning application;
• transmitting, to the UE in response to the first UE capability and the second UE capability, control signaling to configure the multi-block machine learning application of the UE;
• and
• communicating with the UE based at least in part on the control signaling.

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, an indication identifying a first UE capability for a backbone block of a multi-block machine learning application and a second UE capability for a task-specific block of the multi-block machine learning application;
• receive, from the base station, control signaling identifying a configuration for the multi-block machine learning application of the UE at least in part in response to the indication identifying the first UE capability and the second UE capability;
• and
• communicate with the base station or a wireless device using the multi-block machine learning application configured in accordance with the control signaling.

An apparatus for wireless communications at a network entity, comprising:

• a processor;
• memory coupled with the processor;
• and
• instructions stored in the memory and executable by the processor to cause the apparatus to:
• receive, from a user equipment (UE), an indication identifying a first UE capability for a backbone block of a multi-block machine learning application and a second UE capability for a task-specific block of the multi-block machine learning application;
• transmit, to the UE in response to the first UE capability and the second UE capability, control signaling to configure the multi-block machine learning application of the UE;
• and
• communicate with the UE based at least in part on the control signaling.