Peter Hawkins 7de9eb20df Reverts 525b646c0ebd5205f4fa0639c94adb2de47e1cf0 ...

PiperOrigin-RevId: 707146329

2024-12-17 10:12:34 -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 525b646c0ebd5205f4fa0639c94adb2de47e1cf0

2024-12-17 10:12:34 -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

Reverts 525b646c0ebd5205f4fa0639c94adb2de47e1cf0

2024-12-17 10:12:34 -08:00

linalg.py

Fix type annotations for NumPy 2.2

2024-12-11 14:24:58 -08:00

polynomial.py

Fix type annotations for NumPy 2.2

2024-12-11 14:24:58 -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

[sharding_in_types] Make jnp.where broadcast shardings properly when a scalar exists

2024-12-11 16:41:18 -08:00

vectorize.py

Set __module__ attribute for objects in jax.numpy

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