A method of wireless communication, by a user equipment (UE), includes receiving multiple neural network training configurations for channel state feedback (CSF). Each configuration corresponds to a different neural network framework. The method also includes training each of a group of neural network decoder/encoder pairs in accordance with the received training configurations. A method of wireless communication, by a base station, includes transmitting multiple neural network training configurations to a user equipment (UE) for channel state feedback (CSF). Each configuration corresponds to a different neural network framework. The method also includes receiving a neural network decoder/encoder pair trained in accordance with the training configurations.