### Description <!-- Describe your request. --> https://docs.jax.dev/en/latest/_autosummary/jax.ffi.ffi_call.html ### Context <!-- What is the motivation for this request? --> https://github.com/NVIDIA/warp/pull/787
