jax.extend.ffi module

jax.extend.ffi module#

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.