jax.scipy package
=================
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
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
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
gmres
jax.scipy.special
-----------------
.. automodule:: jax.scipy.special
.. autosummary::
:toctree: _autosummary
betainc
digamma
entr
erf
erfc
erfinv
expit
gammainc
gammaincc
gammaln
i0
i0e
i1
i1e
log_ndtr
logit
logsumexp
multigammaln
ndtr
ndtri
polygamma
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.cauchy
~~~~~~~~~~~~~~~~~~~~~~
.. automodule:: jax.scipy.stats.cauchy
.. 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