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