From 4ffe7f26c3cea75b8475c704c9d00efa667e0b0e Mon Sep 17 00:00:00 2001 From: Nathaniel Starkman Date: Sat, 14 Sep 2024 23:45:43 -0400 Subject: [PATCH] feat: add numpy.linalg functions (#77) Signed-off-by: nstarman --- src/quaxed/numpy/linalg.py | 24 ++++++++++++++++++++++++ 1 file changed, 24 insertions(+) diff --git a/src/quaxed/numpy/linalg.py b/src/quaxed/numpy/linalg.py index e80a15b..290ecf7 100644 --- 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