hand

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

@@ -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

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

@@ -0,0 +1,18 @@
+# Copyright 2022 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.scipy.cluster import vq as vq

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

@@ -0,0 +1,18 @@
+# Copyright 2022 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.cluster.vq import vq as vq

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

@@ -0,0 +1,23 @@
+# Copyright 2021 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.fft import (
+  dct as dct,
+  dctn as dctn,
+  idct as idct,
+  idctn as idctn
+)

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

@@ -0,0 +1,20 @@
+# Copyright 2023 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.integrate import (
+  trapezoid as trapezoid
+)

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

@@ -0,0 +1,18 @@
+# Copyright 2022 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.
+
+# Already deprecate namespaces that will be removed in SciPy v2.0.0
+
+from jax._src.third_party.scipy.interpolate import (
+    RegularGridInterpolator as RegularGridInterpolator)

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,
+)

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

@@ -0,0 +1,20 @@
+# 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.ndimage import (
+  map_coordinates as map_coordinates,
+)

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

@@ -0,0 +1,21 @@
+# 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.optimize.minimize import (
+  minimize as minimize,
+  OptimizeResults as OptimizeResults,
+)

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

@@ -0,0 +1,29 @@
+# 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.signal import (
+  fftconvolve as fftconvolve,
+  convolve as convolve,
+  convolve2d as convolve2d,
+  correlate as correlate,
+  correlate2d as correlate2d,
+  detrend as detrend,
+  csd as csd,
+  istft as istft,
+  stft as stft,
+  welch as welch,
+)

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

@@ -0,0 +1,18 @@
+# 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.scipy.sparse import linalg as linalg

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

@@ -0,0 +1,22 @@
+# 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.sparse.linalg import (
+  cg as cg,
+  gmres as gmres,
+  bicgstab as bicgstab,
+)

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

@@ -0,0 +1,13 @@
+# Copyright 2023 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.

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

@@ -0,0 +1,21 @@
+# Copyright 2023 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.spatial.transform import (
+  Rotation as Rotation,
+  Slerp as Slerp,
+)

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

@@ -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

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

@@ -0,0 +1,45 @@
+# 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 jax.scipy.stats import bernoulli as bernoulli
+from jax.scipy.stats import beta as beta
+from jax.scipy.stats import binom as binom
+from jax.scipy.stats import cauchy as cauchy
+from jax.scipy.stats import dirichlet as dirichlet
+from jax.scipy.stats import expon as expon
+from jax.scipy.stats import gamma as gamma
+from jax.scipy.stats import geom as geom
+from jax.scipy.stats import laplace as laplace
+from jax.scipy.stats import logistic as logistic
+from jax.scipy.stats import multinomial as multinomial
+from jax.scipy.stats import multivariate_normal as multivariate_normal
+from jax.scipy.stats import nbinom as nbinom
+from jax.scipy.stats import norm as norm
+from jax.scipy.stats import pareto as pareto
+from jax.scipy.stats import poisson as poisson
+from jax.scipy.stats import t as t
+from jax.scipy.stats import uniform as uniform
+from jax.scipy.stats import chi2 as chi2
+from jax.scipy.stats import betabinom as betabinom
+from jax.scipy.stats import gennorm as gennorm
+from jax.scipy.stats import truncnorm as truncnorm
+from jax._src.scipy.stats.kde import gaussian_kde as gaussian_kde
+from jax._src.scipy.stats._core import mode as mode, rankdata as rankdata, sem as sem
+from jax.scipy.stats import vonmises as vonmises
+from jax.scipy.stats import wrapcauchy as wrapcauchy
+from jax.scipy.stats import gumbel_r as gumbel_r
+from jax.scipy.stats import gumbel_l as gumbel_l

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

@@ -0,0 +1,23 @@
+# 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.stats.bernoulli import (
+  logpmf as logpmf,
+  pmf as pmf,
+  cdf as cdf,
+  ppf as ppf
+)

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

@@ -0,0 +1,25 @@
+# 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.stats.beta import (
+  cdf as cdf,
+  logcdf as logcdf,
+  logpdf as logpdf,
+  logsf as logsf,
+  pdf as pdf,
+  sf as sf,
+)

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

@@ -0,0 +1,21 @@
+# Copyright 2021 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.stats.betabinom import (
+  logpmf as logpmf,
+  pmf as pmf,
+)

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

@@ -0,0 +1,18 @@
+# Copyright 2023 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.
+
+from jax._src.scipy.stats.binom import (
+  logpmf as logpmf,
+  pmf as pmf,
+)

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

@@ -0,0 +1,27 @@
+# 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.stats.cauchy import (
+  cdf as cdf,
+  isf as isf,
+  logcdf as logcdf,
+  logpdf as logpdf,
+  logsf as logsf,
+  pdf as pdf,
+  ppf as ppf,
+  sf as sf,
+)

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

