jax.ffi module
jax.ffi module#
| Call a foreign function interface (FFI) target. |
| Build a lowering rule for an foreign function interface (FFI) target. |
| Wrap a ctypes function pointer in a PyCapsule. |
| Registers a foreign function target. |
| Registers a custom type for a FFI target. |
