bundles / scipy 1.18.0 / scipy / _external / array_api_extra / _lib / _utils / _helpers / _AutoJITWrapper

class

`scipy._external.array_api_extra._lib._utils._helpers:_AutoJITWrapper`

source: /scipy/_external/array_api_extra/_lib/_utils/_helpers.py:516

Signature

class _AutoJITWrapper(obj: T) → None

Members

Summary

Helper of jax_autojit.

Extended Summary

Wrap arbitrary inputs and outputs of the jitted function and convert them to/from PyTrees.

Aliases

scipy.differentiate.xpx._delegation._funcs._helpers._AutoJITWrapper