jax.ffi module

ffi_call()	Call a foreign function interface (FFI) target.
ffi_lowering(call_target_name, *[, ...])	Build a lowering rule for an foreign function interface (FFI) target.
pycapsule(funcptr)	Wrap a ctypes function pointer in a PyCapsule.
register_ffi_target(name, fn[, platform, ...])	Registers a foreign function target.
register_ffi_type_id(name, obj[, platform])	Registers a custom type ID for a FFI target.

jax.extend.ffi module (deprecated)

The jax.extend.ffi module has been moved to jax.ffi, and that import path should be used instead, but these functions remain documented here while the legacy import is being deprecated.

ffi_call()	Call a foreign function interface (FFI) target.
ffi_lowering(call_target_name, *[, ...])	Build a lowering rule for an foreign function interface (FFI) target.
pycapsule(funcptr)	Wrap a ctypes function pointer in a PyCapsule.
register_ffi_target(name, fn[, platform, ...])	Registers a foreign function target.