Revision 01194bd2fb1f7b6a46db88be14324f7acdb1723e authored by Sergei Lebedev on 16 May 2024, 18:27:56 UTC, committed by jax authors on 16 May 2024, 18:29:09 UTC
The callback APIs were migrated to use jax.Arrays for both inputs and outputs in JAX 0.4.27. PiperOrigin-RevId: 634473890
1 parent 380503b
jax.scipy.rst
``jax.scipy`` module
====================
jax.scipy.cluster
-----------------
.. automodule:: jax.scipy.cluster.vq
.. autosummary::
:toctree: _autosummary
vq
jax.scipy.fft
-------------
.. automodule:: jax.scipy.fft
.. autosummary::
:toctree: _autosummary
dct
dctn
idct
idctn
jax.scipy.integrate
-------------------
.. automodule:: jax.scipy.integrate
.. autosummary::
:toctree: _autosummary
trapezoid
jax.scipy.linalg
----------------
.. automodule:: jax.scipy.linalg
.. autosummary::
:toctree: _autosummary
block_diag
cho_factor
cho_solve
cholesky
det
eigh
eigh_tridiagonal
expm
expm_frechet
funm
hessenberg
hilbert
inv
lu
lu_factor
lu_solve
polar
qr
rsf2csf
schur
solve
solve_triangular
sqrtm
svd
toeplitz
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
fftconvolve
convolve
convolve2d
correlate
correlate2d
csd
detrend
istft
stft
welch
jax.scipy.spatial.transform
---------------------------
.. automodule:: jax.scipy.spatial.transform
.. autosummary::
:toctree: _autosummary
Rotation
Slerp
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
bernoulli
beta
betainc
betaln
digamma
entr
erf
erfc
erfinv
exp1
expi
expit
expn
factorial
gamma
gammainc
gammaincc
gammaln
gammasgn
hyp1f1
i0
i0e
i1
i1e
log_ndtr
logit
logsumexp
lpmn
lpmn_values
multigammaln
ndtr
ndtri
poch
polygamma
spence
sph_harm
xlog1py
xlogy
zeta
kl_div
rel_entr
jax.scipy.stats
---------------
.. automodule:: jax.scipy.stats
.. autosummary::
:toctree: _autosummary
mode
rankdata
sem
jax.scipy.stats.bernoulli
~~~~~~~~~~~~~~~~~~~~~~~~~
.. automodule:: jax.scipy.stats.bernoulli
.. autosummary::
:toctree: _autosummary
logpmf
pmf
cdf
ppf
jax.scipy.stats.beta
~~~~~~~~~~~~~~~~~~~~
.. automodule:: jax.scipy.stats.beta
.. autosummary::
:toctree: _autosummary
logpdf
pdf
cdf
logcdf
sf
logsf
jax.scipy.stats.betabinom
~~~~~~~~~~~~~~~~~~~~~~~~~
.. automodule:: jax.scipy.stats.betabinom
.. autosummary::
:toctree: _autosummary
logpmf
pmf
jax.scipy.stats.binom
~~~~~~~~~~~~~~~~~~~~~
.. automodule:: jax.scipy.stats.binom
.. autosummary::
:toctree: _autosummary
logpmf
pmf
jax.scipy.stats.cauchy
~~~~~~~~~~~~~~~~~~~~~~
.. automodule:: jax.scipy.stats.cauchy
.. autosummary::
:toctree: _autosummary
logpdf
pdf
cdf
logcdf
sf
logsf
isf
ppf
jax.scipy.stats.chi2
~~~~~~~~~~~~~~~~~~~~
.. automodule:: jax.scipy.stats.chi2
.. autosummary::
:toctree: _autosummary
logpdf
pdf
cdf
logcdf
sf
logsf
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
cdf
logcdf
sf
logsf
jax.scipy.stats.gennorm
~~~~~~~~~~~~~~~~~~~~~~~
.. automodule:: jax.scipy.stats.gennorm
.. autosummary::
:toctree: _autosummary
cdf
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.multinomial
~~~~~~~~~~~~~~~~~~~~~~~~~~~
.. automodule:: jax.scipy.stats.multinomial
.. autosummary::
:toctree: _autosummary
logpmf
pmf
jax.scipy.stats.multivariate_normal
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
.. automodule:: jax.scipy.stats.multivariate_normal
.. autosummary::
:toctree: _autosummary
logpdf
pdf
jax.scipy.stats.nbinom
~~~~~~~~~~~~~~~~~~~~~~
.. automodule:: jax.scipy.stats.nbinom
.. autosummary::
:toctree: _autosummary
logpmf
pmf
jax.scipy.stats.norm
~~~~~~~~~~~~~~~~~~~~
.. automodule:: jax.scipy.stats.norm
.. autosummary::
:toctree: _autosummary
logpdf
pdf
cdf
logcdf
ppf
sf
logsf
isf
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
cdf
jax.scipy.stats.t
~~~~~~~~~~~~~~~~~
.. automodule:: jax.scipy.stats.t
.. autosummary::
:toctree: _autosummary
logpdf
pdf
jax.scipy.stats.truncnorm
~~~~~~~~~~~~~~~~~~~~~~~~~
.. automodule:: jax.scipy.stats.truncnorm
.. autosummary::
:toctree: _autosummary
cdf
logcdf
logpdf
logsf
pdf
sf
jax.scipy.stats.uniform
~~~~~~~~~~~~~~~~~~~~~~~
.. automodule:: jax.scipy.stats.uniform
.. autosummary::
:toctree: _autosummary
logpdf
pdf
cdf
ppf
jax.scipy.stats.gaussian_kde
~~~~~~~~~~~~~~~~~~~~~~~~~~~~
.. currentmodule:: jax.scipy.stats
.. autosummary::
:toctree: _autosummary
gaussian_kde
gaussian_kde.evaluate
gaussian_kde.integrate_gaussian
gaussian_kde.integrate_box_1d
gaussian_kde.integrate_kde
gaussian_kde.resample
gaussian_kde.pdf
gaussian_kde.logpdf
jax.scipy.stats.vonmises
~~~~~~~~~~~~~~~~~~~~~~~~
.. automodule:: jax.scipy.stats.vonmises
.. autosummary::
:toctree: _autosummary
logpdf
pdf
jax.scipy.stats.wrapcauchy
~~~~~~~~~~~~~~~~~~~~~~~~~~
.. automodule:: jax.scipy.stats.wrapcauchy
.. autosummary::
:toctree: _autosummary
logpdf
pdf
![swh spinner](/static/img/swh-spinner.gif)
Computing file changes ...