Before revision:  We first estimate marginal effects and treatment propensities  to form an objective function that isolates the heterogeneous treatment effects, and then optimize the learned objective .
After revision:  We first estimate marginal effects and treatment propensities in order to form an objective function that isolates the heterogeneous treatment effects, and then optimize the learned objective .