jax authors f73fa7a7ad Merge pull request #25290 from jakevdp:reduction-where ...

PiperOrigin-RevId: 703182008

2024-12-05 11:17:15 -08:00

..

__init__.py

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

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

array_api_metadata.py

[array api] Finalize array API in jax.numpy & deprecate jax.experimental.array_api

2024-08-01 11:19:17 -07:00

array_methods.py

Reverts 2075b091c4e83f0bdbd0d47812a72114fb8b937a

2024-11-19 14:47:24 -08:00

fft.py

Add lax.FftType.

2024-10-10 08:07:35 -07:00

index_tricks.py

Set __module__ attribute for objects in jax.numpy

2024-11-15 06:03:54 -08:00

lax_numpy.py

Fix indexing corner case with empty ellipses

2024-12-03 17:20:40 -08:00

linalg.py

jnp.linalg.vector_norm: properly support multiple axes

2024-12-05 09:48:32 -08:00

polynomial.py

Set __module__ attribute for objects in jax.numpy

2024-11-15 06:03:54 -08:00

reductions.py

jax.numpy: require boolean dtype for where argument

2024-12-05 09:27:19 -08:00

setops.py

Set __module__ attribute for objects in jax.numpy

2024-11-15 06:03:54 -08:00

ufunc_api.py

Set __module__ attribute for objects in jax.numpy

2024-11-15 06:03:54 -08:00

ufuncs.py

Make logaddexp and logaddexp2 into ufuncs

2024-11-18 09:27:36 -08:00

util.py

Remove obsolete implements() decorator & fix tests

2024-10-28 15:22:09 -07:00

vectorize.py

Set __module__ attribute for objects in jax.numpy

2024-11-15 06:03:54 -08:00