rocm_jax/docs/jax.scipy.rst
2019-07-02 15:00:47 -04:00

111 lines
1.5 KiB
ReStructuredText

jax.scipy package
=================
jax.scipy.linalg
-----------------------
.. automodule:: jax.scipy.linalg
.. autosummary::
:toctree: _autosummary
cho_factor
cho_solve
cholesky
det
eigh
inv
lu_factor
lu
qr
solve
solve_triangular
svd
tril
triu
jax.scipy.special
------------------------
.. automodule:: jax.scipy.special
.. autosummary::
:toctree: _autosummary
digamma
erf
erfc
erfinv
expit
gammaln
log_ndtr
logit
logsumexp
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