According to some embodiments, a method performed by a network node for mobility management comprises obtaining data samples for modeling a wireless network environment that comprises a plurality of cells and building a machine learning model of the wireless network using the obtained data samples. The machine learning model is trained to determine a sequence of handovers for a wireless device among the plurality of cells for the wireless device to traverse from a source cell to a destination cell. The method further comprises receiving mobility information for a wireless device, determining one or more handover operations for the wireless device based on the mobility information, and transmitting the one or more handover operations to the wireless device.