Estimating Marginal Treatment Effects in Heterogeneous Populations