let
l2 p =
MX
.(0.5 $* (average_rows (p *@ p)))