Add extra tests for SparseLinear and fix bug where scale was not being multiplied into bias updates