bundles / scipy 1.17.1 / scipy / _lib / array_api_compat / common / _helpers / is_jax_array

function

`scipy._lib.array_api_compat.common._helpers:is_jax_array`

source: /scipy/_lib/array_api_compat/common/_helpers.py:217

Signature

def is_jax_array(x: object) → TypeIs[jax.Array]

Summary

Return True if x is a JAX array.

Extended Summary

This function does not import JAX if it has not already been imported and is therefore cheap to use.

See also

array_namespace
is_array_api_obj
is_cupy_array
is_dask_array
is_ndonnx_array
is_numpy_array
is_pydata_sparse_array
is_torch_array

Aliases

scipy.differentiate.xpx._delegation._funcs.is_jax_array