docs/jax.scipy.rst
jax.scipy module.. automodule:: jax.scipy.cluster.vq
.. autosummary:: :toctree: _autosummary
vq
.. automodule:: jax.scipy.fft
.. autosummary:: :toctree: _autosummary
dct dctn idct idctn
.. automodule:: jax.scipy.integrate
.. autosummary:: :toctree: _autosummary
trapezoid
.. automodule:: jax.scipy.interpolate
.. autosummary:: :toctree: _autosummary
RegularGridInterpolator
.. automodule:: jax.scipy.linalg
.. autosummary:: :toctree: _autosummary
block_diag cho_factor cho_solve cholesky det eigh eigh_tridiagonal expm expm_frechet funm hessenberg hankel hilbert inv lu lu_factor lu_solve pascal polar qr rsf2csf schur solve solve_sylvester solve_triangular sqrtm svd toeplitz
.. automodule:: jax.scipy.ndimage
.. autosummary:: :toctree: _autosummary
map_coordinates
.. automodule:: jax.scipy.optimize
.. autosummary:: :toctree: _autosummary
minimize OptimizeResults
.. automodule:: jax.scipy.signal
.. autosummary:: :toctree: _autosummary
fftconvolve convolve convolve2d correlate correlate2d csd detrend istft stft welch
.. automodule:: jax.scipy.spatial.transform
.. autosummary:: :toctree: _autosummary
Rotation Slerp
.. automodule:: jax.scipy.sparse.linalg
.. autosummary:: :toctree: _autosummary
bicgstab cg gmres
.. automodule:: jax.scipy.special
.. autosummary:: :toctree: _autosummary
bernoulli beta betainc betaln digamma entr erf erfc erfinv exp1 expi expit expn factorial fresnel gamma gammainc gammaincc gammaln gammasgn hyp1f1 hyp2f1 i0 i0e i1 i1e kl_div log_ndtr log_softmax logit logsumexp lpmn lpmn_values multigammaln ndtr ndtri poch polygamma rel_entr sici softmax spence sph_harm xlog1py xlogy zeta
.. 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
logcdf
cdf
logsf
sf
ppf
jax.scipy.stats.gamma
.. automodule:: jax.scipy.stats.gamma .. autosummary:: :toctree: _autosummary
logpdf pdf cdf logcdf sf logsf
jax.scipy.stats.gumbel_l
.. automodule:: jax.scipy.stats.gumbel_l
.. autosummary::
:toctree: _autosummary
logpdf
pdf
cdf
logcdf
sf
logsf
ppf
jax.scipy.stats.gumbel_r
.. automodule:: jax.scipy.stats.gumbel_r .. autosummary:: :toctree: _autosummary
logpdf pdf cdf logcdf sf logsf ppf
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
logcdf
logpdf
logsf
cdf
pdf
ppf
sf
jax.scipy.stats.poisson
~~~~~~~~~~~~~~~~~~~~~~~
.. automodule:: jax.scipy.stats.poisson
.. autosummary::
:toctree: _autosummary
logpmf
pmf
cdf
entropy
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