hand

2026-05-06 19:47:31 +07:00
parent 94d8682530
commit 12dbb7731b
9963 changed files with 2747894 additions and 0 deletions
@@ -0,0 +1,47 @@
+# Copyright 2018 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 typing import TYPE_CHECKING
+
+if TYPE_CHECKING:
+  from jax.scipy import interpolate as interpolate
+  from jax.scipy import linalg as linalg
+  from jax.scipy import ndimage as ndimage
+  from jax.scipy import signal as signal
+  from jax.scipy import sparse as sparse
+  from jax.scipy import special as special
+  from jax.scipy import stats as stats
+  from jax.scipy import fft as fft
+  from jax.scipy import cluster as cluster
+  from jax.scipy import integrate as integrate
+else:
+  import jax._src.lazy_loader as _lazy
+  __getattr__, __dir__, __all__ = _lazy.attach(__name__, [
+    "interpolate",
+    "linalg",
+    "ndimage",
+    "signal",
+    "sparse",
+    "special",
+    "stats",
+    "fft",
+    "cluster",
+    "integrate",
+  ])
+  del _lazy
+
+del TYPE_CHECKING