Estimating a non-neutral production function: a heterogeneous treatment effect approach