Improve linear algebra module, support for the derivative of Abs