Systems, methods, apparatuses, and computer program products for a user equipment downlink transmission beam prediction framework with machine learning are provided. For example, a method can include indicating a beam prediction capability of a user equipment to a network. The method can also include receiving network antenna configuration identification responsive to the indicated beam prediction capability. The method can further include selecting a model for beam prediction based on the network antenna configuration identification. The method can additionally include receiving a configuration of reference signal resources. The method can also include measuring reference signals based on the configuration of reference signal resources. The method can further include performing beam prediction based on the measured references signals using the selected model. The method can additionally include reporting a beam prediction to the network.