A method of training an artificial intelligence (AI) model including obtaining M training pairs and training the AI model using the M training pairs. Obtaining the M training pairs includes inputting an mth set of injection values in a reservoir simulator and an mth set of output values from the reservoir simulator. A method of extracting hydrocarbons from and storing carbon dioxide in a formation including, for N cycles, inputting an nth set of output values into a trained AI model, obtaining an (n+1)th set of injection values from the trained AI model, injecting an (n+1)th amount of carbon dioxide and an (n+1)th amount of treatment fluid into a hydrocarbon reservoir based on the (n+1)th set of injection values, determining an (n+1)th set of output values from the formation, and rewarding or penalizing the trained AI model based on the nth set and the (n+1)th set of output values.