papyri
{ } Raw JSON

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

function

scipy._lib.array_api_compat.common._helpers:is_jax_namespace

source: /scipy/_lib/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