Linear Regression: Add grouping support and heterskedasticity

PIVOTAL: 40576747

Added new api for linear regression along with support for grouping
columns. Also added computation of heteroskedasticity values as an
aggregate.