Jean-Eric Campagne 4beee13ba0 Add implementation of jax.scipy.fftconvolve

2023-04-07 17:19:08 +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

Remove leading underscores in jax._src.numpy.util

2023-03-13 12:18:36 -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

Add implementation of jax.scipy.fftconvolve

2023-04-07 17:19:08 +02:00

special.py

Avoid imports from the public jax.* namespace in more places internally.

2023-04-04 11:41:40 -07:00