Machine learning (ML) assisted channel state information (CSI) reporting or ML assisted CSI prediction includes receiving CSI reporting configurations that include indications that enable or disable at least one of: ML-assisted CSI prediction and artificial intelligence channel feature information (AI-CFI) reporting. ML model training is performed or trained ML model parameters are received, and CSI reference signals corresponding to at least one of the CSI reporting configurations are received. If ML-assisted CSI prediction is enabled, the CSI reporting configurations further include: a timing offset for future CSI prediction, and ML configurations including indication of an ML model used for the ML-assisted CSI prediction. If AI-CFI reporting is enabled, the CSI reporting configurations further include: a configuration for a report of the AI-CFI, and ML configurations including indication of an ML model used for the ML assisted-CSI feedback determination.