papyri
{ } Raw JSON

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