hand

venv/lib/python3.12/site-packages/jax/scipy/linalg.py

@@ -0,0 +1,51 @@
+# Copyright 2020 The JAX Authors.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     https://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# Note: import <name> as <name> is required for names to be exported.
+# See PEP 484 & https://github.com/jax-ml/jax/issues/7570
+
+from jax._src.scipy.linalg import (
+  block_diag as block_diag,
+  cholesky as cholesky,
+  cho_factor as cho_factor,
+  cho_solve as cho_solve,
+  det as det,
+  eigh as eigh,
+  eigh_tridiagonal as eigh_tridiagonal,
+  expm as expm,
+  expm_frechet as expm_frechet,
+  hessenberg as hessenberg,
+  hankel as hankel,
+  hilbert as hilbert,
+  inv as inv,
+  lu as lu,
+  lu_factor as lu_factor,
+  lu_solve as lu_solve,
+  pascal as pascal,
+  polar as polar,
+  qr as qr,
+  qr_multiply as qr_multiply,
+  rsf2csf as rsf2csf,
+  schur as schur,
+  sqrtm as sqrtm,
+  solve as solve,
+  solve_sylvester as solve_sylvester,
+  solve_triangular as solve_triangular,
+  svd as svd,
+  toeplitz as toeplitz,
+)
+
+from jax._src.third_party.scipy.linalg import (
+  funm as funm,
+)