bundles / scipy 1.17.1 / scipy / _lib / array_api_extra / _lib / _lazy / _is_jax_jit_enabled

function

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

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