@@ -0,0 +1,25 @@
+# Copyright 2021 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.stats.chi2 import (
+  cdf as cdf,
+  logcdf as logcdf,
+  logpdf as logpdf,
+  logsf as logsf,
+  pdf as pdf,
+  sf as sf,
+)

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

@@ -0,0 +1,21 @@
+# 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.stats.dirichlet import (
+  logpdf as logpdf,
+  pdf as pdf,
+)

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

@@ -0,0 +1,26 @@
+# 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.stats.expon import (
+  cdf as cdf,
+  logcdf as logcdf,
+  logpdf as logpdf,
+  logsf as logsf,
+  pdf as pdf,
+  ppf as ppf,
+  sf as sf,
+)

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

@@ -0,0 +1,25 @@
+# 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.stats.gamma import (
+  cdf as cdf,
+  logcdf as logcdf,
+  logpdf as logpdf,
+  logsf as logsf,
+  pdf as pdf,
+  sf as sf,
+)

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

@@ -0,0 +1,22 @@
+# Copyright 2022 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.stats.gennorm import (
+  cdf as cdf,
+  logpdf as logpdf,
+  pdf as pdf,
+)

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

@@ -0,0 +1,21 @@
+# 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.stats.geom import (
+  logpmf as logpmf,
+  pmf as pmf,
+)

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

@@ -0,0 +1,26 @@
+# Copyright 2025 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.stats.gumbel_l import (
+  logpdf as logpdf,
+  pdf as pdf,
+  logcdf as logcdf,
+  cdf as cdf,
+  ppf as ppf,
+  sf as sf,
+  logsf as logsf
+)

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

@@ -0,0 +1,26 @@
+# Copyright 2025 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.stats.gumbel_r import (
+  logpdf as logpdf,
+  pdf as pdf,
+  logcdf as logcdf,
+  cdf as cdf,
+  ppf as ppf,
+  sf as sf,
+  logsf as logsf
+)

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

@@ -0,0 +1,22 @@
+# 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.stats.laplace import (
+  cdf as cdf,
+  logpdf as logpdf,
+  pdf as pdf,
+)

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

@@ -0,0 +1,25 @@
+# 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.stats.logistic import (
+  cdf as cdf,
+  isf as isf,
+  logpdf as logpdf,
+  pdf as pdf,
+  ppf as ppf,
+  sf as sf,
+)

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

@@ -0,0 +1,21 @@
+# Copyright 2022 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.stats.multinomial import (
+  logpmf as logpmf,
+  pmf as pmf,
+)

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

@@ -0,0 +1,21 @@
+# 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.stats.multivariate_normal import (
+  logpdf as logpdf,
+  pdf as pdf,
+)

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

@@ -0,0 +1,18 @@
+# Copyright 2021 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.
+
+from jax._src.scipy.stats.nbinom import (
+  logpmf as logpmf,
+  pmf as pmf,
+)

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

@@ -0,0 +1,27 @@
+# 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.stats.norm import (
+  cdf as cdf,
+  logcdf as logcdf,
+  logpdf as logpdf,
+  logsf as logsf,
+  pdf as pdf,
+  ppf as ppf,
+  sf as sf,
+  isf as isf,
+)

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

@@ -0,0 +1,26 @@
+# 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.stats.pareto import (
+  logcdf as logcdf,
+  logpdf as logpdf,
+  logsf as logsf,
+  cdf as cdf,
+  pdf as pdf,
+  ppf as ppf,
+  sf as sf,
+)

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

@@ -0,0 +1,23 @@
+# 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.stats.poisson import (
+  logpmf as logpmf,
+  pmf as pmf,
+  cdf as cdf,
+  entropy as entropy
+)

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

@@ -0,0 +1,21 @@
+# 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.stats.t import (
+  logpdf as logpdf,
+  pdf as pdf,
+)

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

@@ -0,0 +1,25 @@
+# Copyright 2022 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.stats.truncnorm import (
+  cdf as cdf,
+  logcdf as logcdf,
+  logpdf as logpdf,
+  pdf as pdf,
+  logsf as logsf,
+  sf as sf
+)

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

@@ -0,0 +1,23 @@
+# 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.stats.uniform import (
+  logpdf as logpdf,
+  pdf as pdf,
+  cdf as cdf,
+  ppf as ppf,
+)

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

@@ -0,0 +1,21 @@
+# Copyright 2022 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.stats.vonmises import (
+  logpdf as logpdf,
+  pdf as pdf,
+)

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

@@ -0,0 +1,21 @@
+# Copyright 2023 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.stats.wrapcauchy import (
+  logpdf as logpdf,
+  pdf as pdf,
+)