papyri
{ } Raw JSON

bundles / scipy 1.18.0 / scipy / _external / array_api_compat / common / _helpers / is_jax_namespace

function

scipy._external.array_api_compat.common._helpers:is_jax_namespace

source: /scipy/_external/array_api_compat/common/_helpers.py:425

Signature

def is_jax_namespace(xp: Namespace) → bool

Summary

Returns True if xp is a JAX namespace.

Extended Summary

This includes jax.numpy and jax.experimental.array_api which existed in older versions of JAX.

See also

array_namespace
is_array_api_strict_namespace
is_cupy_namespace
is_dask_namespace
is_ndonnx_namespace
is_numpy_namespace
is_pydata_sparse_namespace
is_torch_namespace

Aliases

  • scipy.signal._filter_design.is_jax