GAM to learn non-linear transforms