mirror of
https://github.com/ROCm/jax.git
synced 2025-04-16 20:06:05 +00:00
125 lines
1.6 KiB
ReStructuredText
125 lines
1.6 KiB
ReStructuredText
jax.scipy package
|
|
=================
|
|
|
|
jax.scipy.linalg
|
|
----------------
|
|
|
|
.. automodule:: jax.scipy.linalg
|
|
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
cho_factor
|
|
cho_solve
|
|
cholesky
|
|
det
|
|
eigh
|
|
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.special
|
|
-----------------
|
|
|
|
.. automodule:: jax.scipy.special
|
|
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
digamma
|
|
entr
|
|
erf
|
|
erfc
|
|
erfinv
|
|
expit
|
|
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.norm
|
|
~~~~~~~~~~~~~~~~~~~~
|
|
.. automodule:: jax.scipy.stats.norm
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
cdf
|
|
logcdf
|
|
logpdf
|
|
pdf
|
|
|
|
jax.scipy.stats.uniform
|
|
~~~~~~~~~~~~~~~~~~~~~~~
|
|
.. automodule:: jax.scipy.stats.uniform
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
logpdf
|
|
pdf
|