feat: add numpy.linalg functions (#77)

Signed-off-by: nstarman <nstarman@users.noreply.github.com>
GalacticDynamics · Sep 15, 2024 · 4ffe7f2 · 4ffe7f2
1 parent d289c6f
commit 4ffe7f2
Showing 1 changed file with 24 additions and 0 deletions.
diff --git a/src/quaxed/numpy/linalg.py b/src/quaxed/numpy/linalg.py
@@ -1,7 +1,31 @@
 """Quaxed :mod:`jax.numpy.linalg`."""
 
 __all__ = [  # noqa: F822
+    "cholesky",
+    "cross",
     "det",
+    "diagonal",
+    "eig",
+    "eigh",
+    "eigvalsh",
+    "inv",
+    "matmul",
+    "matrix_norm",
+    "matrix_power",
+    "matrix_rank",
+    "matrix_transpose",
+    "outer",
+    "pinv",
+    "qr",
+    "slogdet",
+    "solve",
+    "svd",
+    "svdvals",
+    "tensordot",
+    "trace",
+    "vecdot",
+    "vector_norm",
+    "norm",
 ]
 
 import sys