mirror of
https://github.com/ROCm/jax.git
synced 2025-04-17 20:36:05 +00:00
3e5ecfe363
Reorder the doc modules into something closer to alphabetical order. Add missing functions from jax.scipy.linalg and jax.scipy.signal to the docs.
305 lines
4.0 KiB
ReStructuredText
305 lines
4.0 KiB
ReStructuredText
jax.scipy package
|
|
=================
|
|
|
|
jax.scipy.fft
|
|
-------------
|
|
|
|
.. automodule:: jax.scipy.fft
|
|
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
dct
|
|
dctn
|
|
|
|
jax.scipy.linalg
|
|
----------------
|
|
|
|
.. automodule:: jax.scipy.linalg
|
|
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
block_diag
|
|
cho_factor
|
|
cho_solve
|
|
cholesky
|
|
det
|
|
eigh
|
|
expm
|
|
expm_frechet
|
|
inv
|
|
lu
|
|
lu_factor
|
|
lu_solve
|
|
qr
|
|
schur
|
|
sqrtm
|
|
solve
|
|
solve_triangular
|
|
svd
|
|
tril
|
|
triu
|
|
|
|
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
|
|
|
|
convolve
|
|
convolve2d
|
|
correlate
|
|
correlate2d
|
|
csd
|
|
stft
|
|
welch
|
|
|
|
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
|
|
|
|
betainc
|
|
digamma
|
|
entr
|
|
erf
|
|
erfc
|
|
erfinv
|
|
exp1
|
|
expi
|
|
expit
|
|
expn
|
|
gammainc
|
|
gammaincc
|
|
gammaln
|
|
i0
|
|
i0e
|
|
i1
|
|
i1e
|
|
log_ndtr
|
|
logit
|
|
logsumexp
|
|
lpmn
|
|
lpmn_values
|
|
multigammaln
|
|
ndtr
|
|
ndtri
|
|
polygamma
|
|
sph_harm
|
|
xlog1py
|
|
xlogy
|
|
zeta
|
|
|
|
|
|
jax.scipy.stats
|
|
---------------
|
|
|
|
jax.scipy.stats.bernoulli
|
|
~~~~~~~~~~~~~~~~~~~~~~~~~
|
|
.. automodule:: jax.scipy.stats.bernoulli
|
|
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
logpmf
|
|
pmf
|
|
|
|
jax.scipy.stats.beta
|
|
~~~~~~~~~~~~~~~~~~~~
|
|
.. automodule:: jax.scipy.stats.beta
|
|
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
logpdf
|
|
pdf
|
|
|
|
jax.scipy.stats.betabinom
|
|
~~~~~~~~~~~~~~~~~~~~~~~~~
|
|
.. automodule:: jax.scipy.stats.betabinom
|
|
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
logpmf
|
|
pmf
|
|
|
|
jax.scipy.stats.cauchy
|
|
~~~~~~~~~~~~~~~~~~~~~~
|
|
.. automodule:: jax.scipy.stats.cauchy
|
|
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
logpdf
|
|
pdf
|
|
|
|
jax.scipy.stats.chi2
|
|
~~~~~~~~~~~~~~~~~~~~
|
|
.. automodule:: jax.scipy.stats.chi2
|
|
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
logpdf
|
|
pdf
|
|
|
|
|
|
|
|
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
|
|
|
|
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.multivariate_normal
|
|
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
|
|
.. automodule:: jax.scipy.stats.multivariate_normal
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
logpdf
|
|
pdf
|
|
|
|
jax.scipy.stats.norm
|
|
~~~~~~~~~~~~~~~~~~~~
|
|
.. automodule:: jax.scipy.stats.norm
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
cdf
|
|
logcdf
|
|
logpdf
|
|
pdf
|
|
ppf
|
|
|
|
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.uniform
|
|
~~~~~~~~~~~~~~~~~~~~~~~
|
|
.. automodule:: jax.scipy.stats.uniform
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
logpdf
|
|
pdf
|