https://stephenallwright.com/l2-loss-function/