Various aspects of the present disclosure relate to artificial intelligence (AI) enabled beam management. A base station may notify a user equipment (UE) of an availability of AI-enabled beam management and provide configuration parameters for an AI algorithm. The UE may utilize the configuration parameters to configure an AI algorithm. The UE may process a first set of beams utilizing the configured AI algorithm to identify a second set of beams. The UE may notify the base station (and/or other network entity) of the second set of beams, and one or more beams of the second set of beams can be utilized for wireless communication between the UE and the base station and/or other network entity or UE.