Stability of the Kim--Milman flow map