__array_namespace_info__

Returns a namespace with Array API namespace inspection utilities.

JAX (Core)

API: jax.numpy.__array_namespace_info__
Strategy: Direct Mapping
Official Docs ↗

NumPy

API: numpy.__array_namespace_info__
Strategy: Direct Mapping
Official Docs ↗

Flax NNX

API: jax.numpy.__array_namespace_info__
Strategy: Direct Mapping
Official Docs ↗