let square_grad x y y' =   let open MX in   (transpose x) $@ (y' -@ y) /$ (float_of_int (row_num x))