jax.typing.ArrayLike

jax.typing.ArrayLike

Type annotation for JAX array-like objects.

alias of Array | ndarray | bool_ | number | bool | int | float | complex