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.