matrix (ndata, nfeatures), loss should be specific to each column/feature.
matrix (ndata, nfeatures), loss should be specific to each column/feature.