jax

8fc7e2b7 - [0.6.0-UT] Fix NumpyLinalgTest test cases (#453)

[0.6.0-UT] Fix NumpyLinalgTest test cases (#453) Co-authored-by: Pham Binh <phambinh@w.amd.com> (cherry picked from commit 29c392d03e9eb217a33bb73046f4b508750140c1)