mirror of
https://github.com/ROCm/jax.git
synced 2025-04-14 10:56:06 +00:00
463 lines
6.3 KiB
ReStructuredText
463 lines
6.3 KiB
ReStructuredText
``jax.scipy`` module
|
|
====================
|
|
|
|
jax.scipy.fft
|
|
-------------
|
|
|
|
.. automodule:: jax.scipy.fft
|
|
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
dct
|
|
dctn
|
|
idct
|
|
idctn
|
|
|
|
jax.scipy.integrate
|
|
-------------------
|
|
|
|
.. automodule:: jax.scipy.integrate
|
|
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
trapezoid
|
|
|
|
jax.scipy.linalg
|
|
----------------
|
|
|
|
.. automodule:: jax.scipy.linalg
|
|
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
block_diag
|
|
cho_factor
|
|
cho_solve
|
|
cholesky
|
|
det
|
|
eigh
|
|
eigh_tridiagonal
|
|
expm
|
|
expm_frechet
|
|
funm
|
|
hessenberg
|
|
inv
|
|
lu
|
|
lu_factor
|
|
lu_solve
|
|
polar
|
|
qr
|
|
rsf2csf
|
|
schur
|
|
sqrtm
|
|
solve
|
|
solve_triangular
|
|
sqrtm
|
|
svd
|
|
toeplitz
|
|
|
|
jax.scipy.ndimage
|
|
-----------------
|
|
|
|
.. automodule:: jax.scipy.ndimage
|
|
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
map_coordinates
|
|
|
|
jax.scipy.optimize
|
|
------------------
|
|
|
|
.. automodule:: jax.scipy.optimize
|
|
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
minimize
|
|
OptimizeResults
|
|
|
|
jax.scipy.signal
|
|
----------------
|
|
|
|
.. automodule:: jax.scipy.signal
|
|
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
fftconvolve
|
|
convolve
|
|
convolve2d
|
|
correlate
|
|
correlate2d
|
|
csd
|
|
istft
|
|
stft
|
|
welch
|
|
|
|
jax.scipy.spatial.transform
|
|
---------------------------
|
|
|
|
.. automodule:: jax.scipy.spatial.transform
|
|
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
Rotation
|
|
Slerp
|
|
|
|
jax.scipy.sparse.linalg
|
|
-----------------------
|
|
|
|
.. automodule:: jax.scipy.sparse.linalg
|
|
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
bicgstab
|
|
cg
|
|
gmres
|
|
|
|
jax.scipy.special
|
|
-----------------
|
|
|
|
.. automodule:: jax.scipy.special
|
|
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
bernoulli
|
|
beta
|
|
betainc
|
|
betaln
|
|
digamma
|
|
entr
|
|
erf
|
|
erfc
|
|
erfinv
|
|
exp1
|
|
expi
|
|
expit
|
|
expn
|
|
factorial
|
|
gamma
|
|
gammainc
|
|
gammaincc
|
|
gammaln
|
|
hyp1f1
|
|
i0
|
|
i0e
|
|
i1
|
|
i1e
|
|
log_ndtr
|
|
logit
|
|
logsumexp
|
|
lpmn
|
|
lpmn_values
|
|
multigammaln
|
|
ndtr
|
|
ndtri
|
|
poch
|
|
polygamma
|
|
spence
|
|
sph_harm
|
|
xlog1py
|
|
xlogy
|
|
zeta
|
|
kl_div
|
|
rel_entr
|
|
|
|
|
|
jax.scipy.stats
|
|
---------------
|
|
|
|
.. automodule:: jax.scipy.stats
|
|
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
mode
|
|
rankdata
|
|
sem
|
|
|
|
jax.scipy.stats.bernoulli
|
|
~~~~~~~~~~~~~~~~~~~~~~~~~
|
|
.. automodule:: jax.scipy.stats.bernoulli
|
|
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
logpmf
|
|
pmf
|
|
cdf
|
|
ppf
|
|
|
|
jax.scipy.stats.beta
|
|
~~~~~~~~~~~~~~~~~~~~
|
|
.. automodule:: jax.scipy.stats.beta
|
|
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
logpdf
|
|
pdf
|
|
cdf
|
|
logcdf
|
|
sf
|
|
logsf
|
|
|
|
jax.scipy.stats.betabinom
|
|
~~~~~~~~~~~~~~~~~~~~~~~~~
|
|
.. automodule:: jax.scipy.stats.betabinom
|
|
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
logpmf
|
|
pmf
|
|
|
|
jax.scipy.stats.binom
|
|
~~~~~~~~~~~~~~~~~~~~~
|
|
.. automodule:: jax.scipy.stats.binom
|
|
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
logpmf
|
|
pmf
|
|
|
|
jax.scipy.stats.cauchy
|
|
~~~~~~~~~~~~~~~~~~~~~~
|
|
.. automodule:: jax.scipy.stats.cauchy
|
|
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
logpdf
|
|
pdf
|
|
cdf
|
|
logcdf
|
|
sf
|
|
logsf
|
|
isf
|
|
ppf
|
|
|
|
jax.scipy.stats.chi2
|
|
~~~~~~~~~~~~~~~~~~~~
|
|
.. automodule:: jax.scipy.stats.chi2
|
|
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
logpdf
|
|
pdf
|
|
cdf
|
|
logcdf
|
|
sf
|
|
logsf
|
|
|
|
|
|
jax.scipy.stats.dirichlet
|
|
~~~~~~~~~~~~~~~~~~~~~~~~~
|
|
.. automodule:: jax.scipy.stats.dirichlet
|
|
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
logpdf
|
|
pdf
|
|
|
|
jax.scipy.stats.expon
|
|
~~~~~~~~~~~~~~~~~~~~~
|
|
.. automodule:: jax.scipy.stats.expon
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
logpdf
|
|
pdf
|
|
|
|
jax.scipy.stats.gamma
|
|
~~~~~~~~~~~~~~~~~~~~~
|
|
.. automodule:: jax.scipy.stats.gamma
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
logpdf
|
|
pdf
|
|
cdf
|
|
logcdf
|
|
sf
|
|
logsf
|
|
|
|
jax.scipy.stats.gennorm
|
|
~~~~~~~~~~~~~~~~~~~~~~~
|
|
.. automodule:: jax.scipy.stats.gennorm
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
cdf
|
|
logpdf
|
|
pdf
|
|
|
|
jax.scipy.stats.geom
|
|
~~~~~~~~~~~~~~~~~~~~
|
|
.. automodule:: jax.scipy.stats.geom
|
|
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
logpmf
|
|
pmf
|
|
|
|
jax.scipy.stats.laplace
|
|
~~~~~~~~~~~~~~~~~~~~~~~
|
|
.. automodule:: jax.scipy.stats.laplace
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
cdf
|
|
logpdf
|
|
pdf
|
|
|
|
jax.scipy.stats.logistic
|
|
~~~~~~~~~~~~~~~~~~~~~~~~
|
|
.. automodule:: jax.scipy.stats.logistic
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
cdf
|
|
isf
|
|
logpdf
|
|
pdf
|
|
ppf
|
|
sf
|
|
|
|
jax.scipy.stats.multinomial
|
|
~~~~~~~~~~~~~~~~~~~~~~~~~~~
|
|
.. automodule:: jax.scipy.stats.multinomial
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
logpmf
|
|
pmf
|
|
|
|
jax.scipy.stats.multivariate_normal
|
|
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
|
|
.. automodule:: jax.scipy.stats.multivariate_normal
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
logpdf
|
|
pdf
|
|
|
|
jax.scipy.stats.nbinom
|
|
~~~~~~~~~~~~~~~~~~~~~~
|
|
.. automodule:: jax.scipy.stats.nbinom
|
|
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
logpmf
|
|
pmf
|
|
|
|
jax.scipy.stats.norm
|
|
~~~~~~~~~~~~~~~~~~~~
|
|
.. automodule:: jax.scipy.stats.norm
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
logpdf
|
|
pdf
|
|
cdf
|
|
logcdf
|
|
ppf
|
|
sf
|
|
logsf
|
|
isf
|
|
|
|
jax.scipy.stats.pareto
|
|
~~~~~~~~~~~~~~~~~~~~~~
|
|
.. automodule:: jax.scipy.stats.pareto
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
logpdf
|
|
pdf
|
|
|
|
jax.scipy.stats.poisson
|
|
~~~~~~~~~~~~~~~~~~~~~~~
|
|
.. automodule:: jax.scipy.stats.poisson
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
logpmf
|
|
pmf
|
|
|
|
jax.scipy.stats.t
|
|
~~~~~~~~~~~~~~~~~
|
|
.. automodule:: jax.scipy.stats.t
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
logpdf
|
|
pdf
|
|
|
|
jax.scipy.stats.truncnorm
|
|
~~~~~~~~~~~~~~~~~~~~~~~~~
|
|
.. automodule:: jax.scipy.stats.truncnorm
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
cdf
|
|
logcdf
|
|
logpdf
|
|
logsf
|
|
pdf
|
|
sf
|
|
|
|
jax.scipy.stats.uniform
|
|
~~~~~~~~~~~~~~~~~~~~~~~
|
|
.. automodule:: jax.scipy.stats.uniform
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
logpdf
|
|
pdf
|
|
cdf
|
|
ppf
|
|
|
|
jax.scipy.stats.gaussian_kde
|
|
~~~~~~~~~~~~~~~~~~~~~~~~~~~~
|
|
.. currentmodule:: jax.scipy.stats
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
gaussian_kde
|
|
gaussian_kde.evaluate
|
|
gaussian_kde.integrate_gaussian
|
|
gaussian_kde.integrate_box_1d
|
|
gaussian_kde.integrate_kde
|
|
gaussian_kde.resample
|
|
gaussian_kde.pdf
|
|
gaussian_kde.logpdf
|
|
|
|
jax.scipy.stats.vonmises
|
|
~~~~~~~~~~~~~~~~~~~~~~~~
|
|
.. automodule:: jax.scipy.stats.vonmises
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
logpdf
|
|
pdf
|
|
|
|
jax.scipy.stats.wrapcauchy
|
|
~~~~~~~~~~~~~~~~~~~~~~~~~~
|
|
.. automodule:: jax.scipy.stats.wrapcauchy
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
logpdf
|
|
pdf
|