mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-15 19:36:06 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/docs/jax.scipy.rst

463 lines
6.3 KiB
ReStructuredText
Raw Normal View History

DOC: make API doc titles more uniform
2023-01-18 10:59:42 -08:00
``jax.scipy`` module
====================
Add Sphinx-generated reference documentation for JAX.
2019-01-15 20:14:19 -05:00
Support scipy.fft.dct/dctn type=2
2021-08-17 15:20:13 +02:00
jax.scipy.fft
-------------
.. automodule:: jax.scipy.fft
.. autosummary::
:toctree: _autosummary
dct
dctn
implement idct and idctn + add function to scipy.rst
2023-04-17 11:57:07 -07:00
idct
idctn
Support scipy.fft.dct/dctn type=2
2021-08-17 15:20:13 +02:00
Deprecate jax.numpy.trapz. Expose the current implementation of jax.numpy.trapz as jax.scipy.integrate.trapezoid instead. Fixes https://github.com/google/jax/issues/17244
2023-08-24 14:01:40 -06:00
jax.scipy.integrate
-------------------
.. automodule:: jax.scipy.integrate
.. autosummary::
:toctree: _autosummary
trapezoid
Add Sphinx-generated reference documentation for JAX.
2019-01-15 20:14:19 -05:00
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
DOC: add missing linalg functionality to docs
2022-03-15 09:55:59 -07:00
eigh_tridiagonal
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
Add jax.scipy.linalg.funm
2022-04-26 18:03:14 +01:00
funm
Add support for Hessenberg and tridiagonal matrix reductions on CPU. * Implement jax.scipy.linalg.hessenberg and jax.lax.linalg.hessenberg. * Export what was previously jax._src.lax.linalg.orgqr as jax.lax.linalg.householder_product, since it can be used with some minor tweaks to compute the unitary matrix of a Hessenberg reduction. * Implement jax.lax.linalg.tridiagonal, which is the symmetric (Hermitian) equivalent of Hessenberg reduction. None of these primitives are differentiable at the moment. PiperOrigin-RevId: 487224934
2022-11-09 06:23:22 -08:00
hessenberg
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
DOC: add missing linalg functionality to docs
2022-03-15 09:55:59 -07:00
polar
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
Add jax.scipy.linalg.rsf2csf
2022-03-27 12:31:12 +01:00
rsf2csf
Add jax.distributed and jax.dlpack to the docs. Reorder the doc modules into something closer to alphabetical order. Add missing functions from jax.scipy.linalg and jax.scipy.signal to the docs.
2022-02-17 16:05:33 -05:00
schur
sqrtm
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
solve
solve_triangular
DOC: add missing linalg functionality to docs
2022-03-15 09:55:59 -07:00
sqrtm
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
svd
Add `jax.scipy.linalg.toeplitz`.
2022-11-15 18:40:52 +09:00
toeplitz
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 jax.scipy.optimize to HTML docs Fixes #5300
2021-01-03 06:24:42 -08:00
jax.scipy.optimize
------------------
.. automodule:: jax.scipy.optimize
.. autosummary::
:toctree: _autosummary
minimize
Remove message field from OptimizeResults to allow for vmapping. Add OptimizeResults to the documentation.
2021-02-16 20:43:53 -05:00
OptimizeResults
Add jax.scipy.optimize to HTML docs Fixes #5300
2021-01-03 06:24:42 -08:00
Add missing functions to autodoc
2020-04-13 13:46:04 -07:00
jax.scipy.signal
----------------
.. automodule:: jax.scipy.signal
.. autosummary::
:toctree: _autosummary
Add implementation of jax.scipy.fftconvolve
2023-04-07 17:19:08 +02:00
fftconvolve
Add missing functions to autodoc
2020-04-13 13:46:04 -07:00
convolve
convolve2d
correlate
correlate2d
Add jax.distributed and jax.dlpack to the docs. Reorder the doc modules into something closer to alphabetical order. Add missing functions from jax.scipy.linalg and jax.scipy.signal to the docs.
2022-02-17 16:05:33 -05:00
csd
Add istft to jax.scipy.signal.
2022-03-28 21:33:40 +09:00
istft
Add jax.distributed and jax.dlpack to the docs. Reorder the doc modules into something closer to alphabetical order. Add missing functions from jax.scipy.linalg and jax.scipy.signal to the docs.
2022-02-17 16:05:33 -05:00
stft
welch
Add missing functions to autodoc
2020-04-13 13:46:04 -07:00
Added scipy.spatial.transform Rotation and Slerp classes
2023-06-02 22:38:15 -05:00
jax.scipy.spatial.transform
---------------------------
.. automodule:: jax.scipy.spatial.transform
.. autosummary::
:toctree: _autosummary
Rotation
Slerp
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
DOC: add bicgstab to doc page
2021-05-04 12:24:50 -07:00
bicgstab
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
cg
Cleanup/fixup jax.scipy.sparse.linalg.gmres and expose it publicly.
2020-11-27 14:44:06 -08:00
gmres
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
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 jax.scipy.special.bernoulli
2023-08-23 12:58:37 -07:00
bernoulli
Add beta function
2023-11-05 15:37:38 -08:00
beta
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
Added more accurate version of the betaln function.
2022-11-16 20:26:20 -08:00
betaln
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
add support for scipy.special.{expn,expi,exp1}
2021-08-21 16:03:43 -04:00
exp1
expi
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
expit
add support for scipy.special.{expn,expi,exp1}
2021-08-21 16:03:43 -04:00
expn
Add special.factorial function
2023-12-04 06:13:14 -08:00
factorial
feat (scipy.special): Add a xla version of scipy.special.gamma function - Add gamma fcn api in scipy.special - Add tests for this purpose - Add function to the docs Currently, there is no implementation of the gamma function in jax but there is one in scipy.special. This breaks some higher level jit-compilation like in the blackjax backend for pymc. This commit adds the missing gamma function. Resolves: #15409
2023-04-18 21:10:22 +02:00
gamma
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
implementation of poch and hyp1f1
2023-11-11 16:36:31 +01:00
hyp1f1
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
Adds associated Legendre functions of the first kind. Co-authored-by: Jake VanderPlas <jakevdp@google.com>
2021-06-02 11:37:37 -07:00
lpmn
Support value computation of associated Legendre functions. Co-authored-by: Jake VanderPlas <jakevdp@google.com>
2021-06-14 14:51:37 -07:00
lpmn_values
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
implementation of poch and hyp1f1
2023-11-11 16:36:31 +01:00
poch
add scipy.special.zeta, scipy.special.polygamma (#3385)
2020-06-15 22:21:26 -04:00
polygamma
added scipy special spence added dtype to arrays in the _spence_poly function
2023-02-10 18:09:31 +00:00
spence
Adds spherical harmonics. Co-authored-by: Jake VanderPlas <jakevdp@google.com>
2021-07-02 10:42:29 -07:00
sph_harm
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
Add kl_div and rel_entr functions Their behavior is the same as functions in scipy.special. The only small difference is in rel_entr function, which unlike scipy.special does not take the optional parameter 'out'. Resolves #16630
2023-07-08 12:27:18 +00:00
kl_div
rel_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
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
Add docs & changelog for jax.scipy.stats.mode
2022-10-20 15:55:57 -07:00
.. automodule:: jax.scipy.stats
.. autosummary::
:toctree: _autosummary
mode
Added scipy.stats.rankdata
2023-02-07 12:07:00 -05:00
rankdata
Add scipy.stats.sem.
2024-01-13 22:17:21 -05:00
sem
Add docs & changelog for jax.scipy.stats.mode
2022-10-20 15:55:57 -07:00
Add missing jax.scipy.stats distributions to the docs. Alphabetize import order.
2020-10-20 09:47:35 -04:00
jax.scipy.stats.bernoulli
Fix RST annotations
2021-01-14 14:38:39 -08:00
~~~~~~~~~~~~~~~~~~~~~~~~~
Add missing jax.scipy.stats distributions to the docs. Alphabetize import order.
2020-10-20 09:47:35 -04:00
.. automodule:: jax.scipy.stats.bernoulli
.. autosummary::
:toctree: _autosummary
logpmf
pmf
Added scipy.stats.bernoulli cdf and ppf.
2022-12-16 10:12:40 +00:00
cdf
ppf
Add missing jax.scipy.stats distributions to the docs. Alphabetize import order.
2020-10-20 09:47:35 -04: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
Add missing jax.scipy.stats distributions to the docs. Alphabetize import order.
2020-10-20 09:47:35 -04:00
.. autosummary::
:toctree: _autosummary
logpdf
pdf
Fix loc and scale parameters in scipy.logistic. Add CDF and SF for several distributions.
2023-03-12 06:53:09 +01:00
cdf
logcdf
sf
jax.scipy.stats: add logsf & make sf more accurate near zero
2023-08-22 14:44:32 -07:00
logsf
Add missing jax.scipy.stats distributions to the docs. Alphabetize import order.
2020-10-20 09:47:35 -04:00
Add betabinomial logpmf/pmf and tests Squash all changes to single commit. Add betabinom Add tests for betabinom. nan where undefefined squash
2021-02-05 15:13:09 -05:00
jax.scipy.stats.betabinom
DOC: fix minor rst formatting issues
2021-02-09 09:49:36 -08:00
~~~~~~~~~~~~~~~~~~~~~~~~~
Add betabinomial logpmf/pmf and tests Squash all changes to single commit. Add betabinom Add tests for betabinom. nan where undefefined squash
2021-02-05 15:13:09 -05:00
.. automodule:: jax.scipy.stats.betabinom
Add missing jax.scipy.stats docs
2023-08-17 12:56:15 -07:00
.. autosummary::
:toctree: _autosummary
logpmf
pmf
jax.scipy.stats.binom
~~~~~~~~~~~~~~~~~~~~~
.. automodule:: jax.scipy.stats.binom
Add betabinomial logpmf/pmf and tests Squash all changes to single commit. Add betabinom Add tests for betabinom. nan where undefefined squash
2021-02-05 15:13:09 -05:00
.. autosummary::
:toctree: _autosummary
logpmf
pmf
Add missing jax.scipy.stats distributions to the docs. Alphabetize import order.
2020-10-20 09:47:35 -04:00
jax.scipy.stats.cauchy
Fix RST annotations
2021-01-14 14:38:39 -08:00
~~~~~~~~~~~~~~~~~~~~~~
Add missing jax.scipy.stats distributions to the docs. Alphabetize import order.
2020-10-20 09:47:35 -04:00
.. automodule:: jax.scipy.stats.cauchy
.. autosummary::
:toctree: _autosummary
logpdf
pdf
Fix loc and scale parameters in scipy.logistic. Add CDF and SF for several distributions.
2023-03-12 06:53:09 +01:00
cdf
logcdf
sf
jax.scipy.stats: add logsf & make sf more accurate near zero
2023-08-22 14:44:32 -07:00
logsf
Fix loc and scale parameters in scipy.logistic. Add CDF and SF for several distributions.
2023-03-12 06:53:09 +01:00
isf
ppf
Add missing jax.scipy.stats distributions to the docs. Alphabetize import order.
2020-10-20 09:47:35 -04:00
Add logdf and pdf for chisquare distribution Add tests Lint with flake8 fails. Should pass now newline at end of file for flake8 docs and changes remove whitespace in changeloc
2021-02-03 15:09:21 -05:00
jax.scipy.stats.chi2
DOC: fix minor rst formatting issues
2021-02-09 09:49:36 -08:00
~~~~~~~~~~~~~~~~~~~~
Add logdf and pdf for chisquare distribution Add tests Lint with flake8 fails. Should pass now newline at end of file for flake8 docs and changes remove whitespace in changeloc
2021-02-03 15:09:21 -05:00
.. automodule:: jax.scipy.stats.chi2
.. autosummary::
:toctree: _autosummary
logpdf
pdf
Fix loc and scale parameters in scipy.logistic. Add CDF and SF for several distributions.
2023-03-12 06:53:09 +01:00
cdf
logcdf
sf
jax.scipy.stats: add logsf & make sf more accurate near zero
2023-08-22 14:44:32 -07:00
logsf
Add logdf and pdf for chisquare distribution Add tests Lint with flake8 fails. Should pass now newline at end of file for flake8 docs and changes remove whitespace in changeloc
2021-02-03 15:09:21 -05:00
Add missing jax.scipy.stats distributions to the docs. Alphabetize import order.
2020-10-20 09:47:35 -04:00
jax.scipy.stats.dirichlet
Fix RST annotations
2021-01-14 14:38:39 -08:00
~~~~~~~~~~~~~~~~~~~~~~~~~
Add missing jax.scipy.stats distributions to the docs. Alphabetize import order.
2020-10-20 09:47:35 -04:00
.. automodule:: jax.scipy.stats.dirichlet
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
Fix loc and scale parameters in scipy.logistic. Add CDF and SF for several distributions.
2023-03-12 06:53:09 +01:00
cdf
logcdf
sf
jax.scipy.stats: add logsf & make sf more accurate near zero
2023-08-22 14:44:32 -07:00
logsf
Add Sphinx-generated reference documentation for JAX.
2019-01-15 20:14:19 -05:00
Added scipy.stats.gennorm.
2022-06-11 14:24:19 -04:00
jax.scipy.stats.gennorm
~~~~~~~~~~~~~~~~~~~~~~~
.. automodule:: jax.scipy.stats.gennorm
.. autosummary::
:toctree: _autosummary
cdf
logpdf
pdf
Add missing jax.scipy.stats distributions to the docs. Alphabetize import order.
2020-10-20 09:47:35 -04:00
jax.scipy.stats.geom
~~~~~~~~~~~~~~~~~~~~
.. automodule:: jax.scipy.stats.geom
.. autosummary::
:toctree: _autosummary
logpmf
pmf
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 missing jax.scipy.stats docs
2023-08-17 12:56:15 -07:00
jax.scipy.stats.multinomial
~~~~~~~~~~~~~~~~~~~~~~~~~~~
.. automodule:: jax.scipy.stats.multinomial
.. autosummary::
:toctree: _autosummary
logpmf
pmf
Add missing jax.scipy.stats distributions to the docs. Alphabetize import order.
2020-10-20 09:47:35 -04:00
jax.scipy.stats.multivariate_normal
Fix RST annotations
2021-01-14 14:38:39 -08:00
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
Add missing jax.scipy.stats distributions to the docs. Alphabetize import order.
2020-10-20 09:47:35 -04:00
.. automodule:: jax.scipy.stats.multivariate_normal
.. autosummary::
:toctree: _autosummary
logpdf
pdf
Add missing jax.scipy.stats docs
2023-08-17 12:56:15 -07:00
jax.scipy.stats.nbinom
~~~~~~~~~~~~~~~~~~~~~~
.. automodule:: jax.scipy.stats.nbinom
.. autosummary::
:toctree: _autosummary
logpmf
pmf
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
logpdf
pdf
jax.scipy.stats: add logsf & make sf more accurate near zero
2023-08-22 14:44:32 -07:00
cdf
logcdf
Add jax.scipy.stats.logistic to documentation. (#2149)
2020-02-03 12:44:57 -05:00
ppf
Fix loc and scale parameters in scipy.logistic. Add CDF and SF for several distributions.
2023-03-12 06:53:09 +01:00
sf
jax.scipy.stats: add logsf & make sf more accurate near zero
2023-08-22 14:44:32 -07:00
logsf
Fix loc and scale parameters in scipy.logistic. Add CDF and SF for several distributions.
2023-03-12 06:53:09 +01:00
isf
Add Sphinx-generated reference documentation for JAX.
2019-01-15 20:14:19 -05:00
Add missing jax.scipy.stats distributions to the docs. Alphabetize import order.
2020-10-20 09:47:35 -04:00
jax.scipy.stats.pareto
Fix RST annotations
2021-01-14 14:38:39 -08:00
~~~~~~~~~~~~~~~~~~~~~~
Add missing jax.scipy.stats distributions to the docs. Alphabetize import order.
2020-10-20 09:47:35 -04:00
.. automodule:: jax.scipy.stats.pareto
.. autosummary::
:toctree: _autosummary
logpdf
pdf
jax.scipy.stats.poisson
~~~~~~~~~~~~~~~~~~~~~~~
.. automodule:: jax.scipy.stats.poisson
.. autosummary::
:toctree: _autosummary
logpmf
pmf
jax.scipy.stats.t
Fix RST annotations
2021-01-14 14:38:39 -08:00
~~~~~~~~~~~~~~~~~
Add missing jax.scipy.stats distributions to the docs. Alphabetize import order.
2020-10-20 09:47:35 -04:00
.. automodule:: jax.scipy.stats.t
.. autosummary::
:toctree: _autosummary
logpdf
pdf
implement truncnorm in jax.scipy.stats fix some shape and type issues import into namespace imports into non-_src library working logpdf test cleanup working tests for cdf and sf after fixing select relax need for x to be in (a, b) ensure behavior with invalid input matches scipy remove enforcing valid parameters in tests added truncnorm to docs whoops alphabetical fix linter error fix circular import issue
2022-10-03 17:46:28 -04:00
jax.scipy.stats.truncnorm
~~~~~~~~~~~~~~~~~~~~~~~~~
.. automodule:: jax.scipy.stats.truncnorm
.. autosummary::
:toctree: _autosummary
cdf
logcdf
logpdf
logsf
pdf
sf
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
Add .hypothesis/ directory to .gitignore and ppf and cdf to scipy.stats.uniform
2024-01-13 13:01:16 +00:00
cdf
ppf
Implementation of jax.scipy.stats.gaussian_kde
2022-06-23 10:00:54 -04:00
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
Added vonmises pdf, logpdf & respective tests. Added vonmises pdf, logpdf & respective tests. Altered type-hinting, added pi as a _lax_const Changed lax constant pi to be created in _pdf instead of passed arg. Changed name in __init__.py Fixed bug in tests. Review related alterations. Review related changes. Added vonmises pdf, logpdf & respective tests. Added vonmises pdf, logpdf & respective tests. Altered type-hinting, added pi as a _lax_const Changed lax constant pi to be created in _pdf instead of passed arg. Changed name in __init__.py Fixed bug in tests. Review related alterations. PR PR PR
2022-11-22 01:16:08 +00:00
jax.scipy.stats.vonmises
~~~~~~~~~~~~~~~~~~~~~~~~
.. automodule:: jax.scipy.stats.vonmises
.. autosummary::
:toctree: _autosummary
logpdf
Add kl_div and rel_entr functions Their behavior is the same as functions in scipy.special. The only small difference is in rel_entr function, which unlike scipy.special does not take the optional parameter 'out'. Resolves #16630
2023-07-08 12:27:18 +00:00
pdf
feat: add wrapcauchy logpdf and pdf
2023-10-16 03:45:24 +02:00
jax.scipy.stats.wrapcauchy
~~~~~~~~~~~~~~~~~~~~~~~~~~
.. automodule:: jax.scipy.stats.wrapcauchy
.. autosummary::
:toctree: _autosummary
logpdf
pdf
Reference in New Issue Copy Permalink