mirror of
https://github.com/ROCm/jax.git
synced 2025-04-24 14:36:07 +00:00
167 lines
2.1 KiB
ReStructuredText
167 lines
2.1 KiB
ReStructuredText
jax.scipy package
|
|
=================
|
|
|
|
jax.scipy.linalg
|
|
----------------
|
|
|
|
.. automodule:: jax.scipy.linalg
|
|
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
block_diag
|
|
cho_factor
|
|
cho_solve
|
|
cholesky
|
|
det
|
|
eigh
|
|
expm
|
|
inv
|
|
lu
|
|
lu_factor
|
|
lu_solve
|
|
qr
|
|
solve
|
|
solve_triangular
|
|
svd
|
|
tril
|
|
triu
|
|
|
|
jax.scipy.ndimage
|
|
-----------------
|
|
|
|
.. automodule:: jax.scipy.ndimage
|
|
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
map_coordinates
|
|
|
|
jax.scipy.signal
|
|
----------------
|
|
|
|
.. automodule:: jax.scipy.signal
|
|
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
convolve
|
|
convolve2d
|
|
correlate
|
|
correlate2d
|
|
|
|
jax.scipy.sparse.linalg
|
|
-----------------------
|
|
|
|
.. automodule:: jax.scipy.sparse.linalg
|
|
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
cg
|
|
|
|
jax.scipy.special
|
|
-----------------
|
|
|
|
.. automodule:: jax.scipy.special
|
|
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
betainc
|
|
digamma
|
|
entr
|
|
erf
|
|
erfc
|
|
erfinv
|
|
expit
|
|
gammainc
|
|
gammaincc
|
|
gammaln
|
|
i0e
|
|
i1e
|
|
log_ndtr
|
|
logit
|
|
logsumexp
|
|
multigammaln
|
|
ndtr
|
|
ndtri
|
|
xlog1py
|
|
xlogy
|
|
|
|
|
|
jax.scipy.stats
|
|
---------------
|
|
|
|
jax.scipy.stats.beta
|
|
~~~~~~~~~~~~~~~~~~~~
|
|
.. automodule:: jax.scipy.stats.beta
|
|
|
|
.. 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
|
|
|
|
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.norm
|
|
~~~~~~~~~~~~~~~~~~~~
|
|
.. automodule:: jax.scipy.stats.norm
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
cdf
|
|
logcdf
|
|
logpdf
|
|
pdf
|
|
ppf
|
|
|
|
jax.scipy.stats.uniform
|
|
~~~~~~~~~~~~~~~~~~~~~~~
|
|
.. automodule:: jax.scipy.stats.uniform
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
logpdf
|
|
pdf
|