jax.scipy.linalg
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
map_coordinates
jax.scipy.signal
convolve convolve2d correlate correlate2d
jax.scipy.sparse.linalg
cg
jax.scipy.special
betainc digamma entr erf erfc erfinv expit gammainc gammaincc gammaln i0e i1e log_ndtr logit logsumexp multigammaln ndtr ndtri polygamma xlog1py xlogy zeta
jax.scipy.stats.beta
logpdf pdf
jax.scipy.stats.expon
logpdf pdf
jax.scipy.stats.gamma
logpdf pdf
jax.scipy.stats.laplace
cdf logpdf pdf
jax.scipy.stats.logistic
cdf isf logpdf pdf ppf sf
jax.scipy.stats.norm
cdf logcdf logpdf pdf ppf
jax.scipy.stats.uniform
logpdf pdf