linalg
================

.. py:module:: jaxns.internals.linalg

.. rubric:: :code:`jaxns.internals.linalg`


.. rubric:: Module Contents

.. py:function:: msqrt(A)

   Computes the matrix square-root using SVD, which is robust to poorly conditioned covariance matrices.
   Computes, M such that M @ M.T = A

   :param A: [N,N] Square matrix to take square root of.

   Returns: [N,N] matrix.


.. py:function:: squared_norm(x1, x2)