Parker Schuh d62fc88fb1 Roll back #14792 ...

Breaks tests. lax.sub requires arguments to have the same dtypes, got float32, float64. (Tip: jnp.subtract is a similar function that does automatic type promotion on inputs).

PiperOrigin-RevId: 514897538

2023-03-07 18:31:19 -08:00

..

cluster

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

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

interpolate

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

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

optimize

Reenable pytype for numpy ufuncs.

2023-03-03 05:01:03 -08:00

sparse

Replace uses of jnp.ndarray with jax.Array inside JAX.

2023-02-15 14:53:00 -08:00

stats

Roll back #14792

2023-03-07 18:31:19 -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

[typing] add annotations to jax.scipy.fft

2022-10-09 05:18:45 -07:00

linalg.py

Removed deprecated polar_unitary as per comment.

2023-01-27 07:24:55 +00:00

ndimage.py

[typing] annotate jax.numpy ufuncs

2022-10-20 11:22:04 -07:00

signal.py

Add Optional to boundary arg of stft.

2022-12-30 22:42:45 -05:00

special.py

DOC: fix typo in sph_harm

2023-02-16 15:34:18 +01:00