Rapid CO2-water multi-phase flow prediction with machine learning