jax.scipy.rst

jax.scipy package

jax.scipy.linalg

.. automodule:: jax.scipy.linalg

.. autosummary::
  :toctree: _autosummary

   block_diag
   cho_factor
   cho_solve
   cholesky
   det
   eigh
   expm
   inv
   lu
   lu_factor
   lu_solve
   qr
   solve
   solve_triangular
   svd
   tril
   triu

jax.scipy.ndimage

.. automodule:: jax.scipy.ndimage

.. autosummary::
  :toctree: _autosummary

   map_coordinates

jax.scipy.signal

.. automodule:: jax.scipy.signal

.. autosummary::
  :toctree: _autosummary

   convolve
   convolve2d
   correlate
   correlate2d

jax.scipy.sparse.linalg

.. automodule:: jax.scipy.sparse.linalg

.. autosummary::
  :toctree: _autosummary

   cg

jax.scipy.special

.. automodule:: jax.scipy.special

.. autosummary::
  :toctree: _autosummary

   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

jax.scipy.stats.beta

.. automodule:: jax.scipy.stats.beta

.. 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

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.norm

.. automodule:: jax.scipy.stats.norm

.. autosummary::
  :toctree: _autosummary

   cdf
   logcdf
   logpdf
   pdf
   ppf

jax.scipy.stats.uniform

.. automodule:: jax.scipy.stats.uniform

.. autosummary::
  :toctree: _autosummary

   logpdf
   pdf