jax.extend.ffi module

ffi_call(target_name, result_shape_dtypes, *args)	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.