This is a pre-release version (2.0.0.dev0+git20260907.41eeb59).Go to latest (1.18.0)
{ } Raw JSON

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

source: build-install/usr/lib/python3.14/site-packages/scipy/_external/array_api_extra/_lib/_utils/_helpers.py:507

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