papyri
This is a pre-release version (2.0.0.dev0+git20260907.41eeb59).Go to latest (1.18.0)
{ } Raw JSON

bundles / scipy 2.0.0.dev0+git20260907.41eeb59 / scipy / _external / array_api_compat / common / _helpers / is_jax_array

function

scipy._external.array_api_compat.common._helpers:is_jax_array

source: build-install/usr/lib/python3.14/site-packages/scipy/_external/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.special._support_alternative_backends.is_jax_array