jax.scipy
module#
jax.scipy.fft#
|
Computes the discrete cosine transform of the input |
|
Computes the multidimensional discrete cosine transform of the input |
|
Computes the inverse discrete cosine transform of the input |
|
Computes the multidimensional inverse discrete cosine transform of the input |
jax.scipy.integrate#
|
Integrate along the given axis using the composite trapezoidal rule. |
jax.scipy.linalg#
|
Create a block diagonal matrix from provided arrays. |
|
Compute the Cholesky decomposition of a matrix, to use in cho_solve |
|
Solve the linear equations A x = b, given the Cholesky factorization of A. |
|
Compute the Cholesky decomposition of a matrix. |
|
Compute the determinant of a matrix |
|
Solve a standard or generalized eigenvalue problem for a complex |
|
Solve eigenvalue problem for a real symmetric tridiagonal matrix. |
|
Compute the matrix exponential of an array. |
Frechet derivative of the matrix exponential of A in the direction E. |
|
|
Evaluate a matrix function specified by a callable. |
Compute Hessenberg form of a matrix. |
|
|
Create a Hilbert matrix of order n. |
|
Compute the inverse of a matrix. |
|
Compute LU decomposition of a matrix with partial pivoting. |
|
Compute pivoted LU decomposition of a matrix. |
|
Solve an equation system, a x = b, given the LU factorization of a |
|
Computes the polar decomposition. |
|
Compute QR decomposition of a matrix. |
|
Convert real Schur form to complex Schur form. |
|
Compute Schur decomposition of a matrix. |
|
Solves the linear equation set |
|
Solve the equation a x = b for x, assuming a is a triangular matrix. |
|
Matrix square root. |
|
Singular Value Decomposition. |
|
Construct a Toeplitz matrix. |
jax.scipy.ndimage#
|
Map the input array to new coordinates by interpolation. |
jax.scipy.optimize#
|
Minimization of scalar function of one or more variables. |
|
Object holding optimization results. |
jax.scipy.signal#
|
Convolve two N-dimensional arrays using FFT. |
|
Convolve two N-dimensional arrays. |
|
Convolve two 2-dimensional arrays. |
|
Cross-correlate two N-dimensional arrays. |
|
Cross-correlate two 2-dimensional arrays. |
|
Estimate the cross power spectral density, Pxy, using Welch's method. |
|
Perform the inverse Short Time Fourier transform (legacy function). |
|
Compute the Short Time Fourier Transform (legacy function). |
|
Estimate power spectral density using Welch's method. |
jax.scipy.spatial.transform#
|
Rotation in 3 dimensions. |
|
Spherical Linear Interpolation of Rotations. |
jax.scipy.sparse.linalg#
|
Use Bi-Conjugate Gradient Stable iteration to solve |
|
Use Conjugate Gradient iteration to solve |
|
GMRES solves the linear system A x = b for x, given A and b. |
jax.scipy.special#
|
Generate the first N Bernoulli numbers. |
|
The beta function |
|
The regularized incomplete beta function. |
|
Natural log of the absolute value of the beta function |
|
The digamma function |
|
The entropy function |
|
The error function |
|
The complement of the error function |
|
The inverse of the error function |
|
Exponential integral function. |
Exponential integral function. |
|
|
The logistic sigmoid (expit) function |
Generalized exponential integral function. |
|
|
Factorial function |
|
The gamma function. |
|
The regularized lower incomplete gamma function. |
|
The regularized upper incomplete gamma function. |
|
Natural log of the absolute value of the gamma function. |
|
Sign of the gamma function. |
The 1F1 hypergeometric function. |
|
|
Modified bessel function of zeroth order. |
|
Exponentially scaled modified bessel function of zeroth order. |
|
Modified bessel function of first order. |
|
Exponentially scaled modified bessel function of first order. |
Log Normal distribution function. |
|
The logit function |
|
Log-sum-exp reduction. |
|
|
The associated Legendre functions (ALFs) of the first kind. |
|
The associated Legendre functions (ALFs) of the first kind. |
|
The natural log of the multivariate gamma function. |
|
Normal distribution function. |
|
The inverse of the CDF of the Normal distribution function. |
The Pochammer symbol. |
|
|
The polygamma function. |
|
Spence's function, also known as the dilogarithm for real values. |
|
Computes the spherical harmonics. |
Compute x*log(1 + y), returning 0 for x=0. |
|
Compute x*log(y), returning 0 for x=0. |
|
The Hurwitz zeta function. |
|
|
The Kullback-Leibler divergence. |
|
The relative entropy function. |
jax.scipy.stats#
|
LAX-backend implementation of |
|
Assign ranks to data, dealing with ties appropriately. |
|
LAX-backend implementation of |
jax.scipy.stats.bernoulli#
|
Log of the probability mass function at k of the given RV. |
|
Probability mass function at k of the given RV. |
|
Cumulative distribution function of the given RV. |
|
Percent point function (inverse of cdf) at q of the given RV. |
jax.scipy.stats.beta#
|
Log of the probability density function at x of the given RV. |
|
Probability density function at x of the given RV. |
|
Cumulative distribution function of the given RV. |
|
Log of the cumulative distribution function at x of the given RV. |
|
Survival function (1 - cdf) at x of the given RV. |
|
Log of the survival function of the given RV. |
jax.scipy.stats.betabinom#
|
Log of the probability mass function at k of the given RV. |
|
Probability mass function at k of the given RV. |
jax.scipy.stats.binom#
|
Log of the probability mass function at k of the given RV. |
|
Probability mass function at k of the given RV. |
jax.scipy.stats.cauchy#
|
Log of the probability density function at x of the given RV. |
|
Probability density function at x of the given RV. |
|
Cumulative distribution function of the given RV. |
|
Log of the cumulative distribution function at x of the given RV. |
|
Survival function (1 - cdf) at x of the given RV. |
|
Log of the survival function of the given RV. |
|
Inverse survival function (inverse of sf) at q of the given RV. |
|
Percent point function (inverse of cdf) at q of the given RV. |
jax.scipy.stats.chi2#
|
Log of the probability density function at x of the given RV. |
|
Probability density function at x of the given RV. |
|
Cumulative distribution function of the given RV. |
|
Log of the cumulative distribution function at x of the given RV. |
|
Survival function (1 - cdf) at x of the given RV. |
|
Log of the survival function of the given RV. |
jax.scipy.stats.dirichlet#
|
Log of the Dirichlet probability density function. |
|
The Dirichlet probability density function. |
jax.scipy.stats.expon#
|
Log of the probability density function at x of the given RV. |
|
Probability density function at x of the given RV. |
jax.scipy.stats.gamma#
|
Log of the probability density function at x of the given RV. |
|
Probability density function at x of the given RV. |
|
Cumulative distribution function of the given RV. |
|
Log of the cumulative distribution function at x of the given RV. |
|
Survival function (1 - cdf) at x of the given RV. |
|
Log of the survival function of the given RV. |
jax.scipy.stats.gennorm#
|
Cumulative distribution function of the given RV. |
|
Log of the probability density function at x of the given RV. |
|
Probability density function at x of the given RV. |
jax.scipy.stats.geom#
|
Log of the probability mass function at k of the given RV. |
|
Probability mass function at k of the given RV. |
jax.scipy.stats.laplace#
|
Cumulative distribution function of the given RV. |
|
Log of the probability density function at x of the given RV. |
|
Probability density function at x of the given RV. |
jax.scipy.stats.logistic#
|
Cumulative distribution function of the given RV. |
|
Inverse survival function (inverse of sf) at q of the given RV. |
|
Log of the probability density function at x of the given RV. |
|
Probability density function at x of the given RV. |
|
Percent point function (inverse of cdf) at q of the given RV. |
|
Survival function (1 - cdf) at x of the given RV. |
jax.scipy.stats.multinomial#
|
Log of the Multinomial probability mass function. |
|
Multinomial probability mass function. |
jax.scipy.stats.multivariate_normal#
|
Log of the multivariate normal probability density function. |
|
Multivariate normal probability density function. |
jax.scipy.stats.nbinom#
|
Log of the probability mass function at k of the given RV. |
|
Probability mass function at k of the given RV. |
jax.scipy.stats.norm#
|
Log of the probability density function at x of the given RV. |
|
Probability density function at x of the given RV. |
|
Cumulative distribution function of the given RV. |
|
Log of the cumulative distribution function at x of the given RV. |
|
Percent point function (inverse of cdf) at q of the given RV. |
|
Survival function (1 - cdf) at x of the given RV. |
|
Log of the survival function of the given RV. |
|
Inverse survival function (inverse of sf) at q of the given RV. |
jax.scipy.stats.pareto#
|
Log of the probability density function at x of the given RV. |
|
Probability density function at x of the given RV. |
jax.scipy.stats.poisson#
|
Log of the probability mass function at k of the given RV. |
|
Probability mass function at k of the given RV. |
jax.scipy.stats.t#
|
Log of the probability density function at x of the given RV. |
|
Probability density function at x of the given RV. |
jax.scipy.stats.truncnorm#
|
Cumulative distribution function of the given RV. |
|
Log of the cumulative distribution function at x of the given RV. |
|
Log of the probability density function at x of the given RV. |
|
Log of the survival function of the given RV. |
|
Probability density function at x of the given RV. |
|
Survival function (1 - cdf) at x of the given RV. |
jax.scipy.stats.uniform#
|
Log of the probability density function at x of the given RV. |
|
Probability density function at x of the given RV. |
|
Cumulative distribution function of the given RV. |
|
Percent point function (inverse of cdf) at q of the given RV. |
jax.scipy.stats.gaussian_kde#
|
Representation of a kernel-density estimate using Gaussian kernels. |
|
Evaluate the estimated pdf on a set of points. |
|
Multiply estimated density by a multivariate Gaussian and integrate |
|
Computes the integral of a 1D pdf between two bounds. |
|
Computes the integral of the product of this kernel density estimate |
|
Randomly sample a dataset from the estimated pdf |
Evaluate the estimated pdf on a provided set of points. |
|
Evaluate the log of the estimated pdf on a provided set of points. |
jax.scipy.stats.vonmises#
|
Log of the probability density function at x of the given RV. |
|
Probability density function at x of the given RV. |
jax.scipy.stats.wrapcauchy#
|
Log of the probability density function at x of the given RV. |
|
Probability density function at x of the given RV. |