bundles / scipy 2.0.0.dev0+git20260907.41eeb59 / scipy / _external / array_api_extra / _lib / _utils / _helpers / _AutoJITWrapper
class
scipy._external.array_api_extra._lib._utils._helpers:_AutoJITWrapper
Signature
class _AutoJITWrapper(obj: T) → NoneMembers
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