bundles / scipy 1.17.1 / scipy / _lib / array_api_extra / _lib / _utils / _helpers / _AutoJITWrapper

class

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

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