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,99 @@
+# 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.special import (
+  bernoulli as bernoulli,
+  bessel_jn as bessel_jn,
+  beta as beta,
+  betainc as betainc,
+  betaln as betaln,
+  digamma as digamma,
+  entr as entr,
+  erf as erf,
+  erfc as erfc,
+  erfinv as erfinv,
+  exp1 as exp1,
+  expi as expi,
+  expit as expit,
+  expn as expn,
+  factorial as factorial,
+  gamma as gamma,
+  gammainc as gammainc,
+  gammaincc as gammaincc,
+  gammaln as gammaln,
+  gammasgn as gammasgn,
+  hyp1f1 as hyp1f1,
+  hyp2f1 as hyp2f1,
+  i0 as i0,
+  i0e as i0e,
+  i1 as i1,
+  i1e as i1e,
+  kl_div as kl_div,
+  log_ndtr as log_ndtr,
+  log_softmax as log_softmax,
+  loggamma as loggamma,
+  logit as logit,
+  logsumexp as logsumexp,
+  lpmn as _deprecated_lpmn,
+  lpmn_values as _deprecated_lpmn_values,
+  multigammaln as multigammaln,
+  ndtr as ndtr,
+  ndtri as ndtri,
+  poch as poch,
+  polygamma as polygamma,
+  rel_entr as rel_entr,
+  sici as sici,
+  softmax as softmax,
+  spence as spence,
+  sph_harm_y as sph_harm_y,
+  xlog1py as xlog1py,
+  xlogy as xlogy,
+  zeta as zeta,
+)
+
+from jax._src.third_party.scipy.special import (
+  fresnel as fresnel,
+)
+
+_deprecations = {
+    # Added Jan 3 2024
+    "lpmn": (
+        "jax.scipy.special.lpmn is deprecated; no replacement is planned.",
+        _deprecated_lpmn,
+    ),
+    "lpmn_values": (
+        "jax.scipy.special.lpmn_values is deprecated; no replacement is planned.",
+        _deprecated_lpmn_values,
+    ),
+    # Deprecated in v0.7.0, removed in v0.10.0.
+    # TODO(jakevdp): remove this entry in v0.11.0.
+    "sph_harm": (
+        "jax.scipy.special.sph_harm was deprecated in v0.7.0 and removed in"
+        " v0.10.0. Use jax.scipy.special.sph_harm_y instead.",
+        None
+    )
+}
+
+import typing as _typing
+if _typing.TYPE_CHECKING:
+  lpmn = _deprecated_lpmn
+  lpmn_values = _deprecated_lpmn_values
+else:
+  from jax._src.deprecations import deprecation_getattr as _deprecation_getattr
+  __getattr__ = _deprecation_getattr(__name__, _deprecations)
+  del _deprecation_getattr
+del _typing