Jake VanderPlas a283aa0cc3 Deprecate three jax.Array methods: ...

- jax.Array.broadcast: use lax.broadcast instead
- jax.Array.broadcast_in_dim: use lax.broadcast_in_dim instead
- jax.Array.split: use jnp.split instead
These are removed because they are not part of the np.ndarray API.

2023-02-23 16:15:09 -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

Deprecate three jax.Array methods:

2023-02-23 16:15:09 -08:00

linalg.py

Remove typing_extensions dependency

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

ndarray.py

Add initial jax.Array base class for instance checks & annotation

2022-09-26 07:48:43 -07:00

polynomial.py

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

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

reductions.py

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

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

setops.py

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

2023-02-14 23:01:11 -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

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

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

vectorize.py

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

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