jax.extend.ffi module .. automodule:: jax.extend.ffi .. autosummary:: :toctree: _autosummary ffi_call ffi_lowering pycapsule register_ffi_target