jax.extend.ffi 模块

ffi_call(target_name, result_shape_dtypes, *args)	调用外部函数接口 (FFI) 目标。
ffi_lowering(call_target_name, *[, ...])	为外部函数接口（FFI）目标构建一个降低规则。
pycapsule(funcptr)	将 ctypes 函数指针封装在 PyCapsule 中。
register_ffi_target(name, fn[, platform, ...])	注册一个外部函数目标。