jax.ffi 模块

ffi_call()	调用外部函数接口 (FFI) 目标。
ffi_lowering(call_target_name, *[, ...])	为外部函数接口 (FFI) 目标构建降低规则。
pycapsule(funcptr)	将 ctypes 函数指针包装在 PyCapsule 中。
register_ffi_target(name, fn[, platform, ...])	注册一个外部函数目标。
register_ffi_type_id(name, obj[, platform])	为 FFI 目标注册自定义类型 ID。

jax.extend.ffi 模块（已弃用）

jax.extend.ffi 模块已移动到 jax.ffi，应使用该导入路径，但这些函数在旧版导入被弃用时仍在此处记录。

ffi_call()	调用外部函数接口 (FFI) 目标。
ffi_lowering(call_target_name, *[, ...])	为外部函数接口 (FFI) 目标构建降低规则。
pycapsule(funcptr)	将 ctypes 函数指针包装在 PyCapsule 中。
register_ffi_target(名称, 函数[, 平台, ...])	注册一个外部函数目标。