Peter Hawkins a4412e2715 Remove internal ndarray type name. Use Array throughout. ...

jax.numpy.ndarray remains an exported alias for jax.Array.

PiperOrigin-RevId: 513046188

2023-02-28 14:51:08 -08: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

[x64] more type safety in jax.scipy.signal

2022-12-01 13:43:07 -08:00

index_tricks.py

migrate more internal dependencies from jax.core to jax._src.core

2023-02-14 23:01:11 -08:00

lax_numpy.py

Remove internal ndarray type name. Use Array throughout.

2023-02-28 14:51:08 -08:00

linalg.py

Remove typing_extensions dependency

2022-12-05 15:42:26 -08:00

polynomial.py

migrate more internal dependencies from jax.core to jax._src.core

2023-02-14 23:01:11 -08:00

reductions.py

Remove internal ndarray type name. Use Array throughout.

2023-02-28 14:51:08 -08:00

setops.py

Replace jax._src.util.prod with math.prod.

2023-02-28 12:41:00 -08:00

ufuncs.py

migrate more internal dependencies from jax.core to jax._src.core

2023-02-14 23:01:11 -08:00

util.py

Remove internal ndarray type name. Use Array throughout.

2023-02-28 14:51:08 -08:00

vectorize.py

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

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