bundles / scipy 1.18.0 / scipy / _external / array_api_extra / _lib / _lazy / _is_jax_jit_enabled

function

`scipy._external.array_api_extra._lib._lazy:_is_jax_jit_enabled`

source: /scipy/_external/array_api_extra/_lib/_lazy.py:305

Signature

def _is_jax_jit_enabled(xp: ModuleType) → bool

Summary

Return True if this function is being called inside jax.jit.

Aliases

scipy.differentiate.xpx._lib._lazy._is_jax_jit_enabled