rocm_jax/docs/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
   expm_frechet
   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
   i0
   i0e
   i1
   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
Add Sphinx-generated reference documentation for JAX. 2019-01-15 20:14:19 -05:00			`jax.scipy package`
			`=================`

			`jax.scipy.linalg`
Add jax.scipy.ndimage to online docs (#1724) 2019-11-20 12:35:10 -08:00			`----------------`
Add Sphinx-generated reference documentation for JAX. 2019-01-15 20:14:19 -05:00
			`.. automodule:: jax.scipy.linalg`

Improve layout of jax.scipy documentation. Fixes #577, since the set of supported distributions is now relatively easy to see. 2019-04-15 12:40:18 -04:00			`.. autosummary::`
			`:toctree: _autosummary`

Implement jax.scipy.linalg.block_diag. (#2113) 2020-01-29 11:24:40 -05:00			`block_diag`
Improve layout of jax.scipy documentation. Fixes #577, since the set of supported distributions is now relatively easy to see. 2019-04-15 12:40:18 -04:00			`cho_factor`
			`cho_solve`
			`cholesky`
			`det`
			`eigh`
Issue1635 expm (#1940) * Issue1635 expm Implemented expm using Pade approximation. The implmentation is wrapped using custom_transforms. Frechet derivatives are provided using defvjp. * Issue1635 expm Implemented expm using Pade approximation based on tf.linalg.expm. * Revert "Revert "Merge remote-tracking branch 'origin/Issue1635' into Issue1635"" This reverts commit dd26c6eeeb60fa556f55abc8acb2f5969b64a2f5, reversing changes made to b63c190c7671ebb9b911a52dcc203285c56a8051. * Issue1635 expm testing Add a test that compares numerical output of scipy.linalg.expm against jax.scipy.linalg.expm * travis build Issue1635 branch * Issue1635 expm testing Use rand_small to get numerical agreeming * Issue1635 expm testing Use @jit to prevent recompilation * Issue1635 expm testing Use rand_small to get numerical agreement * Revert "travis build Issue1635 branch" This reverts commit 6139772555e3af79dc0307fce88838a480e42d38. * Issue1635 Replace construct with jax.numpy.select * Issue1635 Restructure to support the docstring from SciPy * Issue1635 Restructure to support the docstring from SciPy * Issue1635 Remove the note that sparsity is not exploited because JAX does not support sparsity. * Issue1635 expm Support for the case where A is upper triangular. Instead of autodetection, the option is specified explicitly. * Issue1635 Rename argument, make it positional. Update documentation Co-authored-by: Jan <j.hueckelheim@imperial.ac.uk> 2020-01-22 00:11:51 -05:00			`expm`
Issue1635 expm frechet (#2062) * Implement Frechet derivatives for expm. * Update expm to use the current custom gradients API. Make some stylistic fixes. Co-authored-by: Peter Hawkins <phawkins@google.com> 2020-06-28 12:11:12 -04:00			`expm_frechet`
Improve layout of jax.scipy documentation. Fixes #577, since the set of supported distributions is now relatively easy to see. 2019-04-15 12:40:18 -04:00			`inv`
			`lu`
Implement jax.scipy.linalg.lu_solve. 2019-09-05 09:59:47 -04:00			`lu_factor`
			`lu_solve`
Improve layout of jax.scipy documentation. Fixes #577, since the set of supported distributions is now relatively easy to see. 2019-04-15 12:40:18 -04:00			`qr`
			`solve`
			`solve_triangular`
			`svd`
			`tril`
			`triu`
Add Sphinx-generated reference documentation for JAX. 2019-01-15 20:14:19 -05:00
Add jax.scipy.ndimage to online docs (#1724) 2019-11-20 12:35:10 -08:00			`jax.scipy.ndimage`
			`-----------------`

			`.. automodule:: jax.scipy.ndimage`

			`.. autosummary::`
			`:toctree: _autosummary`

			`map_coordinates`
Add Sphinx-generated reference documentation for JAX. 2019-01-15 20:14:19 -05:00
Add missing functions to autodoc 2020-04-13 13:46:04 -07:00			`jax.scipy.signal`
			`----------------`

			`.. automodule:: jax.scipy.signal`

			`.. autosummary::`
			`:toctree: _autosummary`

			`convolve`
			`convolve2d`
			`correlate`
			`correlate2d`

Implement scipy.sparse.linalg.cg (second try) (#2566) * super minimal starter code * Update optimizers.py * implement flip with axis = None * Create sparse.py * fix some imports * Update sparse.py * add partial function & test * Update lax_scipy_sparse_test.py * Update lax_scipy_sparse_test.py * add a test case for sparse pd matrix & add bigger dim * address comments * fix info return & create matrix with rng_factory * Update lax_scipy_sparse_test.py * Update lax_scipy_sparse_test.py * Update sparse.py * Update sparse.py * Update sparse.py * Update lax_scipy_sparse_test.py * Update lax_scipy_sparse_test.py * cast jax arrays into numpy array for scipy compatibility * Update sparse.py * Update sparse.py * fix None issue, but algo is not working * fix return of build_and_solve and output of while_loop * fix condition func of while loop * clearer variable names * mismatch error * Update lax_scipy_sparse_test.py * Fixes to jax.experimental.sparse.cg * Fix tests for gradients * Add support for preconditioners to cg * Move cg into scipy, update docs * doc tweak Co-authored-by: Tuan Nguyen <anhtuan277@gmail.com> 2020-04-03 13:37:11 -07:00			`jax.scipy.sparse.linalg`
			`-----------------------`

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

			`.. autosummary::`
			`:toctree: _autosummary`

			`cg`

Add Sphinx-generated reference documentation for JAX. 2019-01-15 20:14:19 -05:00			`jax.scipy.special`
Add jax.scipy.ndimage to online docs (#1724) 2019-11-20 12:35:10 -08:00			`-----------------`
Add Sphinx-generated reference documentation for JAX. 2019-01-15 20:14:19 -05:00
			`.. automodule:: jax.scipy.special`
Improve layout of jax.scipy documentation. Fixes #577, since the set of supported distributions is now relatively easy to see. 2019-04-15 12:40:18 -04:00
			`.. autosummary::`
			`:toctree: _autosummary`

Add betainc to JAX (#1998) Adds betaln, a wrapper for the Beta function (scipy.special.betaln). 2020-01-15 13:13:11 -08:00			`betainc`
Improve layout of jax.scipy documentation. Fixes #577, since the set of supported distributions is now relatively easy to see. 2019-04-15 12:40:18 -04:00			`digamma`
Fix test breakage at head. Add new numpy/scipy functions to documentation. 2019-08-02 08:55:22 -04:00			`entr`
Improve layout of jax.scipy documentation. Fixes #577, since the set of supported distributions is now relatively easy to see. 2019-04-15 12:40:18 -04:00			`erf`
			`erfc`
			`erfinv`
			`expit`
Add gammainc/gammaincc to JAX (#2064) 2020-01-29 08:25:21 -08:00			`gammainc`
			`gammaincc`
Improve layout of jax.scipy documentation. Fixes #577, since the set of supported distributions is now relatively easy to see. 2019-04-15 12:40:18 -04:00			`gammaln`
Add Bessel functions in jax.numpy & jax.scipy.special (#4007) 2020-08-10 10:10:59 -07:00			`i0`
Add i0e and i1e Bessel functions. (#1541) 2019-10-21 10:30:55 -04:00			`i0e`
Add Bessel functions in jax.numpy & jax.scipy.special (#4007) 2020-08-10 10:10:59 -07:00			`i1`
Add i0e and i1e Bessel functions. (#1541) 2019-10-21 10:30:55 -04:00			`i1e`
Add some missing special functions to the documentation. 2019-05-15 09:54:24 -04:00			`log_ndtr`
Improve layout of jax.scipy documentation. Fixes #577, since the set of supported distributions is now relatively easy to see. 2019-04-15 12:40:18 -04:00			`logit`
			`logsumexp`
Fix test breakage at head. Add new numpy/scipy functions to documentation. 2019-08-02 08:55:22 -04:00			`multigammaln`
Improve layout of jax.scipy documentation. Fixes #577, since the set of supported distributions is now relatively easy to see. 2019-04-15 12:40:18 -04:00			`ndtr`
			`ndtri`
add scipy.special.zeta, scipy.special.polygamma (#3385) 2020-06-15 22:21:26 -04:00			`polygamma`
Add some missing special functions to the documentation. 2019-05-15 09:54:24 -04:00			`xlog1py`
			`xlogy`
add scipy.special.zeta, scipy.special.polygamma (#3385) 2020-06-15 22:21:26 -04:00			`zeta`
Improve layout of jax.scipy documentation. Fixes #577, since the set of supported distributions is now relatively easy to see. 2019-04-15 12:40:18 -04:00
Add Sphinx-generated reference documentation for JAX. 2019-01-15 20:14:19 -05:00
			`jax.scipy.stats`
Add jax.scipy.ndimage to online docs (#1724) 2019-11-20 12:35:10 -08:00			`---------------`
Add Sphinx-generated reference documentation for JAX. 2019-01-15 20:14:19 -05:00
			`jax.scipy.stats.beta`
Add jax.scipy.ndimage to online docs (#1724) 2019-11-20 12:35:10 -08:00			`~~~~~~~~~~~~~~~~~~~~`
Add Sphinx-generated reference documentation for JAX. 2019-01-15 20:14:19 -05:00			`.. automodule:: jax.scipy.stats.beta`
Improve layout of jax.scipy documentation. Fixes #577, since the set of supported distributions is now relatively easy to see. 2019-04-15 12:40:18 -04:00
			`.. autosummary::`
			`:toctree: _autosummary`

			`logpdf`
			`pdf`
Add Sphinx-generated reference documentation for JAX. 2019-01-15 20:14:19 -05:00
			`jax.scipy.stats.expon`
Add jax.scipy.ndimage to online docs (#1724) 2019-11-20 12:35:10 -08:00			`~~~~~~~~~~~~~~~~~~~~~`
Add Sphinx-generated reference documentation for JAX. 2019-01-15 20:14:19 -05:00			`.. automodule:: jax.scipy.stats.expon`
Improve layout of jax.scipy documentation. Fixes #577, since the set of supported distributions is now relatively easy to see. 2019-04-15 12:40:18 -04:00			`.. autosummary::`
			`:toctree: _autosummary`

			`logpdf`
			`pdf`
Add Sphinx-generated reference documentation for JAX. 2019-01-15 20:14:19 -05:00
			`jax.scipy.stats.gamma`
Add jax.scipy.ndimage to online docs (#1724) 2019-11-20 12:35:10 -08:00			`~~~~~~~~~~~~~~~~~~~~~`
Add Sphinx-generated reference documentation for JAX. 2019-01-15 20:14:19 -05:00			`.. automodule:: jax.scipy.stats.gamma`
Improve layout of jax.scipy documentation. Fixes #577, since the set of supported distributions is now relatively easy to see. 2019-04-15 12:40:18 -04:00			`.. autosummary::`
			`:toctree: _autosummary`

			`logpdf`
			`pdf`
Add Sphinx-generated reference documentation for JAX. 2019-01-15 20:14:19 -05:00
			`jax.scipy.stats.laplace`
Add jax.scipy.ndimage to online docs (#1724) 2019-11-20 12:35:10 -08:00			`~~~~~~~~~~~~~~~~~~~~~~~`
Add Sphinx-generated reference documentation for JAX. 2019-01-15 20:14:19 -05:00			`.. automodule:: jax.scipy.stats.laplace`
Improve layout of jax.scipy documentation. Fixes #577, since the set of supported distributions is now relatively easy to see. 2019-04-15 12:40:18 -04:00			`.. autosummary::`
			`:toctree: _autosummary`

Minor doc fixes. 2019-07-02 15:00:47 -04:00			`cdf`
Improve layout of jax.scipy documentation. Fixes #577, since the set of supported distributions is now relatively easy to see. 2019-04-15 12:40:18 -04:00			`logpdf`
			`pdf`
Add Sphinx-generated reference documentation for JAX. 2019-01-15 20:14:19 -05:00
Add jax.scipy.stats.logistic to documentation. (#2149) 2020-02-03 12:44:57 -05:00			`jax.scipy.stats.logistic`
			`~~~~~~~~~~~~~~~~~~~~~~~~`
			`.. automodule:: jax.scipy.stats.logistic`
			`.. autosummary::`
			`:toctree: _autosummary`

			`cdf`
			`isf`
			`logpdf`
			`pdf`
			`ppf`
			`sf`

Add Sphinx-generated reference documentation for JAX. 2019-01-15 20:14:19 -05:00			`jax.scipy.stats.norm`
Add jax.scipy.ndimage to online docs (#1724) 2019-11-20 12:35:10 -08:00			`~~~~~~~~~~~~~~~~~~~~`
Add Sphinx-generated reference documentation for JAX. 2019-01-15 20:14:19 -05:00			`.. automodule:: jax.scipy.stats.norm`
Improve layout of jax.scipy documentation. Fixes #577, since the set of supported distributions is now relatively easy to see. 2019-04-15 12:40:18 -04:00			`.. autosummary::`
			`:toctree: _autosummary`

			`cdf`
			`logcdf`
			`logpdf`
			`pdf`
Add jax.scipy.stats.logistic to documentation. (#2149) 2020-02-03 12:44:57 -05:00			`ppf`
Add Sphinx-generated reference documentation for JAX. 2019-01-15 20:14:19 -05:00
			`jax.scipy.stats.uniform`
Add jax.scipy.ndimage to online docs (#1724) 2019-11-20 12:35:10 -08:00			`~~~~~~~~~~~~~~~~~~~~~~~`
Add Sphinx-generated reference documentation for JAX. 2019-01-15 20:14:19 -05:00			`.. automodule:: jax.scipy.stats.uniform`
Improve layout of jax.scipy documentation. Fixes #577, since the set of supported distributions is now relatively easy to see. 2019-04-15 12:40:18 -04:00			`.. autosummary::`
			`:toctree: _autosummary`

			`logpdf`
			`pdf`