Methods and devices utilizing artificial intelligence (AI) or machine learning (ML) for customization of a device specific air interface configuration in a wireless communication network are provided. An over the air information exchange to facilitate the training of one or more AI/ML modules involves the exchange of AI/ML capability information identifying whether a device supports AI/ML for optimization of the air interface.