2019-01-15 20:14:19 -05:00
|
|
|
jax.scipy package
|
|
|
|
=================
|
|
|
|
|
|
|
|
jax.scipy.linalg
|
|
|
|
-----------------------
|
|
|
|
|
|
|
|
.. automodule:: jax.scipy.linalg
|
|
|
|
|
2019-04-15 12:40:18 -04:00
|
|
|
.. autosummary::
|
|
|
|
:toctree: _autosummary
|
|
|
|
|
|
|
|
cho_factor
|
|
|
|
cho_solve
|
|
|
|
cholesky
|
|
|
|
det
|
|
|
|
eigh
|
|
|
|
inv
|
|
|
|
lu_factor
|
|
|
|
lu
|
|
|
|
qr
|
|
|
|
solve
|
|
|
|
solve_triangular
|
|
|
|
svd
|
|
|
|
tril
|
|
|
|
triu
|
2019-01-15 20:14:19 -05:00
|
|
|
|
|
|
|
|
|
|
|
jax.scipy.special
|
|
|
|
------------------------
|
|
|
|
|
|
|
|
.. automodule:: jax.scipy.special
|
2019-04-15 12:40:18 -04:00
|
|
|
|
|
|
|
.. autosummary::
|
|
|
|
:toctree: _autosummary
|
|
|
|
|
|
|
|
digamma
|
|
|
|
erf
|
|
|
|
erfc
|
|
|
|
erfinv
|
|
|
|
expit
|
|
|
|
gammaln
|
2019-05-15 09:54:24 -04:00
|
|
|
log_ndtr
|
2019-04-15 12:40:18 -04:00
|
|
|
logit
|
|
|
|
logsumexp
|
|
|
|
ndtr
|
|
|
|
ndtri
|
2019-05-15 09:54:24 -04:00
|
|
|
xlog1py
|
|
|
|
xlogy
|
2019-04-15 12:40:18 -04:00
|
|
|
|
2019-01-15 20:14:19 -05:00
|
|
|
|
|
|
|
jax.scipy.stats
|
|
|
|
-----------------------
|
|
|
|
|
|
|
|
jax.scipy.stats.beta
|
|
|
|
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
|
|
|
|
.. automodule:: jax.scipy.stats.beta
|
2019-04-15 12:40:18 -04:00
|
|
|
|
|
|
|
.. autosummary::
|
|
|
|
:toctree: _autosummary
|
|
|
|
|
|
|
|
logpdf
|
|
|
|
pdf
|
2019-01-15 20:14:19 -05:00
|
|
|
|
|
|
|
jax.scipy.stats.expon
|
|
|
|
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
|
|
|
|
.. automodule:: jax.scipy.stats.expon
|
2019-04-15 12:40:18 -04:00
|
|
|
.. autosummary::
|
|
|
|
:toctree: _autosummary
|
|
|
|
|
|
|
|
logpdf
|
|
|
|
pdf
|
2019-01-15 20:14:19 -05:00
|
|
|
|
|
|
|
jax.scipy.stats.gamma
|
|
|
|
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
|
|
|
|
.. automodule:: jax.scipy.stats.gamma
|
2019-04-15 12:40:18 -04:00
|
|
|
.. autosummary::
|
|
|
|
:toctree: _autosummary
|
|
|
|
|
|
|
|
logpdf
|
|
|
|
pdf
|
2019-01-15 20:14:19 -05:00
|
|
|
|
|
|
|
jax.scipy.stats.laplace
|
|
|
|
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
|
|
|
|
.. automodule:: jax.scipy.stats.laplace
|
2019-04-15 12:40:18 -04:00
|
|
|
.. autosummary::
|
|
|
|
:toctree: _autosummary
|
|
|
|
|
|
|
|
logpdf
|
|
|
|
pdf
|
2019-01-15 20:14:19 -05:00
|
|
|
|
|
|
|
jax.scipy.stats.norm
|
|
|
|
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
|
|
|
|
.. automodule:: jax.scipy.stats.norm
|
2019-04-15 12:40:18 -04:00
|
|
|
.. autosummary::
|
|
|
|
:toctree: _autosummary
|
|
|
|
|
|
|
|
cdf
|
|
|
|
logcdf
|
|
|
|
logpdf
|
|
|
|
pdf
|
2019-01-15 20:14:19 -05:00
|
|
|
|
|
|
|
jax.scipy.stats.uniform
|
|
|
|
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
|
|
|
|
.. automodule:: jax.scipy.stats.uniform
|
2019-04-15 12:40:18 -04:00
|
|
|
.. autosummary::
|
|
|
|
:toctree: _autosummary
|
|
|
|
|
|
|
|
logpdf
|
|
|
|
pdf
|