laqua-stack d742733bea 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

..

cluster

Remove leading underscores in jax._src.numpy.util

2023-03-13 12:18:36 -07:00

interpolate

Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".

2022-09-22 12:27:19 -07:00

optimize

Remove leading underscores in jax._src.numpy.util

2023-03-13 12:18:36 -07:00

sparse

Replace uses of jnp.ndarray with jax.Array inside JAX.

2023-02-15 14:53:00 -08:00

stats

Fix loc and scale parameters in scipy.logistic. Add CDF and SF for several distributions.

2023-03-21 00:16:13 +01:00

__init__.py

Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".

2022-09-22 12:27:19 -07:00

fft.py

implement idct and idctn + add function to scipy.rst

2023-04-17 12:12:51 -07:00

linalg.py

jax.scipy.linalg.expm: support batched inputs

2023-03-27 16:39:48 -07:00

ndimage.py

internal: avoid unused imports in lax_numpy

2023-03-08 10:29:04 -08:00

signal.py

Remove use of int casting in STFT collapse of batch dimensions.

2023-04-13 15:15:11 -07:00

special.py

feat (scipy.special): Add a xla version of scipy.special.gamma function

2023-04-18 21:10:22 +02:00