Jake VanderPlas 9e01afe7af Add jax.numpy.trapezoid ...

This function has been added to NumPy in version 2.0, as a replacement
for the already deprecated trapz function.

2024-04-01 13:05:20 -07:00

..

__init__.py

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

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

array_methods.py

jnp.ndarray.item(): add args support

2024-01-03 13:03:47 -08:00

fft.py

Updated

2024-02-05 18:01:48 -05:00

index_tricks.py

Replaced most usages of abc.ABC with util.StrictABC

2024-01-29 12:40:43 +00:00

lax_numpy.py

Add jax.numpy.trapezoid

2024-04-01 13:05:20 -07:00

linalg.py

[JAX] Add an option subset_by_index that allows computing a contiguous subset of singular components from svd.

2024-02-15 16:33:09 -08:00

polynomial.py

Repair various type errors

2024-03-13 15:13:56 -04:00

reductions.py

Fix axis=None and keepdims=True for jnp.quantile and jnp.median

2024-03-15 09:05:02 +08:00

setops.py

Rename _wraps to implements

2024-01-24 14:14:19 -08:00

ufunc_api.py

Import submodules from jax._src explicitly, instead of relying on import side-effects. It will lead to the missing x-refs in code search according to go/pywald-sawmill-analysis.

2024-02-06 15:47:16 -08:00

ufuncs.py

Rename _wraps to implements

2024-01-24 14:14:19 -08:00

util.py

Rename _wraps to implements

2024-01-24 14:14:19 -08:00

vectorize.py

jnp.vectorize: support excluding arguments by keyword

2023-12-20 15:38:19 -08:00