let square_loss y y' =   let open MX in   average_rows ((y' -@ y) **@ 2.)